00bb1843 by 周伟奇

change sheet name

1 parent 23c4a341
...@@ -17,7 +17,7 @@ class BSWorkbook(Workbook): ...@@ -17,7 +17,7 @@ class BSWorkbook(Workbook):
17 def __init__(self, interest_keyword, salary_keyword, loan_keyword, wechat_keyword, *args, **kwargs): 17 def __init__(self, interest_keyword, salary_keyword, loan_keyword, wechat_keyword, *args, **kwargs):
18 super().__init__(*args, **kwargs) 18 super().__init__(*args, **kwargs)
19 locale.setlocale(locale.LC_NUMERIC, 'en_US.UTF-8') 19 locale.setlocale(locale.LC_NUMERIC, 'en_US.UTF-8')
20 self.meta_sheet_title = '关键信息提取和展示' 20 self.meta_sheet_title = 'Key info'
21 self.blank_row = (None,) 21 self.blank_row = (None,)
22 self.code_header = ('页数', '电子回单验证码') 22 self.code_header = ('页数', '电子回单验证码')
23 self.date_header = ('打印时间', '起始日期', '终止日期', '流水区间结果') 23 self.date_header = ('打印时间', '起始日期', '终止日期', '流水区间结果')
...@@ -45,7 +45,7 @@ class BSWorkbook(Workbook): ...@@ -45,7 +45,7 @@ class BSWorkbook(Workbook):
45 if not isinstance(card, str): 45 if not isinstance(card, str):
46 return consts.ERROR_CARD 46 return consts.ERROR_CARD
47 try: 47 try:
48 new_card = card.translate(consts.SHEET_TITLE_TRANS).strip()[-6:] 48 new_card = card.translate(consts.SHEET_TITLE_TRANS).strip()[-4:]
49 if len(new_card) == 0: 49 if len(new_card) == 0:
50 new_card = consts.ERROR_CARD 50 new_card = consts.ERROR_CARD
51 except Exception as e: 51 except Exception as e:
...@@ -327,9 +327,11 @@ class BSWorkbook(Workbook): ...@@ -327,9 +327,11 @@ class BSWorkbook(Workbook):
327 ) 327 )
328 return metadata_rows 328 return metadata_rows
329 329
330 def build_meta_sheet(self, card, confidence, code, print_time, start_date, end_date): 330 def build_meta_sheet(self, role_name, card, confidence, code, print_time, start_date, end_date):
331 metadata_rows = self.build_metadata_rows(confidence, code, print_time, start_date, end_date) 331 metadata_rows = self.build_metadata_rows(confidence, code, print_time, start_date, end_date)
332 ms = self.create_sheet('{0}({1})'.format(self.meta_sheet_title, card)) 332 if not isinstance(role_name, str):
333 role_name = consts.UNKNOWN_ROLE
334 ms = self.create_sheet('{0}{1}({2})'.format(self.meta_sheet_title, role_name, card))
333 for row in metadata_rows: 335 for row in metadata_rows:
334 ms.append(row) 336 ms.append(row)
335 return ms 337 return ms
...@@ -614,7 +616,8 @@ class BSWorkbook(Workbook): ...@@ -614,7 +616,8 @@ class BSWorkbook(Workbook):
614 616
615 # 2.元信息提取表 617 # 2.元信息提取表
616 confidence = self.get_confidence(max_find_count) 618 confidence = self.get_confidence(max_find_count)
617 ms = self.build_meta_sheet(new_card, 619 ms = self.build_meta_sheet(summary.get('role', consts.UNKNOWN_ROLE),
620 new_card,
618 confidence, 621 confidence,
619 summary.get('code'), 622 summary.get('code'),
620 summary.get('print_time'), 623 summary.get('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!