27db93d6 by 周伟奇

fix bug

1 parent 266f2739
...@@ -667,9 +667,9 @@ RP_FIELD_ORDER_1 = IC_FIELD_ORDER_1 ...@@ -667,9 +667,9 @@ RP_FIELD_ORDER_1 = IC_FIELD_ORDER_1
667 VAT_CN_NAME = 'VAT普票' 667 VAT_CN_NAME = 'VAT普票'
668 VAT_CLASSIFY = 0 668 VAT_CLASSIFY = 0
669 VAT_FIELD_ORDER = (('发票代码', '发票代码'), 669 VAT_FIELD_ORDER = (('发票代码', '发票代码'),
670 ('发票代码(开具)', '发票代码(开具)'), 670 ('发票代码_开具', '发票代码(开具)'),
671 ('发票号码', '发票号码'), 671 ('发票号码', '发票号码'),
672 ('发票号码(开具)', '发票号码(开具)'), 672 ('发票号码_开具', '发票号码(开具)'),
673 ('开票日期', '开票日期'), 673 ('开票日期', '开票日期'),
674 ('校验码', '校验码'), 674 ('校验码', '校验码'),
675 ('货物或应税劳务、服务名称', '货物或应税劳务、服务名称'), 675 ('货物或应税劳务、服务名称', '货物或应税劳务、服务名称'),
...@@ -686,7 +686,7 @@ VAT_FIELD_ORDER = (('发票代码', '发票代码'), ...@@ -686,7 +686,7 @@ VAT_FIELD_ORDER = (('发票代码', '发票代码'),
686 ('销方纳税人识别号', '销售方纳税人识别号'), 686 ('销方纳税人识别号', '销售方纳税人识别号'),
687 ('销方地址、电话', '销售方地址、电话'), 687 ('销方地址、电话', '销售方地址、电话'),
688 ('销方开户行及账号', '销售方开户行及账号'), 688 ('销方开户行及账号', '销售方开户行及账号'),
689 ('销售方:(章)', '销售方:(章)'), 689 ('下盖章', '销售方:(章)'),
690 ('备注', '备注'),) 690 ('备注', '备注'),)
691 # 机动车登记证书 691 # 机动车登记证书
692 MVC_CN_NAME = '机动车登记证书' 692 MVC_CN_NAME = '机动车登记证书'
......
...@@ -440,10 +440,23 @@ class Command(BaseCommand, LoggerMixin): ...@@ -440,10 +440,23 @@ 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 for summary in role_dict.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'])
444 card = '{0}_{1}'.format(consts.UNKNOWN_CARD, card_num) 453 card = '{0}_{1}'.format(consts.UNKNOWN_CARD, card_num)
445 card_num += 1 454 merged_bs_summary[card] = summary_dict
446 merged_bs_summary[card] = summary 455 else:
456 for summary in role_dict.values():
457 card = '{0}_{1}'.format(consts.UNKNOWN_CARD, card_num)
458 card_num += 1
459 merged_bs_summary[card] = summary
447 else: 460 else:
448 # 1卡号 461 # 1卡号
449 one_card = False 462 one_card = False
...@@ -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'])
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!