b17b3c65 by 周伟奇

add img count

1 parent 615edaa5
...@@ -99,6 +99,7 @@ RES_FAILED = '识别失败' ...@@ -99,6 +99,7 @@ RES_FAILED = '识别失败'
99 RES_FAILED_1 = '识别失败(阶段1)' 99 RES_FAILED_1 = '识别失败(阶段1)'
100 RES_FAILED_2 = '识别失败(阶段2)' 100 RES_FAILED_2 = '识别失败(阶段2)'
101 RES_FAILED_3 = '识别失败(阶段1数据格式错误)' 101 RES_FAILED_3 = '识别失败(阶段1数据格式错误)'
102 RES_FAILED_SET = {RES_FAILED, RES_FAILED_1, RES_FAILED_2, RES_FAILED_3}
102 103
103 CARD_RATIO = 0.9 104 CARD_RATIO = 0.9
104 UNKNOWN_CARD = '未知卡号' 105 UNKNOWN_CARD = '未知卡号'
......
...@@ -310,13 +310,15 @@ class BSWorkbook(Workbook): ...@@ -310,13 +310,15 @@ class BSWorkbook(Workbook):
310 month_mapping.setdefault(item, []).append( 310 month_mapping.setdefault(item, []).append(
311 (ws.title, idx_list[i] + min_row, idx_list[i + 1] + min_row - 1, day_mean)) 311 (ws.title, idx_list[i] + min_row, idx_list[i + 1] + min_row - 1, day_mean))
312 312
313 def build_metadata_rows(self, confidence, code, print_time, start_date, end_date): 313 def build_metadata_rows(self, confidence, code, print_time, start_date, end_date, res_count_tuple):
314 if start_date is None or end_date is None: 314 if start_date is None or end_date is None:
315 timedelta = None 315 timedelta = None
316 else: 316 else:
317 timedelta = (end_date - start_date).days 317 timedelta = (end_date - start_date).days
318 metadata_rows = [ 318 metadata_rows = [
319 ('流水识别置信度', confidence), 319 ('流水识别置信度', confidence),
320 ('图片总数', res_count_tuple[0]),
321 ('识别成功', res_count_tuple[1]),
320 self.blank_row, 322 self.blank_row,
321 self.code_header, 323 self.code_header,
322 ] 324 ]
...@@ -330,8 +332,8 @@ class BSWorkbook(Workbook): ...@@ -330,8 +332,8 @@ class BSWorkbook(Workbook):
330 ) 332 )
331 return metadata_rows 333 return metadata_rows
332 334
333 def build_meta_sheet(self, role_name, card, confidence, code, print_time, start_date, end_date): 335 def build_meta_sheet(self, role_name, card, confidence, code, print_time, start_date, end_date, res_count_tuple):
334 metadata_rows = self.build_metadata_rows(confidence, code, print_time, start_date, end_date) 336 metadata_rows = self.build_metadata_rows(confidence, code, print_time, start_date, end_date, res_count_tuple)
335 if not isinstance(role_name, str): 337 if not isinstance(role_name, str):
336 role_name = consts.UNKNOWN_ROLE 338 role_name = consts.UNKNOWN_ROLE
337 ms = self.create_sheet('{0}{1}({2})'.format(self.meta_sheet_title, role_name, card)) 339 ms = self.create_sheet('{0}{1}({2})'.format(self.meta_sheet_title, role_name, card))
...@@ -595,7 +597,7 @@ class BSWorkbook(Workbook): ...@@ -595,7 +597,7 @@ class BSWorkbook(Workbook):
595 ms.append(row) 597 ms.append(row)
596 self.remove(tmp2_ws) 598 self.remove(tmp2_ws)
597 599
598 def bs_rebuild(self, bs_summary): 600 def bs_rebuild(self, bs_summary, res_count_tuple):
599 # bs_summary = { 601 # bs_summary = {
600 # '卡号': { 602 # '卡号': {
601 # 'classify': 0, 603 # 'classify': 0,
...@@ -656,7 +658,8 @@ class BSWorkbook(Workbook): ...@@ -656,7 +658,8 @@ class BSWorkbook(Workbook):
656 summary.get('code'), 658 summary.get('code'),
657 summary.get('print_time'), 659 summary.get('print_time'),
658 start_date, 660 start_date,
659 end_date) 661 end_date,
662 res_count_tuple)
660 663
661 # 3.创建月份表、提取/高亮关键行 664 # 3.创建月份表、提取/高亮关键行
662 # 倒序处理 665 # 倒序处理
...@@ -737,21 +740,35 @@ class BSWorkbook(Workbook): ...@@ -737,21 +740,35 @@ class BSWorkbook(Workbook):
737 res_list.sort(key=lambda x: (x[0], x[1], x[2])) 740 res_list.sort(key=lambda x: (x[0], x[1], x[2]))
738 ws = self.create_sheet(consts.RES_SHEET_NAME) 741 ws = self.create_sheet(consts.RES_SHEET_NAME)
739 ws.append(consts.RES_SHEET_HEADER) 742 ws.append(consts.RES_SHEET_HEADER)
743 success_count = 0
740 for res_tuple in res_list: 744 for res_tuple in res_list:
745 if res_tuple[-1] not in consts.RES_FAILED_SET:
746 success_count += 1
741 ws.append(res_tuple) 747 ws.append(res_tuple)
748 return len(res_list), success_count
749 else:
750 return 0, 0
751
752 def move_res_sheet(self):
753 sheet = self.get_sheet_by_name(consts.RES_SHEET_NAME)
754 idx = self._sheets.index(sheet)
755 del self._sheets[idx]
756 self._sheets.append(sheet)
742 757
743 def remove_base_sheet(self): 758 def remove_base_sheet(self):
744 if len(self.sheetnames) > 1: 759 if len(self.sheetnames) > 1:
745 self.remove(self.get_sheet_by_name('Sheet')) 760 self.remove(self.get_sheet_by_name('Sheet'))
746 761
747 def rebuild(self, bs_summary, license_summary, res_list, document_scheme): 762 def rebuild(self, bs_summary, license_summary, res_list, document_scheme):
763 res_count_tuple = self.res_sheet(res_list)
764
748 count_list = [(consts.MODEL_FIELD_BS, len(bs_summary))] 765 count_list = [(consts.MODEL_FIELD_BS, len(bs_summary))]
749 if document_scheme == consts.DOC_SCHEME_LIST[1]: 766 if document_scheme == consts.DOC_SCHEME_LIST[1]:
750 self.license_rebuild(license_summary, document_scheme, count_list) 767 self.license_rebuild(license_summary, document_scheme, count_list)
751 self.bs_rebuild(bs_summary) 768 self.bs_rebuild(bs_summary, res_count_tuple)
752 else: 769 else:
753 self.bs_rebuild(bs_summary) 770 self.bs_rebuild(bs_summary, res_count_tuple)
754 self.license_rebuild(license_summary, document_scheme, count_list) 771 self.license_rebuild(license_summary, document_scheme, count_list)
755 self.res_sheet(res_list) 772 self.move_res_sheet()
756 self.remove_base_sheet() 773 self.remove_base_sheet()
757 return count_list 774 return count_list
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!