change sheet name
Showing
1 changed file
with
8 additions
and
5 deletions
... | @@ -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'), | ... | ... |
-
Please register or sign in to post a comment