fix bs compare

周伟奇
Showing 2 changed files with 40 additions and 18 deletions
src/apps/doc/consts.py
src/celery_compare/tasks.py
--- a/src/apps/doc/consts.py
View file @6e11b40
+++ b/src/apps/doc/consts.py
View file @6e11b40
@@ -1671,7 +1671,7 @@ BD_COMPARE_LOGIC = {
 BS_COMPARE_LOGIC = {
    '户名': ('role', 'se_one_compare', {}, '主共借人未提供银行流水'),
-    '(担保人)户名': ('role', 'se_one_compare', {}, '担保人未提供银行流水，需人工查看直系亲属关系'),
+    '(担保人)户名': ('role', 'se_one_compare', {}, '担保人未提供银行流水'),
    '打印日期': ('print_time', 'se_bs_print_date_compare', {}, '主共借人银行流水打印日期异常'),
    '(担保人)打印日期': ('print_time', 'se_bs_print_date_compare', {}, '担保人银行流水打印日期异常'),
    '流水日期': ('date', 'se_bs_date_compare', {}, '主共借人银行流水日期异常'),
@@ -1679,6 +1679,7 @@ BS_COMPARE_LOGIC = {
 }
 SPECIAL_REASON = '主共借人未提供银行流水，含担保人需人工查看直系亲属关系'
+SPECIAL_REASON_2 = '担保人未提供银行流水，需人工查看直系亲属关系'
 HT_COMPARE_LOGIC = {
    '合同编号-每页': ('合同编号-每页', 'se_list_compare', {}, '合同编号与系统不一致'),
--- a/src/celery_compare/tasks.py
View file @6e11b40
+++ b/src/celery_compare/tasks.py
View file @6e11b40
@@ -1484,25 +1484,9 @@ def se_bs_compare(license_en, ocr_res_dict, strip_list):
                max_correct_count = correct_count
                pre_best_res = pre_tmp_res_part
-        for name, value in pre_field_list:
-            if len(pre_best_res) > 0:
-                result, ocr_str, reason = pre_best_res[name]
-            else:
-                result = consts.RESULT_N
-                ocr_str = empty_str
-                if len(suf_field_list) > 0:
-                    reason = consts.SPECIAL_REASON
-                else:
-                    reason = compare_logic[pre_field_list[0][0]][3]
-            img_path = empty_str
-            error_type = empty_error_type if result == consts.RESULT_Y else ErrorType.OCR.value
-            if isinstance(value, list):
-                value = json.dumps(value, ensure_ascii=False)
-            result_field_list.append((name, value, result, ocr_str, img_path, error_type, reason))
        # 担保人
-        if len(suf_field_list) > 0:
        suf_best_res = {}
+        if len(suf_field_list) > 0:
            max_correct_count = 0
            for ocr_res in ocr_res_list:
                correct_count = 0
@@ -1532,18 +1516,39 @@ def se_bs_compare(license_en, ocr_res_dict, strip_list):
                    max_correct_count = correct_count
                    suf_best_res = suf_tmp_res_part
+        for name, value in pre_field_list:
+            if len(pre_best_res) > 0:
+                result, ocr_str, reason = pre_best_res[name]
+            else:
+                result = consts.RESULT_N
+                ocr_str = empty_str
+                if len(suf_field_list) > 0 and len(suf_best_res) > 0:
+                    reason = consts.SPECIAL_REASON
+                else:
+                    reason = compare_logic[pre_field_list[0][0]][3]
+            img_path = empty_str
+            error_type = empty_error_type if result == consts.RESULT_Y else ErrorType.OCR.value
+            if isinstance(value, list):
+                value = json.dumps(value, ensure_ascii=False)
+            result_field_list.append((name, value, result, ocr_str, img_path, error_type, reason))
+        if len(suf_field_list) > 0:
            for name, value in suf_field_list:
                if len(suf_best_res) > 0:
                    result, ocr_str, reason = suf_best_res[name]
                else:
                    result = consts.RESULT_N
                    ocr_str = empty_str
+                    if len(pre_best_res) > 0:
+                        reason = consts.SPECIAL_REASON_2
+                    else:
                        reason = compare_logic[suf_field_list[0][0]][3]
                img_path = empty_str
                error_type = empty_error_type if result == consts.RESULT_Y else ErrorType.OCR.value
                if isinstance(value, list):
                    value = json.dumps(value, ensure_ascii=False)
                result_field_list.append((name, value, result, ocr_str, img_path, error_type, reason))
    else:
        for name, value in strip_list:
            if isinstance(value, list):
@@ -1921,6 +1926,14 @@ def se_compare_process(compare_info, ocr_res_dict):
                        for a, b in field_list:
                            if isinstance(b, str):
                                strip_list.append((a, b.strip()))
+                            elif isinstance(b, list):
+                                c = []
+                                for i in b:
+                                    if isinstance(i, str):
+                                        c.append(i.strip())
+                                    else:
+                                        c.append(i)
+                                strip_list.append((a, c))
                            else:
                                strip_list.append((a, b))
                        failure_field = []
@@ -1955,6 +1968,14 @@ def se_compare_process(compare_info, ocr_res_dict):
                for a, b in field_list:
                    if isinstance(b, str):
                        strip_list.append((a, b.strip()))
+                    elif isinstance(b, list):
+                        c = []
+                        for i in b:
+                            if isinstance(i, str):
+                                c.append(i.strip())
+                            else:
+                                c.append(i)
+                        strip_list.append((a, c))
                    else:
                        strip_list.append((a, b))
                failure_field = []