添加fsm 任务

王聪
Showing 1 changed file with 8 additions and 3 deletions
src/apps/doc/management/commands/ocr_process.py
--- a/src/apps/doc/management/commands/ocr_process.py
View file @ce75072
+++ b/src/apps/doc/management/commands/ocr_process.py
View file @ce75072
@@ -41,7 +41,7 @@ from apps.doc.models import (
    IDBCRecords,
    Configs,
 )
-from celery_compare.tasks import compare
+from celery_compare.tasks import compare, fsm_compare
 class Command(BaseCommand, LoggerMixin):
@@ -2014,8 +2014,13 @@ class Command(BaseCommand, LoggerMixin):
                                        self.log_base, task_str, res_obj.id))
                                    # 触发比对
                                    try:
-                                        # pass
+                                        fsm_activited = getattr(res_obj, 'fsm_activited')
-                                        compare.apply_async((doc.application_id, business_type, None, res_obj.id,
+                                        if fsm_activited:
+                                            fsm_compare.apply_async((doc.application_id, business_type, None, res_obj.id, is_ca, True),
+                                                queue='queue_compare')
+                                        else:
+                                            # pass
+                                            compare.apply_async((doc.application_id, business_type, None, res_obj.id,
                                                             is_ca, True), queue='queue_compare')
                                    except Exception as e:
                                        self.online_log.error(