mirror of
https://github.com/luzhisheng/js_reverse.git
synced 2025-04-22 07:08:48 +08:00
117 lines
4.4 KiB
Python
117 lines
4.4 KiB
Python
import image_compare
|
|
import pandas as pd
|
|
import pdfplumber
|
|
from image_text_ocr import ImageTextOcr
|
|
import os
|
|
import cv2
|
|
|
|
|
|
class Discern(object):
|
|
|
|
def __init__(self):
|
|
self.image_text_ocr = ImageTextOcr()
|
|
self.xlsx_keys = {}
|
|
self.num = 0
|
|
|
|
def export_excel(self, export):
|
|
# 将字典列表转换为DataFrame
|
|
pf = pd.DataFrame(list([export]))
|
|
file_path = pd.ExcelWriter('../docs/结果.xlsx')
|
|
# 替换空单元格
|
|
pf.fillna(' ', inplace=True)
|
|
# 输出
|
|
pf.to_excel(file_path, index=False)
|
|
# 保存表格
|
|
file_path.close()
|
|
|
|
def pdf_text(self, pdf_path):
|
|
with pdfplumber.open(pdf_path) as pdf:
|
|
# 遍历每个页面
|
|
page = pdf.pages[0]
|
|
# 提取页面文本
|
|
text = page.extract_text()
|
|
lines = text.split("\n")
|
|
line_str = ''
|
|
for line in lines:
|
|
line_str += line
|
|
if '报告编号:' in line:
|
|
self.xlsx_keys['报告编号'] = line.split("报告编号")[1].strip().replace(': ', '')
|
|
if 'Report Number:' in line:
|
|
self.xlsx_keys['报告编号'] = line.split("Report Number")[1].strip().replace(': ', '')
|
|
if '样品名称:' in line:
|
|
self.xlsx_keys['样品名称'] = line.split("样品名称")[1].strip().replace(': ', '')
|
|
if 'Article Name:' in line:
|
|
self.xlsx_keys['样品名称'] = line.split("Article Name")[1].strip().replace(': ', '')
|
|
if '公司' in line and '中检华通' not in line and '制造商' not in line:
|
|
self.xlsx_keys['公司名称'] = line.strip()
|
|
if '最终报告' in line:
|
|
self.xlsx_keys['检测项目'] = line_str.replace('最终报告', '')
|
|
self.xlsx_keys['标志'] = ''
|
|
|
|
def pdf_images(self, pdf_path):
|
|
self.num = 0
|
|
with pdfplumber.open(pdf_path) as pdf:
|
|
for page in pdf.pages:
|
|
images = page.images
|
|
for img in images:
|
|
# 获取图片的二进制流
|
|
self.num += 1
|
|
image_file = f"../target_img/image_{self.num}.jpg"
|
|
with open(image_file, "wb") as f:
|
|
f.write(img['stream'].get_data())
|
|
|
|
def get_images_text(self):
|
|
for i in range(1, self.num + 1):
|
|
try:
|
|
cma_flag = image_compare.run(f'../target_img/image_{i}.jpg', '../img/cma.jpg')
|
|
cnas_flag = image_compare.run(f'../target_img/image_{i}.jpg', '../img/cnas.jpg')
|
|
text_list = self.image_text_ocr.run(f'../target_img/image_{i}.jpg')
|
|
except cv2.error as c:
|
|
pass
|
|
if cma_flag:
|
|
self.xlsx_keys['标志'] = '国cma'
|
|
if cnas_flag:
|
|
self.xlsx_keys['标志'] += ',cnas中文'
|
|
if text_list:
|
|
self.xlsx_keys['方案编号'] = text_list[0]
|
|
self.xlsx_keys['签发日期'] = text_list[1]
|
|
|
|
def remove_file(self, folder_path):
|
|
with os.scandir(folder_path) as entries:
|
|
for entry in entries:
|
|
if entry.is_file():
|
|
file_path = entry.path
|
|
try:
|
|
os.remove(file_path)
|
|
except Exception as e:
|
|
pass
|
|
|
|
def run(self, folder_path):
|
|
with os.scandir(folder_path) as entries:
|
|
for entry in entries:
|
|
self.xlsx_keys = {
|
|
'登记日期': '',
|
|
'方案编号': '',
|
|
'样品编号': '',
|
|
'报告编号': '',
|
|
'样品名称': '',
|
|
'检测项目': '',
|
|
'标志': '',
|
|
'签发日期': '',
|
|
'公司名称': ''
|
|
}
|
|
self.remove_file('../target_img')
|
|
if entry.is_file():
|
|
file_path = entry.path
|
|
file_name = entry.name
|
|
self.xlsx_keys['文件名'] = file_name
|
|
self.pdf_text(file_path)
|
|
self.pdf_images(file_path)
|
|
self.get_images_text()
|
|
self.export_excel(self.xlsx_keys)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
discern = Discern()
|
|
discern.run('../file')
|