b0be45bc by 周伟奇

fix dda

1 parent 39fadc7a
...@@ -79,16 +79,30 @@ class Command(BaseCommand, LoggerMixin): ...@@ -79,16 +79,30 @@ class Command(BaseCommand, LoggerMixin):
79 79
80 def license1_process(self, ocr_data, all_res, classify): 80 def license1_process(self, ocr_data, all_res, classify):
81 # 类别:'0'身份证, '1'居住证 81 # 类别:'0'身份证, '1'居住证
82 license_data = ocr_data.get('data', []) 82 license_data = ocr_data.get('data')
83 if not license_data: 83 if not license_data:
84 return 84 return
85 if isinstance(license_data, dict):
86 license_data.pop('base64_img', '')
85 if classify == consts.IC_CLASSIFY: 87 if classify == consts.IC_CLASSIFY:
86 for id_card_dict in license_data: 88 id_card_dict = {}
87 try: 89 card_type = license_data.get('type', '')
88 id_card_dict.pop('base64_img') 90 is_ic = card_type.startswith('身份证')
89 except Exception as e: 91 is_info_side = card_type.endswith('信息面')
90 continue 92 id_card_dict['类别'] = '0' if is_ic else '1'
91 all_res.setdefault(classify, []).extend(license_data) 93 if is_ic:
94 field_map = consts.IC_MAP_0 if is_info_side else consts.IC_MAP_1
95 else:
96 field_map = consts.RP_MAP_0 if is_info_side else consts.RP_MAP_1
97 for write_field, search_field in field_map:
98 id_card_dict[write_field] = license_data.get('words_result', {}).get(search_field, {}).get('words', '')
99 if not is_info_side:
100 start_time = license_data.get('words_result', {}).get('签发日期', {}).get('words', '')
101 end_time = license_data.get('words_result', {}).get('失效日期', {}).get('words', '')
102 id_card_dict['有效期限'] = '{0}-{1}'.format(start_time, end_time)
103 all_res.setdefault(classify, []).append(id_card_dict)
104 else:
105 all_res.setdefault(classify, []).extend(license_data)
92 106
93 def license2_process(self, ocr_data, all_res, classify, img_path): 107 def license2_process(self, ocr_data, all_res, classify, img_path):
94 pid, _, _, _, _, _ = consts.LICENSE_CLASSIFY_MAPPING.get(classify) 108 pid, _, _, _, _, _ = consts.LICENSE_CLASSIFY_MAPPING.get(classify)
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!