c8a54f58 by 周伟奇

Merge branch 'feature/license' into feature/mssql

2 parents 540f8777 56281e38
...@@ -35,3 +35,4 @@ src/*.sh ...@@ -35,3 +35,4 @@ src/*.sh
35 35
36 test* 36 test*
37 ocr_test.py 37 ocr_test.py
38 ocr_process.py
...\ No newline at end of file ...\ No newline at end of file
......
...@@ -440,6 +440,19 @@ class Command(BaseCommand, LoggerMixin): ...@@ -440,6 +440,19 @@ class Command(BaseCommand, LoggerMixin):
440 merged_bs_summary = {} 440 merged_bs_summary = {}
441 card_num = 1 441 card_num = 1
442 for role_dict in unknown_summary.values(): 442 for role_dict in unknown_summary.values():
443 if len(role_dict) == 2 and consts.UNKNOWN_ROLE in role_dict:
444 summary_dict = role_dict.pop(consts.UNKNOWN_ROLE, {})
445 for summary in role_dict.values():
446 summary_dict['confidence'].extend(summary['confidence'])
447 summary_dict['role'] = summary['role']
448 summary_dict['code'].extend(summary['code'])
449 summary_dict['print_time'].extend(summary['print_time'])
450 summary_dict['start_date'].extend(summary['start_date'])
451 summary_dict['end_date'].extend(summary['end_date'])
452 summary_dict['sheet'].extend(summary['sheet'])
453 card = '{0}_{1}'.format(consts.UNKNOWN_CARD, card_num)
454 merged_bs_summary[card] = summary_dict
455 else:
443 for summary in role_dict.values(): 456 for summary in role_dict.values():
444 card = '{0}_{1}'.format(consts.UNKNOWN_CARD, card_num) 457 card = '{0}_{1}'.format(consts.UNKNOWN_CARD, card_num)
445 card_num += 1 458 card_num += 1
...@@ -460,6 +473,7 @@ class Command(BaseCommand, LoggerMixin): ...@@ -460,6 +473,7 @@ class Command(BaseCommand, LoggerMixin):
460 for role, summary in classify_summary.items(): 473 for role, summary in classify_summary.items():
461 if one_card or role in card_summary['role_set']: 474 if one_card or role in card_summary['role_set']:
462 merge_role.append(role) 475 merge_role.append(role)
476 card_summary['confidence'].extend(summary['confidence'])
463 card_summary['sheet'].extend(summary['sheet']) 477 card_summary['sheet'].extend(summary['sheet'])
464 card_summary['code'].extend(summary['code']) 478 card_summary['code'].extend(summary['code'])
465 card_summary['print_time'].extend(summary['print_time']) 479 card_summary['print_time'].extend(summary['print_time'])
......
...@@ -18,7 +18,7 @@ class DocHandler: ...@@ -18,7 +18,7 @@ class DocHandler:
18 18
19 def get_doc_list(self, doc_queryset, business_type): 19 def get_doc_list(self, doc_queryset, business_type):
20 for doc_dict in doc_queryset: 20 for doc_dict in doc_queryset:
21 if doc_dict['status'] != DocStatus.COMPLETE.value: 21 if doc_dict['status'] not in [DocStatus.COMPLETE.value, DocStatus.UPLOAD_FAILED.value]:
22 continue 22 continue
23 doc_id = doc_dict.get('id') 23 doc_id = doc_dict.get('id')
24 doc_dict['pdf_link'] = self.get_link(doc_id, business_type) 24 doc_dict['pdf_link'] = self.get_link(doc_id, business_type)
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!