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