fix bug
Showing
2 changed files
with
5 additions
and
3 deletions
... | @@ -457,7 +457,7 @@ class Command(BaseCommand, LoggerMixin): | ... | @@ -457,7 +457,7 @@ class Command(BaseCommand, LoggerMixin): |
457 | 457 | ||
458 | wb = BSWorkbook(set(), set(), set(), set(), set()) | 458 | wb = BSWorkbook(set(), set(), set(), set(), set()) |
459 | ws = wb.create_sheet(self.sheet_name) | 459 | ws = wb.create_sheet(self.sheet_name) |
460 | row_idx = 1 | 460 | row_idx = 0 |
461 | code_idx = 1 | 461 | code_idx = 1 |
462 | mode_code = None | 462 | mode_code = None |
463 | for write_field, field_value in results.items(): | 463 | for write_field, field_value in results.items(): | ... | ... |
... | @@ -34,6 +34,7 @@ class PDFHandler: | ... | @@ -34,6 +34,7 @@ class PDFHandler: |
34 | self.img_suffixs = {'.jpeg', '.jpg', '.png', '.webp', '.bmp'} | 34 | self.img_suffixs = {'.jpeg', '.jpg', '.png', '.webp', '.bmp'} |
35 | self.suffix = self.get_suffix(document_name) | 35 | self.suffix = self.get_suffix(document_name) |
36 | self.is_ebank = False | 36 | self.is_ebank = False |
37 | self.is_e_pdf = False | ||
37 | self.page_text_list = [] | 38 | self.page_text_list = [] |
38 | 39 | ||
39 | def get_suffix(self, file_name): | 40 | def get_suffix(self, file_name): |
... | @@ -293,7 +294,8 @@ class PDFHandler: | ... | @@ -293,7 +294,8 @@ class PDFHandler: |
293 | # 'text': text_list | 294 | # 'text': text_list |
294 | # } | 295 | # } |
295 | # ) | 296 | # ) |
296 | self.is_ebank = True | 297 | # self.is_ebank = True |
298 | self.is_e_pdf = True | ||
297 | # self.page_text_list = page_text_list | 299 | # self.page_text_list = page_text_list |
298 | 300 | ||
299 | def extract_image(self, max_img_count=None): | 301 | def extract_image(self, max_img_count=None): |
... | @@ -316,7 +318,7 @@ class PDFHandler: | ... | @@ -316,7 +318,7 @@ class PDFHandler: |
316 | # (xref, smask, width, height, bpc, colorspace, alt.colorspace, name, filter, invoker) | 318 | # (xref, smask, width, height, bpc, colorspace, alt.colorspace, name, filter, invoker) |
317 | 319 | ||
318 | # 1.页面图片对象数目为0时,保存整个页面为png图片 | 320 | # 1.页面图片对象数目为0时,保存整个页面为png图片 |
319 | if self.is_ebank or len(il) == 0: | 321 | if self.is_e_pdf or self.is_ebank or len(il) == 0: |
320 | page = pdf.loadPage(pno) | 322 | page = pdf.loadPage(pno) |
321 | self.page_to_png(page) | 323 | self.page_to_png(page) |
322 | # 2.页面图片对象数目为1时: | 324 | # 2.页面图片对象数目为1时: | ... | ... |
-
Please register or sign in to post a comment