f83bf474 by 周伟奇

fix bug

1 parent 92b21d6a
...@@ -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时:
......
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!