zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复历史数据状态错误 (#1732)

Browse Source
This commit is contained in:
shaohuzhang1 2024-12-02 14:02:01 +08:00 committed by GitHub
parent 2084112d54
commit 8664840f7b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
apps/dataset/migrations/0011_document_status_meta_paragraph_status_meta_and_more.py
View File

@ -9,18 +9,16 @@ from dataset.models import State, TaskType
sql = """ sql = """
UPDATE "document" 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): def updateDocumentStatus(apps, schema_editor):
ParagraphModel = apps.get_model('dataset', 'Paragraph')
DocumentModel = apps.get_model('dataset', 'Document') 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))() ListenerManagement.get_aggregation_document_status_by_query_set(QuerySet(DocumentModel))()
@ -50,6 +48,7 @@ class Migration(migrations.Migration):
name='status', name='status',
field=models.CharField(default=dataset.models.data_set.Status.__str__, max_length=20, verbose_name='状态'), field=models.CharField(default=dataset.models.data_set.Status.__str__, max_length=20, verbose_name='状态'),
), ),
migrations.RunSQL(sql_paragraph),
migrations.RunSQL(sql), migrations.RunSQL(sql),
migrations.RunPython(updateDocumentStatus) migrations.RunPython(updateDocumentStatus)
] ]