From 8664840f7bdbe5a8ff99a05f202ae46a94dadcbc Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Mon, 2 Dec 2024 14:02:01 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=8E=86=E5=8F=B2?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E7=8A=B6=E6=80=81=E9=94=99=E8=AF=AF=20(#1732?= =?UTF-8?q?)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ...nt_status_meta_paragraph_status_meta_and_more.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/apps/dataset/migrations/0011_document_status_meta_paragraph_status_meta_and_more.py b/apps/dataset/migrations/0011_document_status_meta_paragraph_status_meta_and_more.py index 4fc9cc8b7..7c5d63755 100644 --- a/apps/dataset/migrations/0011_document_status_meta_paragraph_status_meta_and_more.py +++ b/apps/dataset/migrations/0011_document_status_meta_paragraph_status_meta_and_more.py @@ -9,18 +9,16 @@ from dataset.models import State, TaskType sql = """ UPDATE "document" -SET status ="replace"(status, '1', '3') +SET status ="replace"("replace"("replace"(status, '2', '3'),'0','3'),'1','2') +""" +sql_paragraph = """ +UPDATE "paragraph" +SET status ="replace"("replace"("replace"(status, '2', '3'),'0','3'),'1','2') """ def updateDocumentStatus(apps, schema_editor): - ParagraphModel = apps.get_model('dataset', 'Paragraph') DocumentModel = apps.get_model('dataset', 'Document') - success_list = QuerySet(DocumentModel).filter(status='2') - if len(success_list) == 0: - return - ListenerManagement.update_status(QuerySet(ParagraphModel).filter(document_id__in=[d.id for d in success_list]), - TaskType.EMBEDDING, State.SUCCESS) ListenerManagement.get_aggregation_document_status_by_query_set(QuerySet(DocumentModel))() @@ -50,6 +48,7 @@ class Migration(migrations.Migration): name='status', field=models.CharField(default=dataset.models.data_set.Status.__str__, max_length=20, verbose_name='状态'), ), + migrations.RunSQL(sql_paragraph), migrations.RunSQL(sql), migrations.RunPython(updateDocumentStatus) ]