From c4c4b6e9cd4185de9d5f9f26b7babe626c47d54d Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Wed, 4 Dec 2024 12:01:09 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E9=A1=B9=E7=9B=AE?=
 =?UTF-8?q?=E9=87=8D=E5=90=AF=E5=90=8E=E6=96=87=E6=A1=A3=E7=8A=B6=E6=80=81?=
 =?UTF-8?q?=E4=B8=8D=E6=AD=A3=E7=A1=AE=20(#1754)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/common/event/__init__.py | 6 ++++++
 apps/smartdoc/settings/lib.py | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/apps/common/event/__init__.py b/apps/common/event/__init__.py
index 6b6d0541a..2ae40525b 100644
--- a/apps/common/event/__init__.py
+++ b/apps/common/event/__init__.py
@@ -10,8 +10,14 @@ import setting.models
 from setting.models import Model
 from .listener_manage import *
 
+update_document_status_sql = """
+UPDATE "public"."document" 
+SET status ="replace"("replace"("replace"(status, '1', '3'), '0', '3'), '4', '3')
+"""
+
 
 def run():
     # QuerySet(Document).filter(status__in=[Status.embedding, Status.queue_up]).update(**{'status': Status.error})
     QuerySet(Model).filter(status=setting.models.Status.DOWNLOAD).update(status=setting.models.Status.ERROR,
                                                                          meta={'message': "下载程序被中断,请重试"})
+    update_execute(update_document_status_sql, [])
diff --git a/apps/smartdoc/settings/lib.py b/apps/smartdoc/settings/lib.py
index a4c1aaabb..fc1d3244f 100644
--- a/apps/smartdoc/settings/lib.py
+++ b/apps/smartdoc/settings/lib.py
@@ -7,6 +7,7 @@
     @desc:
 """
 import os
+import shutil
 
 from smartdoc.const import CONFIG, PROJECT_DIR
 
@@ -34,6 +35,11 @@ CELERY_TASK_SOFT_TIME_LIMIT = 3600
 CELERY_WORKER_CANCEL_LONG_RUNNING_TASKS_ON_CONNECTION_LOSS = True
 CELERY_ACKS_LATE = True
 celery_once_path = os.path.join(celery_data_dir, "celery_once")
+try:
+    if os.path.exists(celery_once_path) and os.path.isdir(celery_once_path):
+        shutil.rmtree(celery_once_path)
+except Exception as e:
+    pass
 CELERY_ONCE = {
     'backend': 'celery_once.backends.File',
     'settings': {'location': celery_once_path}