15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refactor: improve document status update logic and enhance post handler functionality

This commit is contained in:
CaptainB 2025-07-02 11:44:33 +08:00
parent b71a854d7c
commit 0ad38e2540
2 changed files with 28 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
apps/common/event/__init__.py
View File

@ -6,25 +6,32 @@
@date：2023/11/10 10:43
@desc:
"""
from models_provider.models import Model, Status
from .listener_manage import *
from django.utils.translation import gettext as _
from ..db.sql_execute import update_execute
from common.lock.impl.file_lock import FileLock
from .listener_manage import *
from ..db.sql_execute import update_execute
lock = FileLock()
update_document_status_sql = """
UPDATE "public"."document"
SET status ="replace"("replace"("replace"(status, '1', '3'), '0', '3'), '4', '3')
WHERE status ~ '1|0|4'
"""
UPDATE "public"."document"
SET status ="replace"("replace"("replace"(status, '1', '3'), '0', '3'), '4', '3')
WHERE status ~ '1|0|4' \
"""
def run():
from models_provider.models import Model, Status
if lock.try_lock('event_init', 30 * 30):
try:
QuerySet(Model).filter(status=Status.DOWNLOAD).update(status=Status.ERROR, meta={'message': _( 'The download process was interrupted, please try again')})
# 修改Model状态为ERROR
QuerySet(Model).filter(
status=Status.DOWNLOAD
).update(
status=Status.ERROR, meta={'message': _('The download process was interrupted, please try again')}
)
# 更新文档状态
update_execute(update_document_status_sql, [])
finally:
lock.un_lock('event_init')

13
apps/maxkb/wsgi.py
View File

@ -14,3 +14,16 @@ from django.core.wsgi import get_wsgi_application
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'maxkb.settings')
application = get_wsgi_application()
def post_handler():
from common.database_model_manage.database_model_manage import DatabaseModelManage
from common import job
from common import event
event.run()
job.run()
DatabaseModelManage.init()
post_handler()