12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237 |
- import copy
- import io
- import os
- import re
- import sys
- from bs4 import BeautifulSoup
- sys.path.append(os.path.dirname(__file__) + "/../")
- from pdfplumber import PDF
- from pdfplumber.table import TableFinder
- from pdfplumber.page import Page as pdfPage
- from format_convert.convert_tree import _Document, _Page, _Image, _Sentence, _Table
- import time
- import math
- from scipy.stats import linregress
- from matplotlib import pyplot as plt
- from shapely.geometry import LineString, Point
- from PIL import Image
- import traceback
- import cv2
- import PyPDF2
- from PyPDF2 import PdfFileReader, PdfFileWriter
- from pdfminer.pdfparser import PDFParser
- from pdfminer.pdfdocument import PDFDocument
- from pdfminer.pdfpage import PDFPage
- from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
- from pdfminer.converter import PDFPageAggregator
- from pdfminer.layout import LTTextBoxHorizontal, LAParams, LTFigure, LTImage, LTCurve, LTText, LTChar, LTRect, \
- LTTextBoxVertical, LTLine, LTTextContainer, LTTextLine
- from format_convert.utils import judge_error_code, get_platform, LineTable, log, \
- memory_decorator, get_garble_code, get_md5_from_bytes, bytes2np, bbox_iou
- import fitz
- from format_convert.wrapt_timeout_decorator import timeout
- from otr.table_line_new import table_line_pdf
- @memory_decorator
- def pdf2Image(path, save_dir):
- log("into pdf2Image")
- try:
- try:
- doc = fitz.open(path)
- except Exception as e:
- log("pdf format error!")
- # print("pdf format error!", e)
- return [-3]
- # output_image_list = []
- output_image_dict = {}
- page_count = doc.page_count
- for page_no in range(page_count):
- # 限制pdf页数,只取前10页后10页
- if page_count > 20:
- if 10 <= page_no < page_count - 10:
- # log("pdf2Image: pdf pages count " + str(doc.page_count)
- # + ", only get 70 pages")
- continue
- try:
- page = doc.loadPage(page_no)
- output = save_dir + "_page" + str(page_no) + ".png"
- rotate = int(0)
- # 每个尺寸的缩放系数为1.3,这将为我们生成分辨率提高2.6的图像。
- # 此处若是不做设置,默认图片大小为:792X612, dpi=96
- # (1.33333333 --> 1056x816) (2 --> 1584x1224)
- # (1.183, 2.28 --> 1920x1080)
- zoom_x = 3.
- zoom_y = 3.
- # mat = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
- mat = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
- pix = page.getPixmap(matrix=mat, alpha=False)
- pix.writePNG(output)
- pdf_image = cv2.imread(output)
- print("pdf_image", page_no, pdf_image.shape)
- # output_image_list.append([page_no, output])
- output_image_dict[int(page_no)] = output
- except ValueError as e:
- traceback.print_exc()
- if str(e) == "page not in document":
- log("pdf2Image page not in document! continue..." + str(page_no))
- continue
- elif "encrypted" in str(e):
- log("pdf2Image document need password " + str(page_no))
- return [-7]
- except RuntimeError as e:
- if "cannot find page" in str(e):
- log("pdf2Image page {} not in document! continue... ".format(str(page_no)) + str(e))
- continue
- else:
- traceback.print_exc()
- return [-3]
- return [output_image_dict]
- except Exception as e:
- log("pdf2Image error!")
- print("pdf2Image", traceback.print_exc())
- return [-1]
- @timeout(10, timeout_exception=TimeoutError)
- def pdf_analyze(interpreter, page, device, page_no):
- log("into pdf_analyze")
- pdf_time = time.time()
- # print("pdf_analyze interpreter process...")
- interpreter.process_page(page)
- # print("pdf_analyze device get_result...")
- layout = device.get_result()
- log("pdf2text page " + str(page_no) + " read time " + str(time.time() - pdf_time))
- return layout
- @memory_decorator
- def pdf2text(path, unique_type_dir):
- return
- def get_single_pdf(path, page_no):
- log("into get_single_pdf")
- try:
- # print("path, ", path)
- pdf_origin = PdfFileReader(path, strict=False)
- pdf_new = PdfFileWriter()
- pdf_new.addPage(pdf_origin.getPage(page_no))
- path_new = path.split(".")[0] + "_split.pdf"
- with open(path_new, "wb") as ff:
- pdf_new.write(ff)
- return path_new
- except PyPDF2.utils.PdfReadError as e:
- raise e
- except Exception as e:
- log("get_single_pdf error! page " + str(page_no))
- traceback.print_exc()
- raise e
- def page_table_connect(has_table_dict):
- log("into page_table_connect")
- if not has_table_dict:
- return [], []
- try:
- # 判断是否有页码的表格相连
- table_connect_list = []
- temp_list = []
- # 离图片顶部或底部距离,页面高度的1/7
- threshold = 7
- page_no_list = list(has_table_dict.keys())
- page_no_list.sort(key=lambda x: x)
- for i in range(1, len(page_no_list)):
- page_info = has_table_dict.get(page_no_list[i])
- last_page_info = has_table_dict.get(page_no_list[i - 1])
- # 页码需相连
- if page_no_list[i] - page_no_list[i - 1] == 1:
- # 上一页最后一个区域的列数和下一页第一个区域列数都为0,且相等
- if not last_page_info[1][-1] and not page_info[1][0] and \
- last_page_info[1][-1] == page_info[1][0]:
- # 上一页的轮廓点要离底部一定距离内,下一页的轮廓点要离顶部一定距离内
- if last_page_info[4][0] - last_page_info[2][-1][1][1] \
- <= int(last_page_info[4][0] / threshold) \
- and page_info[2][0][0][1] - 0 \
- <= int(page_info[4][0] / threshold):
- temp_list.append(page_no_list[i - 1])
- temp_list.append(page_no_list[i])
- continue
- # 条件不符合的,存储之前保存的连接页码
- if len(temp_list) > 1:
- temp_list = list(set(temp_list))
- temp_list.sort(key=lambda x: x)
- table_connect_list.append(temp_list)
- temp_list = []
- if len(temp_list) > 1:
- temp_list = list(set(temp_list))
- temp_list.sort(key=lambda x: x)
- table_connect_list.append(temp_list)
- temp_list = []
- # 连接两页内容
- connect_text_list = []
- for area in table_connect_list:
- first_page_no = area[0]
- area_page_text = str(has_table_dict.get(first_page_no)[0])
- for i in range(1, len(area)):
- current_page_no = area[i]
- current_page_text = str(has_table_dict.get(current_page_no)[0])
- # 连接两个table
- table_prefix = re.finditer('<table border="1">', current_page_text)
- index_list = []
- for t in table_prefix:
- index_list.append(t.span())
- delete_index = index_list[0]
- current_page_text = current_page_text[:delete_index[0]] \
- + current_page_text[delete_index[1]:]
- table_suffix = re.finditer('</table>', area_page_text)
- index_list = []
- for t in table_suffix:
- index_list.append(t.span())
- delete_index = index_list[-1]
- area_page_text = area_page_text[:delete_index[0]] \
- + area_page_text[delete_index[1]:]
- area_page_text = area_page_text + current_page_text
- connect_text_list.append([area_page_text, area])
- return table_connect_list, connect_text_list
- except Exception as e:
- # print("page_table_connect", e)
- log("page_table_connect error!")
- traceback.print_exc()
- return [-1], [-1]
- @timeout(30, timeout_exception=TimeoutError)
- def read_pdf(path, package_name, packages):
- log(package_name)
- laparams = LAParams(line_overlap=0.01,
- char_margin=0.3,
- line_margin=0.01,
- word_margin=0.01,
- boxes_flow=0.1, )
- if package_name == packages[0]:
- fp = open(path, 'rb')
- parser = PDFParser(fp)
- doc_pdfminer = PDFDocument(parser)
- rsrcmgr = PDFResourceManager()
- device = PDFPageAggregator(rsrcmgr, laparams=laparams)
- interpreter = PDFPageInterpreter(rsrcmgr, device)
- return doc_pdfminer, device, interpreter
- elif package_name == packages[1]:
- doc_pymupdf = fitz.open(path)
- return doc_pymupdf
- elif package_name == packages[2]:
- doc_pypdf2 = PdfFileReader(path, strict=False)
- doc_pypdf2_new = PdfFileWriter()
- return doc_pypdf2, doc_pypdf2_new
- elif package_name == packages[3]:
- fp = open(path, 'rb')
- lt = LineTable()
- doc_top = 0
- doc_pdfplumber = read_pdfplumber(fp, laparams)
- return lt, doc_top, doc_pdfplumber
- @timeout(25, timeout_exception=TimeoutError)
- def read_pdfminer(path, laparams):
- fp = open(path, 'rb')
- parser = PDFParser(fp)
- doc_pdfminer = PDFDocument(parser)
- rsrcmgr = PDFResourceManager()
- device = PDFPageAggregator(rsrcmgr, laparams=laparams)
- interpreter = PDFPageInterpreter(rsrcmgr, device)
- return doc_pdfminer, device, interpreter
- @timeout(15, timeout_exception=TimeoutError)
- def read_pymupdf(path):
- return fitz.open(path)
- @timeout(15, timeout_exception=TimeoutError)
- def read_pypdf2(path):
- doc_pypdf2 = PdfFileReader(path, strict=False)
- doc_pypdf2_new = PdfFileWriter()
- return doc_pypdf2, doc_pypdf2_new
- @timeout(25, timeout_exception=TimeoutError, use_signals=False)
- def read_pdfplumber(path, laparams):
- fp = open(path, 'rb')
- lt = LineTable()
- doc_top = 0
- doc_pdfplumber = PDF(fp, laparams=laparams.__dict__)
- return lt, doc_top, doc_pdfplumber
- class PDFConvert:
- def __init__(self, path, unique_type_dir, need_page_no):
- self._doc = _Document(path)
- self.path = path
- self.unique_type_dir = unique_type_dir
- if not os.path.exists(self.unique_type_dir):
- os.mkdir(self.unique_type_dir)
- # 指定提取的页码范围
- self.need_page_no = need_page_no
- self.start_page_no = None
- self.end_page_no = None
- # 默认使用limit_page_cnt控制,前10页后10页
- if self.need_page_no is None:
- self.limit_page_cnt = 20
- else:
- # 使用start_page_no,end_page_no范围控制,例如2,5
- ss = self.need_page_no.split(',')
- if len(ss) != 2:
- self._doc.error_code = [-14]
- else:
- self.start_page_no = int(ss[0])
- self.end_page_no = int(ss[-1])
- if self.end_page_no == -1:
- self.end_page_no = 1000000
- self.start_page_no -= 1
- self.end_page_no -= 1
- if self.end_page_no <= self.start_page_no or self.start_page_no < 0 or self.end_page_no < -1:
- self._doc.error_code = [-14]
- self.packages = ["pdfminer", "PyMuPDF", "PyPDF2", "pdfplumber"]
- self.has_init_pdf = [0] * len(self.packages)
- # 记录图片对象的md5,用于去除大量重复图片
- self.md5_image_obj_list = []
- @memory_decorator
- def init_package(self, package_name):
- # 各个包初始化
- try:
- laparams = LAParams(line_overlap=0.01,
- char_margin=0.3,
- line_margin=0.01,
- word_margin=0.01,
- boxes_flow=0.1, )
- if package_name == self.packages[0]:
- # fp = open(self.path, 'rb')
- # parser = PDFParser(fp)
- # self.doc_pdfminer = PDFDocument(parser)
- # rsrcmgr = PDFResourceManager()
- # self.laparams = LAParams(line_overlap=0.01,
- # char_margin=0.3,
- # line_margin=0.01,
- # word_margin=0.01,
- # boxes_flow=0.1,)
- # self.device = PDFPageAggregator(rsrcmgr, laparams=self.laparams)
- # self.interpreter = PDFPageInterpreter(rsrcmgr, self.device)
- self.doc_pdfminer, self.device, self.interpreter = read_pdfminer(self.path, laparams)
- self.has_init_pdf[0] = 1
- elif package_name == self.packages[1]:
- self.doc_pymupdf = read_pymupdf(self.path)
- self.has_init_pdf[1] = 1
- elif package_name == self.packages[2]:
- # self.doc_pypdf2 = PdfFileReader(self.path, strict=False)
- # self.doc_pypdf2_new = PdfFileWriter()
- self.doc_pypdf2, self.doc_pypdf2_new = read_pypdf2(self.path)
- self.has_init_pdf[2] = 1
- elif package_name == self.packages[3]:
- # self.fp = open(self.path, 'rb')
- # self.lt = LineTable()
- # self.doc_top = 0
- # self.doc_pdfplumber = PDF(self.fp, laparams=self.laparams.__dict__)
- self.lt, self.doc_top, self.doc_pdfplumber = read_pdfplumber(self.path, laparams)
- self.has_init_pdf[3] = 0
- else:
- log("Only Support Packages " + str(self.packages))
- raise Exception
- except Exception as e:
- log(package_name + " cannot open pdf!")
- traceback.print_exc()
- self._doc.error_code = [-3]
- def convert(self, limit_page_cnt=20):
- if self.has_init_pdf[0] == 0:
- self.init_package("pdfminer")
- if self._doc.error_code is not None:
- self._doc.error_code = None
- # pdfminer读不了直接转成图片识别
- self.get_all_page_image()
- return
- # 判断是否能读pdf
- try:
- pages = PDFPage.create_pages(self.doc_pdfminer)
- for page in pages:
- break
- pages = list(pages)
- # except pdfminer.psparser.PSEOF as e:
- except:
- # pdfminer 读不了空白页的对象,直接使用pymupdf转换出的图片进行ocr识别
- log("pdf2text pdfminer read failed! read by pymupdf!")
- traceback.print_exc()
- try:
- self.get_all_page_image()
- return
- except:
- traceback.print_exc()
- log("pdf2text use pymupdf read failed!")
- self._doc.error_code = [-3]
- return
- # 每一页进行处理
- pages = PDFPage.create_pages(self.doc_pdfminer)
- pages = list(pages)
- page_count = len(pages)
- page_no = 0
- for page in pages:
- # 指定pdf页码
- if self.start_page_no is not None and self.end_page_no is not None:
- if page_count < self.end_page_no:
- self.end_page_no = page_count
- if page_no < self.start_page_no or page_no >= self.end_page_no:
- page_no += 1
- continue
- # 限制pdf页数,只取前后各10页
- else:
- if page_count > limit_page_cnt and int(limit_page_cnt/2) <= page_no < page_count - int(limit_page_cnt/2):
- page_no += 1
- continue
- # 解析单页
- self._page = _Page(page, page_no)
- self.convert_page(page, page_no)
- if self._doc.error_code is None and self._page.error_code is not None:
- if self._page.error_code[0] in [-4, -3, 0]:
- page_no += 1
- continue
- else:
- self._doc.error_code = self._page.error_code
- break
- self._doc.add_child(self._page)
- page_no += 1
- self.delete_same_image()
- self.delete_header_footer()
- # self.delete_bold_text_duplicate()
- def delete_same_image(self, show=0):
- # 剔除大量重复图片
- md5_dict = {}
- for _md5, image_obj in self.md5_image_obj_list:
- if _md5 in md5_dict.keys():
- md5_dict[_md5] += [image_obj]
- else:
- md5_dict[_md5] = [image_obj]
- cnt_threshold = 10
- delete_obj_list = []
- for _md5 in md5_dict.keys():
- img_list = md5_dict.get(_md5)
- print('len(md5_dict.get(_md5))', _md5, len(img_list))
- if len(img_list) >= cnt_threshold:
- if show:
- img_np = bytes2np(img_list[0].content)
- cv2.namedWindow('delete same img_np', cv2.WINDOW_NORMAL)
- cv2.imshow('delete same img_np', img_np)
- cv2.waitKey(0)
- delete_obj_list += img_list
- for page in self._doc.children:
- for obj in delete_obj_list:
- if obj in page.children:
- page.children.remove(obj)
- if show:
- for page in self._doc.children:
- for obj in page.children:
- if isinstance(obj, _Image):
- img_np = bytes2np(obj.content)
- cv2.imshow('page img_np', img_np)
- cv2.waitKey(0)
- def delete_header_footer(self):
- sen_dict = {}
- for page in self._doc.children:
- for obj in page.children:
- if isinstance(obj, _Sentence):
- key = str(obj.content) + ' ' + str(int(obj.y))
- # print('key', key)
- if key in sen_dict.keys():
- sen_dict[key] += [obj]
- else:
- sen_dict[key] = [obj]
- for key in sen_dict.keys():
- l = sen_dict.get(key)
- if len(l) >= 2/3 * max(10, len(self._doc.children)):
- for page in self._doc.children:
- new_children = []
- for obj in page.children:
- if isinstance(obj, _Sentence):
- if obj not in l:
- new_children.append(obj)
- else:
- new_children.append(obj)
- page.children = new_children
- print('len(l)', len(l), len(self._doc.children))
- print('delete_header_footer l[0]', l[0].content, l[0].y)
- return
- def delete_bold_text_duplicate(self, lt_text_box_list):
- # 拿出所有LTChar
- lt_char_list = []
- for lt_text_box in lt_text_box_list:
- for lt_text_line in lt_text_box:
- for lt_char in lt_text_line:
- if isinstance(lt_char, LTChar):
- lt_char_list.append(lt_char)
- # 找出需剔除的
- lt_char_list.sort(key=lambda x: (int(x.bbox[1]), x.bbox[0]))
- delete_list = []
- for i in range(len(lt_char_list)-1):
- lt_char1 = lt_char_list[i]
- bbox1 = lt_char1.bbox
- lt_char2 = lt_char_list[i+1]
- bbox2 = lt_char2.bbox
- if lt_char1 in delete_list:
- continue
- if lt_char2 in delete_list:
- continue
- if lt_char1.get_text() == lt_char2.get_text() and bbox1[0] <= bbox2[0] <= bbox1[2] <= bbox2[2] \
- and int(bbox1[1]) == int(bbox2[1]) and int(bbox1[3]) == int(bbox2[3]) \
- and re.search('[\u4e00-\u9fff():、,。]', lt_char1.get_text()):
- delete_list.append(lt_char2)
- # 重新组装
- new_lt_text_box_list = []
- for lt_text_box in lt_text_box_list:
- new_lt_text_box = LTTextBoxHorizontal()
- for lt_text_line in lt_text_box:
- new_lt_text_line = LTTextLine(0.01)
- for lt_char in lt_text_line:
- if lt_char in delete_list:
- continue
- if isinstance(lt_char, LTChar):
- new_lt_text_line.add(lt_char)
- new_lt_text_box.add(new_lt_text_line)
- new_lt_text_box_list.append(new_lt_text_box)
- return new_lt_text_box_list
- def clean_text(self, _text):
- return re.sub("\s", "", _text)
- def get_text_lines(self, page, page_no):
- lt_line_list = []
- page_plumber = pdfPage(self.doc_pdfplumber, page, page_number=page_no, initial_doctop=self.doc_top)
- self.doc_top += page_plumber.height
- table_finder = TableFinder(page_plumber)
- all_width_zero = True
- for _edge in table_finder.get_edges():
- if _edge.get('linewidth') and _edge.get('linewidth') > 0:
- all_width_zero = False
- break
- for _edge in table_finder.get_edges():
- # print(_edge)
- if _edge.get('linewidth', 0.1) > 0 or all_width_zero:
- lt_line_list.append(LTLine(1, (float(_edge["x0"]), float(_edge["y0"])),
- (float(_edge["x1"]), float(_edge["y1"]))))
- log("pdf page %s has %s lines" % (str(page_no), str(len(lt_line_list))))
- return lt_line_list
- def get_page_lines(self, layout, page_no, show=0):
- def _plot(_line_list, title, mode=1):
- if not show:
- return
- for _line in _line_list:
- if mode == 1:
- x0, y0, x1, y1 = _line.__dict__.get("bbox")
- elif mode == 2:
- x0, y0, x1, y1 = _line
- plt.plot([x0, x1], [y0, y1])
- plt.title(title)
- plt.show()
- return
- def is_cross(A, B, C, D):
- if A[0] == B[0] == C[0] == D[0]:
- if A[1] <= C[1] <= B[1] or A[1] <= D[1] <= B[1] \
- or C[1] <= A[1] <= D[1] or C[1] <= B[1] <= D[1]:
- return True
- if A[1] == B[1] == C[1] == D[1]:
- if A[0] <= C[0] <= B[0] or A[0] <= D[0] <= B[0] \
- or C[0] <= A[0] <= D[0] or C[0] <= B[0] <= D[0]:
- return True
- line1 = LineString([A, B])
- line2 = LineString([C, D])
- int_pt = line1.intersection(line2)
- try:
- point_of_intersection = int_pt.x, int_pt.y
- return True
- except:
- return False
- def calculate_k(bbox):
- x = [bbox[0], bbox[2]]
- y = [bbox[1], bbox[3]]
- slope, intercept, r_value, p_value, std_err = linregress(x, y)
- # print('k', slope)
- if math.isnan(slope):
- slope = 0
- return slope
- def line_iou(line1, line2, axis=0):
- if line1[0][axis] <= line2[0][axis] <= line2[1][axis] <= line1[1][axis]:
- return 1.0
- if line2[0][axis] <= line1[0][axis] <= line1[1][axis] <= line2[1][axis]:
- return 1.0
- inter = min(line1[1][axis], line2[1][axis]) - max(line1[0][axis], line2[0][axis])
- # union = max(line1[1][axis], line2[1][axis]) - min(line1[0][axis], line2[0][axis])
- union = min(abs(line1[0][axis] - line1[1][axis]), abs(line2[0][axis] - line2[1][axis]))
- if union in [0, 0.]:
- iou = 0.
- else:
- iou = inter / union
- return iou
- def get_cross_line(_line_list, threshold=1, cross_times=0):
- # 根据是否有交点判断表格线
- _cross_line_list = []
- for line1 in _line_list:
- if line1 in _cross_line_list:
- continue
- if abs(line1[2] - line1[0]) > abs(line1[3] - line1[1]):
- p1 = [max(0, line1[0] - threshold), line1[1]]
- p2 = [min(line1[2] + threshold, page_w), line1[3]]
- else:
- p1 = [line1[0], max(0, line1[1] - threshold)]
- p2 = [line1[2], min(line1[3] + threshold, page_h)]
- line1 = [p1[0], p1[1], p2[0], p2[1]]
- _times = 0
- for line2 in _line_list:
- if abs(line2[2] - line2[0]) > abs(line2[3] - line2[1]):
- p3 = [max(0, line2[0] - threshold), line2[1]]
- p4 = [min(line2[2] + threshold, page_w), line2[3]]
- else:
- p3 = [line2[0], max(0, line2[1] - threshold)]
- p4 = [line2[2], min(line2[3] + threshold, page_h)]
- line2 = [p3[0], p3[1], p4[0], p4[1]]
- if line1 == line2:
- continue
- if is_cross(p1, p2, p3, p4):
- _times += 1
- if _times >= cross_times:
- _cross_line_list += [line1]
- break
- return _cross_line_list
- def repair_bias_line(_line_list):
- temp_list = []
- for line in _line_list:
- x0, y0, x1, y1 = line
- _y = min(y0, y1)
- _x = min(x0, x1)
- if abs(x0 - x1) > abs(y0 - y1):
- temp_list.append([x0, _y, x1, _y])
- else:
- temp_list.append([_x, y0, _x, y1])
- _line_list = temp_list
- return _line_list
- def repair_col_line(_straight_list, _bias_list, threshold=2, min_width=7):
- if not _straight_list or not _bias_list:
- print('add_col_bias_line empty', len(_straight_list), len(_bias_list))
- return []
- # 分列
- _straight_list.sort(key=lambda x: (x[0], x[1]))
- cols = []
- col = []
- current_w = _straight_list[0][0]
- for line in _straight_list:
- if abs(line[0] - line[2]) > abs(line[1] - line[3]):
- continue
- if min(line[0], line[2]) - threshold <= current_w <= max(line[0], line[2]) + threshold:
- col.append(line)
- else:
- if col:
- cols.append(col)
- col = [line]
- current_w = line[0]
- if col:
- cols.append(col)
- # 补充col
- new_list = []
- for line in bias_line_list:
- if abs(line[0] - line[2]) > abs(line[1] - line[3]):
- continue
- for col in cols:
- w = col[0][0]
- if w - threshold <= line[0] <= w + threshold or w - threshold <= line[2] <= w + threshold:
- new_list.append([w, line[1] - 3, w, line[3] + 3])
- new_list += _straight_list
- # 去重
- new_list = [str(x) for x in new_list]
- new_list = list(set(new_list))
- new_list = [eval(x) for x in new_list]
- # 分列
- new_list.sort(key=lambda x: (x[0], x[1]))
- cols = []
- col = []
- current_w = new_list[0][0]
- for line in new_list:
- if abs(line[0] - line[2]) > abs(line[1] - line[3]):
- continue
- if min(line[0], line[2]) - threshold <= current_w <= max(line[0], line[2]) + threshold:
- col.append(line)
- else:
- if col:
- cols.append(col)
- col = [line]
- current_w = line[0]
- if col:
- cols.append(col)
- # 删除col
- for col1 in cols:
- for col2 in cols:
- if col1 == col2 or abs(col1[0][0] - col2[0][0]) > min_width:
- continue
- col1_len, col2_len = 0, 0
- for c in col1:
- col1_len += abs(c[1] - c[3])
- for c in col2:
- col2_len += abs(c[1] - c[3])
- if col1_len > col2_len * 3:
- for c in col2:
- if c in new_list:
- new_list.remove(c)
- if col2_len > col1_len * 3:
- for c in col1:
- if c in new_list:
- new_list.remove(c)
- return new_list
- def merge_line(_line_list, threshold=2):
- new_line_list = []
- # 分列
- _line_list.sort(key=lambda x: (x[0], x[1]))
- cols = []
- col = []
- current_w = None
- for line in _line_list:
- if abs(line[0] - line[2]) > abs(line[1] - line[3]):
- continue
- if not col:
- col.append(line)
- current_w = line[0]
- _iou = line_iou([[0, line[1]], [0, line[3]]], [[0, col[0][1]], [0, col[0][3]]], axis=1)
- if min(line[0], line[2]) - threshold <= current_w <= max(line[0], line[2]) + threshold \
- and is_cross(line[0:2], line[2:4], col[-1][0:2], col[-1][2:4]):
- col.append(line)
- elif min(line[0], line[2]) - 2*threshold <= current_w <= max(line[0], line[2]) + 2*threshold \
- and _iou >= 0.1:
- col.append(line)
- else:
- if col:
- cols.append(col)
- col = [line]
- current_w = line[0]
- if col:
- cols.append(col)
- for col in cols:
- temp_c = col[0]
- col_w = col[0][0]
- for i in range(len(col) - 1):
- c = col[i]
- next_c = col[i + 1]
- if is_cross(c[0:2], c[2:4], next_c[0:2], next_c[2:4]) \
- or line_iou([[0, c[1]], [0, c[3]]], [[0, next_c[1]], [0, next_c[3]]], axis=1) >= 0.1:
- temp_c = [col_w, min(temp_c[1], c[1], c[3], next_c[1], next_c[3]), col_w,
- max(temp_c[3], c[1], c[3], next_c[1], next_c[3])]
- else:
- new_line_list.append(temp_c)
- temp_c = next_c
- if not new_line_list or (new_line_list and new_line_list[-1] != temp_c):
- new_line_list.append(temp_c)
- # 分行
- _line_list.sort(key=lambda x: (x[1], x[0]))
- rows = []
- row = []
- current_h = None
- for line in _line_list:
- if abs(line[0] - line[2]) < abs(line[1] - line[3]):
- continue
- if not row:
- row = [line]
- current_h = line[1]
- if min(line[1], line[3]) - threshold <= current_h <= max(line[1], line[3]) + threshold:
- row.append(line)
- else:
- if row:
- rows.append(row)
- row = [line]
- current_h = line[1]
- if row:
- rows.append(row)
- for row in rows:
- temp_r = row[0]
- row_h = row[0][1]
- for i in range(len(row) - 1):
- r = row[i]
- next_r = row[i + 1]
- # if is_cross(r[0:2], r[2:4], next_r[0:2], next_r[2:4]):
- if line_iou([r[0:2], r[2:4]], [next_r[0:2], next_r[2:4]], axis=0) >= 0.1:
- temp_r = [min(temp_r[0], r[0], r[2], next_r[0], next_r[2]), row_h,
- max(temp_r[2], r[0], r[2], next_r[0], next_r[2]), row_h]
- else:
- new_line_list.append(temp_r)
- temp_r = next_r
- if not new_line_list or (new_line_list and new_line_list[-1] != temp_r):
- new_line_list.append(temp_r)
- return new_line_list
- def remove_outline_no_cross(_line_list):
- row_list = []
- col_list = []
- for line in _line_list:
- # 存所有行
- if abs(line[0] - line[2]) > abs(line[1] - line[3]):
- row_list.append(line)
- # 存所有列
- if abs(line[0] - line[2]) < abs(line[1] - line[3]):
- col_list.append(line)
- if not col_list:
- return _line_list
- # 左右两条边框
- col_list.sort(key=lambda x: (x[0], x[1]))
- left_col = col_list[0]
- right_col = col_list[-1]
- # 判断有交点但中间区域无交点
- compare_list = []
- for col in [left_col, right_col]:
- add_h = abs(col[1]-col[3]) / 8
- center_area = [col[1]+add_h, col[3]-add_h]
- cross_cnt = 0
- center_cross_cnt = 0
- center_row_cnt = 0
- for row in row_list:
- if is_cross(row[0:2], row[2:4], col[0:2], col[2:4]):
- if center_area[0] <= row[1] <= center_area[1]:
- center_cross_cnt += 1
- else:
- cross_cnt += 1
- else:
- if center_area[0] <= row[1] <= center_area[1]:
- center_row_cnt += 1
- compare_list.append([cross_cnt, center_cross_cnt, center_row_cnt])
- _flag = True
- for c in compare_list:
- if c[0] >= 2 and c[1] == 0 and c[2] >= 2:
- continue
- _flag = False
- print('compare_list', compare_list)
- if _flag and compare_list[0][1] == compare_list[1][1] \
- and compare_list[0][2] == compare_list[1][2]:
- for col in [left_col, right_col]:
- if col in _line_list:
- _line_list.remove(col)
- return _line_list
- def cross_line_process(_cross_line_list, _bias_line_list):
- if show:
- print('cross_line_process len(_cross_line_list) -1', len(_cross_line_list))
- # 斜线校正
- if _cross_line_list:
- _cross_line_list = repair_bias_line(_cross_line_list)
- if show:
- print('cross_line_process len(_cross_line_list) 0', len(_cross_line_list))
- # 修复竖线
- if _bias_line_list:
- _cross_line_list = repair_col_line(_cross_line_list, _bias_line_list)
- if show:
- print('cross_line_process len(_cross_line_list) 1', len(_cross_line_list))
- # 根据是否有交点判断表格线
- _cross_line_list = get_cross_line(_cross_line_list, threshold=1, cross_times=1)
- if show:
- print('cross_line_process len(_cross_line_list) 2', len(_cross_line_list))
- # 合并线条
- if not _cross_line_list:
- return []
- _cross_line_list = merge_line(_cross_line_list)
- if show:
- print('cross_line_process len(_cross_line_list) 3', len(_cross_line_list))
- # 删除最外层嵌套边框
- _cross_line_list = remove_outline_no_cross(_cross_line_list)
- if show:
- print('cross_line_process len(_cross_line_list) 4', len(_cross_line_list))
- # 复用otr的部分后处理,补线
- _cross_line_list = table_line_pdf(_cross_line_list, page_w, page_h)
- if show:
- print('cross_line_process len(_cross_line_list) 5', len(_cross_line_list))
- return _cross_line_list
- log('into get_page_lines')
- page_h = layout.height
- page_w = layout.width
- element_list = []
- line_list = []
- bias_line_list = []
- text_container_list = []
- lt_rect_list = []
- # 先提取文本框和矩形框
- min_y = 10000
- max_x, max_y = 0, 0
- for element in layout:
- if isinstance(element, LTTextContainer):
- text_container_list.append(element)
- if isinstance(element, LTRect):
- # print(element.bbox, element.height, element.width)
- lt_rect_list.append(element)
- if element.bbox[1] <= min_y:
- min_y = element.bbox[1]
- if element.bbox[3] <= min_y:
- min_y = element.bbox[3]
- if element.bbox[1] > max_y:
- max_y = element.bbox[1]
- if element.bbox[3] > max_y:
- max_y = element.bbox[3]
- if element.bbox[0] > max_x:
- max_x = element.bbox[0]
- if element.bbox[2] > max_x:
- max_x = element.bbox[2]
- if max_y > page_h:
- page_h = max_y + 20
- if max_x > page_w:
- page_w = max_x + 20
- # 矩形框y有负数
- if min_y < 0:
- for lt_rect in lt_rect_list:
- if lt_rect.y0 < 0 or lt_rect.y1 < 0:
- new_y0 = 10 if lt_rect.y0 < 0 else lt_rect.y0
- new_y1 = 10 if lt_rect.y1 < 0 else lt_rect.y1
- lt_rect.set_bbox((lt_rect.x0, new_y0, lt_rect.x1, new_y1))
- _plot([x.bbox for x in lt_rect_list], 'get_page_lines start', mode=2)
- # 合并矩形框
- delete_lt_rect_list = []
- for i in range(len(lt_rect_list)):
- lt_rect1 = lt_rect_list[i]
- b1 = lt_rect1.bbox
- if lt_rect1 in delete_lt_rect_list:
- continue
- for j in range(i+1, len(lt_rect_list)):
- lt_rect2 = lt_rect_list[j]
- b2 = lt_rect2.bbox
- if lt_rect2 in delete_lt_rect_list:
- continue
- if bbox_iou(b1, b2) >= 0.9:
- delete_lt_rect_list.append(lt_rect2)
- # 筛选出线形矩形和非线形矩形
- line_rect_list = []
- non_line_rect_list = []
- for lt_rect in lt_rect_list:
- if (lt_rect.height <= 2) ^ (lt_rect.width <= 2):
- line_rect_list.append(lt_rect)
- if lt_rect.height > 2 and lt_rect.width > 2:
- non_line_rect_list.append(lt_rect)
- # 非线形矩形若与线形矩形距离较近,则删除
- threshold = 5
- for n_rect in non_line_rect_list:
- if n_rect in delete_lt_rect_list:
- continue
- middle_x = (n_rect.x0 + n_rect.x1) / 2
- middle_y = (n_rect.y0 + n_rect.y1) / 2
- for rect in line_rect_list:
- if rect in delete_lt_rect_list:
- continue
- if rect.height >= rect.width:
- if n_rect.width / 2 - threshold <= abs(rect.x0 - middle_x) <= n_rect.width / 2 + threshold:
- delete_lt_rect_list.append(n_rect)
- else:
- if n_rect.height / 2 - threshold <= abs(rect.y0 - middle_y) <= n_rect.height / 2 + threshold:
- delete_lt_rect_list.append(n_rect)
- # print('len(delete_lt_rect_list)', len(delete_lt_rect_list))
- # 寻找每个文本框对应的最小矩形框
- text_lt_rect_list = []
- for text_lt_rect in text_container_list:
- text_box = text_lt_rect.bbox
- min_area = 1000000
- min_lt_rect = None
- for lt_rect in lt_rect_list:
- if lt_rect in delete_lt_rect_list:
- continue
- if lt_rect in text_lt_rect_list:
- continue
- if lt_rect.height <= 5 or lt_rect.width <= 5:
- continue
- _bbox = lt_rect.bbox
- _area = abs(_bbox[2] - _bbox[0]) * abs(_bbox[3] - _bbox[1])
- _iou = bbox_iou(_bbox, text_box)
- if _iou >= 0.5 and _area < min_area:
- min_area = _area
- min_lt_rect = lt_rect
- # 匹配不到再去小框里找
- if min_lt_rect is None:
- for lt_rect in lt_rect_list:
- if lt_rect in delete_lt_rect_list:
- continue
- if lt_rect in text_lt_rect_list:
- continue
- if lt_rect.height <= 5 or lt_rect.width <= 5:
- _bbox = lt_rect.bbox
- _area = abs(_bbox[2] - _bbox[0]) * abs(_bbox[3] - _bbox[1])
- _iou = bbox_iou(_bbox, text_box)
- if _iou >= 0.5 and _area < min_area:
- min_area = _area
- min_lt_rect = lt_rect
- # print(text_lt_rect.get_text(), min_lt_rect)
- if min_lt_rect is not None:
- text_lt_rect_list.append(min_lt_rect)
- # 删除包含关系的矩形框
- delete_lt_rect_list += text_lt_rect_list
- # for i in range(len(lt_rect_list)):
- # lt_rect1 = lt_rect_list[i]
- # b1 = lt_rect1.bbox
- # if lt_rect1 in delete_lt_rect_list:
- # continue
- # for j in range(i+1, len(lt_rect_list)):
- # lt_rect2 = lt_rect_list[j]
- # b2 = lt_rect2.bbox
- # if lt_rect2 in delete_lt_rect_list:
- # continue
- # if b1[0] <= b2[0] <= b2[1] <= b1[2] and b1[1] <= b2[1] <= b2[3] <= b1[3]:
- # delete_lt_rect_list.append(b2)
- # if b2[0] <= b1[0] <= b1[1] <= b2[2] and b2[1] <= b1[1] <= b1[3] <= b2[3]:
- # delete_lt_rect_list.append(b1)
- # 在提取表格框,并排除文本框
- for element in layout:
- _b = element.bbox
- # 只取这三种类型的bbox
- if isinstance(element, LTRect):
- if element in delete_lt_rect_list:
- continue
- line_list += [[_b[0], _b[1], _b[0], _b[3]], [_b[0], _b[1], _b[2], _b[1]],
- [_b[2], _b[1], _b[2], _b[3]], [_b[0], _b[3], _b[2], _b[3]]]
- elif isinstance(element, (LTCurve, LTLine)):
- # element_list.append(element)
- if element.height > 10 or element.width > 10:
- # print('element.height, element.width', element.bbox, type(element))
- # k = calculate_k(element.bbox)
- # if 1.73 / 3 < abs(k) < 1.73:
- # continue
- # else:
- # bias_line_list.append(element.bbox)
- # continue
- if element.height >= element.width:
- line_list += [[_b[0], _b[1], _b[0], _b[3]], [_b[2], _b[1], _b[2], _b[3]]]
- else:
- line_list += [[_b[0], _b[1], _b[2], _b[1]], [_b[0], _b[3], _b[2], _b[3]]]
- if show:
- print('get_page_lines line_list', line_list)
- print('get_page_lines bias_line_list', bias_line_list)
- _plot(line_list+bias_line_list, 'line_list+bias_line_list', mode=2)
- if not line_list and not bias_line_list:
- return []
- # # 是否使用斜线来生成表格
- line_list_copy = copy.deepcopy(line_list)
- # if len(line_list) < 6 and len(bias_line_list) > len(line_list) * 2:
- # if show:
- # print('use bias line')
- # # bias_line_list += add_col_bias_line(line_list, bias_line_list)
- # line_list = bias_line_list
- # 去重
- line_list = [str(x) for x in line_list]
- line_list = list(set(line_list))
- line_list = [eval(x) for x in line_list]
- # 根据是否有交点判断表格线
- cross_line_list = get_cross_line(line_list_copy, threshold=2, cross_times=1)
- if show:
- print('get_page_lines cross_line_list', cross_line_list)
- if not cross_line_list:
- # 将线全部合并再获取一次
- cross_line_list = get_cross_line(line_list_copy, threshold=2, cross_times=1)
- if not cross_line_list:
- return []
- _plot(line_list, 'get_cross_line', mode=2)
- cross_line_list = cross_line_process(cross_line_list, bias_line_list)
- if not cross_line_list:
- cross_line_list = get_cross_line(line_list_copy, threshold=2, cross_times=1)
- cross_line_list = cross_line_process(cross_line_list, bias_line_list)
- if show:
- print('get_page_lines cross_line_list2', cross_line_list)
- # show
- if show:
- print('len(cross_line_list)', len(cross_line_list))
- _plot(cross_line_list, 'cross_line_process', mode=2)
- lt_line_list = []
- for line in cross_line_list:
- lt_line_list.append(LTLine(1, (float(line[0]), float(line[1])),
- (float(line[2]), float(line[3]))))
- log("pdf page %s has %s lines" % (str(page_no), str(len(lt_line_list))))
- return lt_line_list
- def recognize_text(self, layout, page_no, lt_text_list, lt_line_list):
- list_tables, filter_objs, _, connect_textbox_list = self.lt.recognize_table(lt_text_list, lt_line_list, from_pdf=True)
- self._page.in_table_objs = filter_objs
- # print("=======text_len:%d:filter_len:%d"%(len(lt_text_list),len(filter_objs)))
- for table in list_tables:
- _table = _Table(table["table"], table["bbox"])
- # self._page.children.append(_table)
- self._page.add_child(_table)
- list_sentences = ParseUtils.recognize_sentences(lt_text_list, filter_objs,
- layout.bbox, page_no)
- for sentence in list_sentences:
- _sen = _Sentence(sentence.text, sentence.bbox)
- self._page.add_child(_sen)
- # pdf对象需反向排序
- self._page.is_reverse = True
- return list_tables
- def is_text_legal(self, lt_text_list, page_no):
- # 无法识别pdf字符编码,整页用ocr
- text_temp = ""
- for _t in lt_text_list:
- text_temp += _t.get_text()
- if re.search('[(]cid:[0-9]+[)]', text_temp):
- log("text has cid! try pymupdf...")
- page_image = self.get_page_image(page_no)
- if judge_error_code(page_image):
- self._page.error_code = page_image
- else:
- _image = _Image(page_image[1], page_image[0])
- self._page.add_child(_image)
- return False
- match1 = re.findall(get_garble_code(), text_temp)
- # match2 = re.search('[\u4e00-\u9fa5]', text_temp)
- if len(match1) > 8 and len(text_temp) > 10:
- log("pdf garbled code! try pymupdf... " + text_temp[:20])
- page_image = self.get_page_image(page_no)
- if judge_error_code(page_image):
- self._page.error_code = page_image
- else:
- _image = _Image(page_image[1], page_image[0])
- self._page.add_child(_image)
- return False
- return True
- def judge_b_table(self, lt_text_list, table_list):
- table_h_list = []
- for table in table_list:
- table_h_list.append([table.get('bbox')[1], table.get('bbox')[3]])
- # 先分行
- lt_text_list.sort(key=lambda x: (x.bbox[1], x.bbox[0]))
- lt_text_row_list = []
- current_h = lt_text_list[0].bbox[1]
- row = []
- threshold = 2
- for lt_text in lt_text_list:
- bbox = lt_text.bbox
- if current_h - threshold <= bbox[1] <= current_h + threshold:
- row.append(lt_text)
- else:
- if row:
- lt_text_row_list.append(row)
- row = [lt_text]
- current_h = lt_text.bbox[1]
- if row:
- lt_text_row_list.append(row)
- # print('lt_text_row_list')
- # for r in lt_text_row_list:
- # print('r', [x.get_text() for x in r])
- # 判断文本中间是否是空格,或一行文本中间有多个
- is_b_table_flag = False
- is_b_table_cnt = 3
- tolerate_cnt = 2
- t_cnt = 0
- row_cnt = 0
- b_table_row_list = []
- for row in lt_text_row_list:
- # 水印行跳过
- if len(row) == 1 and len(row[0].get_text()[:-1]) == 1:
- continue
- # 目录行跳过
- continue_flag = False
- for r in row:
- if re.search('[.·]{7,}', r.get_text()):
- continue_flag = True
- break
- if continue_flag:
- continue
- if len(row) == 1:
- text = row[0].get_text()
- bbox = row[0].bbox
- match = re.search('[ ]{3,}', text)
- if match and re.search('[\u4e00-\u9fff]{2,}', text[:match.span()[0]]) \
- and re.search('[\u4e00-\u9fff]{2,}', text[match.span()[1]:]):
- row_cnt += 1
- t_cnt = 0
- b_table_row_list += row
- else:
- # 容忍
- if t_cnt < tolerate_cnt:
- t_cnt += 1
- continue
- row_cnt = 0
- b_table_row_list = []
- else:
- row_cnt += 1
- t_cnt = 0
- b_table_row_list += row
- if row_cnt >= is_b_table_cnt:
- # 判断在不在有边框表格的范围
- in_flag = False
- for table_h in table_h_list:
- for b in b_table_row_list:
- # print('b.bbox', b.bbox)
- # print(table_h)
- if table_h[1] <= b.bbox[1] <= table_h[0] or table_h[1] <= b.bbox[3] <= table_h[0]:
- in_flag = True
- break
- if in_flag:
- break
- if in_flag:
- is_b_table_flag = False
- t_cnt = 0
- row_cnt = 0
- else:
- # print('True b_table_row_list', b_table_row_list)
- # print('table_h_list', table_h_list)
- is_b_table_flag = True
- break
- log('pdf is_b_table_flag ' + str(is_b_table_flag))
- return is_b_table_flag
- def convert_page(self, page, page_no):
- layout = self.get_layout(page, page_no)
- if self._doc.error_code is not None:
- return
- if judge_error_code(layout):
- self._page.error_code = layout
- return
- # 判断该页的对象类型,并存储
- lt_text_list = []
- lt_image_list = []
- for x in layout:
- if isinstance(x, (LTTextBoxHorizontal, LTTextBoxVertical)):
- lt_text_list.append(x)
- if isinstance(x, LTFigure):
- for y in x:
- if isinstance(y, LTImage):
- # 小的图忽略
- if y.width <= 300 and y.height <= 300:
- continue
- # 图的width超过layout width,很大可能是水印
- if y.width > layout.width + 20:
- continue
- lt_image_list.append(y)
- # 解决重复粗体字问题
- lt_text_list = self.delete_bold_text_duplicate(lt_text_list)
- # 删除水印字
- lt_text_list = self.delete_water_mark(lt_text_list, layout.bbox, 15)
- log("convert_pdf page " + str(page_no))
- log("len(lt_image_list), len(lt_text_list) " + str(len(lt_image_list)) + " " + str(len(lt_text_list)))
- log('layout.width, layout.height ' + str(layout.width) + str(layout.height))
- # 若只有文本且图片数为0,直接提取文字及表格
- # if only_image == 0 and image_count == 0:
- # if len(lt_image_list) == 0 and len(lt_text_list) > 0:
- # # PDFPlumber
- # if self.has_init_pdf[3] == 0:
- # self.init_package("pdfplumber")
- # if self._doc.error_code is not None:
- # self._doc.error_code = None
- # log("init pdfplumber failed! try pymupdf...")
- # # 调用pdfplumber获取pdf图片报错,则使用pypdf2将pdf转html
- # page_image = self.get_page_image(page_no)
- # if judge_error_code(page_image):
- # self._page.error_code = page_image
- # else:
- # _image = _Image(page_image[1], page_image[0])
- # self._page.add_child(_image)
- # return
- #
- # if not self.is_text_legal(lt_text_list, page_no):
- # return
- #
- # # 根据text规律,判断该页是否可能有无边框表格
- # start_time = time.time()
- # if self.judge_b_table(lt_text_list):
- # page_image = self.get_page_image(page_no)
- # if judge_error_code(page_image):
- # self._page.error_code = page_image
- # else:
- # _image = _Image(page_image[1], page_image[0])
- # _image.is_from_pdf = True
- # _image.b_table_from_text = True
- # _image.b_table_text_obj_list = lt_text_list
- # _image.b_table_layout_size = (layout.width, layout.height)
- # self._page.add_child(_image)
- # log('convert_pdf judge_b_table set image cost: ' + str(time.time()-start_time))
- #
- # try:
- # lt_line_list = self.get_page_lines(layout, page_no)
- # except:
- # traceback.print_exc()
- # lt_line_list = []
- # self._page.error_code = [-13]
- # try:
- # # lt_line_list = self.get_text_lines(page,page_no)
- # self.recognize_text(layout, page_no, lt_text_list, lt_line_list)
- # except:
- # traceback.print_exc()
- # self._page.error_code = [-8]
- # 若该页图片数量过多,或无文本,则直接ocr整页识别
- # elif image_count > 3 or only_image == 1:
- if len(lt_image_list) > 4 or len(lt_text_list) == 0:
- page_image = self.get_page_image(page_no)
- if judge_error_code(page_image):
- self._page.error_code = page_image
- else:
- _image = _Image(page_image[1], page_image[0])
- _image.is_from_pdf = True
- self._page.add_child(_image)
- # 正常读取该页对象
- else:
- # 图表对象
- for image in lt_image_list:
- try:
- print("pdf2text LTImage size", page_no, image.width, image.height)
- image_stream = image.stream.get_data()
- # 小的图忽略
- if image.width <= 300 and image.height <= 300:
- continue
- # 查看提取的图片高宽,太大则用pdf输出图进行ocr识别
- img_test = Image.open(io.BytesIO(image_stream))
- if image.height >= 1000 and image.width >= 1000:
- page_image = self.get_page_image(page_no)
- if judge_error_code(page_image):
- self._page.error_code = page_image
- else:
- _image = _Image(page_image[1], page_image[0])
- _image.is_from_pdf = True
- self._page.add_child(_image)
- image_md5 = get_md5_from_bytes(page_image[1])
- self.md5_image_obj_list.append([image_md5, _image])
- return
- # 比较小的图则直接保存用ocr识别
- else:
- temp_path = self.unique_type_dir + 'page' + str(page_no) \
- + '_lt' + str(lt_image_list.index(image)) + '.jpg'
- img_test.save(temp_path)
- with open(temp_path, "rb") as ff:
- image_stream = ff.read()
- _image = _Image(image_stream, temp_path, image.bbox)
- self._page.add_child(_image)
- image_md5 = get_md5_from_bytes(image_stream)
- self.md5_image_obj_list.append([image_md5, _image])
- except Exception:
- log("pdf2text pdfminer read image in page " + str(page_no) +
- " fail! use pymupdf read image...")
- traceback.print_exc()
- # pdf对象需反向排序
- self._page.is_reverse = True
- self.init_package("pdfplumber")
- if not self.is_text_legal(lt_text_list, page_no):
- return
- try:
- lt_line_list = self.get_page_lines(layout, page_no)
- except:
- traceback.print_exc()
- lt_line_list = []
- self._page.error_code = [-13]
- table_list = self.recognize_text(layout, page_no, lt_text_list, lt_line_list)
- # 根据text规律,判断该页是否可能有无边框表格
- if self.judge_b_table(lt_text_list, table_list):
- page_image = self.get_page_image(page_no)
- if judge_error_code(page_image):
- self._page.error_code = page_image
- else:
- _image = _Image(page_image[1], page_image[0])
- _image.is_from_pdf = True
- _image.b_table_from_text = True
- _image.b_table_text_obj_list = lt_text_list
- _image.b_table_layout_size = (layout.width, layout.height)
- self._page.add_child(_image)
- def get_layout(self, page, page_no):
- log("get_layout")
- if self.has_init_pdf[0] == 0:
- self.init_package("pdfminer")
- if self._doc.error_code is not None:
- return
- # 获取该页layout
- start_time = time.time()
- try:
- if get_platform() == "Windows":
- # origin_pdf_analyze = pdf_analyze.__wrapped__
- # layout = origin_pdf_analyze(self.interpreter, page, self.device)
- layout = pdf_analyze(self.interpreter, page, self.device, page_no)
- else:
- layout = pdf_analyze(self.interpreter, page, self.device, page_no)
- except TimeoutError as e:
- log("pdf2text pdfminer read pdf page " + str(page_no) + " time out! " + str(time.time() - start_time))
- layout = [-4]
- except Exception:
- traceback.print_exc()
- log("pdf2text pdfminer read pdf page " + str(page_no) + " error! continue...")
- layout = [-3]
- return layout
- def get_page_image(self, page_no):
- log("")
- try:
- if self.has_init_pdf[1] == 0:
- self.init_package("PyMuPDF")
- if self._doc.error_code is not None:
- return
- # save_dir = self.path.split(".")[-2] + "_" + self.path.split(".")[-1]
- output = self.unique_type_dir + "page" + str(page_no) + ".png"
- page = self.doc_pymupdf.loadPage(page_no)
- rotate = int(0)
- zoom_x = 2.
- zoom_y = 2.
- mat = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)
- pix = page.getPixmap(matrix=mat, alpha=False)
- pix.writePNG(output)
- # 输出图片resize
- self.resize_image(output)
- with open(output, "rb") as f:
- pdf_image = f.read()
- return [output, pdf_image]
- except ValueError as e:
- traceback.print_exc()
- if str(e) == "page not in document":
- log("pdf2Image page not in document! continue... page " + str(page_no))
- return [0]
- elif "encrypted" in str(e):
- log("pdf2Image document need password " + str(page_no))
- return [-7]
- except RuntimeError as e:
- if "cannot find page" in str(e):
- log("pdf2Image page {} not in document! continue... ".format(str(page_no)) + str(e))
- return [0]
- else:
- traceback.print_exc()
- return [-3]
- def get_all_page_image(self):
- log("")
- if self.has_init_pdf[1] == 0:
- self.init_package("PyMuPDF")
- if self._doc.error_code is not None:
- return
- page_count = self.doc_pymupdf.page_count
- for page_no in range(page_count):
- # 限制pdf页数,只取前10页后10页
- if page_count > 20:
- if 10 <= page_no < page_count - 10:
- continue
- self._page = _Page(None, page_no)
- page_image = self.get_page_image(page_no)
- if judge_error_code(page_image):
- self._page.error_code = page_image
- else:
- _image = _Image(page_image[1], page_image[0])
- self._page.add_child(_image)
- # 报错继续读后面页面
- if self._doc.error_code is None and self._page.error_code is not None:
- continue
- self._doc.add_child(self._page)
- def connect_table(self, html_list):
- if not html_list:
- return html_list
- # 判断初始条件1
- # 0: 前一页最后一个表格为A,后一页第一个表格为B
- # 1.1: A后无文本(除了页码),且B前无文本(除了页码)
- # 1.2: B前有文字(可能是页眉,小于60字),且B的第一行前几个单元格为空,且第一行不为空的单元格有文字较多的格子
- # 1.3: B前有文字(可能是页眉,小于60字),且B的第一行第一个单元格为空,且有文字的格子数量占所有格子的一半
- # 1.4: B前有文字(可能是页眉,小于60字),且B的第一行第一个单元格为纯数字序号
- # 1.5: A后有文字(除了页码还有页眉),且A的后面只有一行且中文不超过15个字
- connect_flag_list = []
- soup_list = []
- for i, h in enumerate(html_list):
- soup = BeautifulSoup(h, 'lxml')
- soup_list.append(soup)
- # 找最后一个表格
- last_table_start, last_table_end = None, None
- match = re.finditer('<table', h)
- for m in match:
- last_table_start = m.span()[0]
- if last_table_start is not None:
- match = re.finditer('</table>', h[last_table_start:])
- for m in match:
- last_table_end = m.span()[1] + last_table_start
- # 最后一个表格后有无除了页码外的内容
- connect_flag1 = False
- if last_table_end is not None:
- match = re.findall('[^-/第页0-9,,]', re.sub('<div>|</div>', '', h[last_table_end:]))
- # print('match', match.group())
- # if not match or match.group() == '':
- if len(match) == 0:
- connect_flag1 = True
- # 有页脚
- if not connect_flag1:
- if len(re.findall('<div>', h[last_table_end:])) <= 1 \
- and len(re.findall('[\u4e00-\u9fff]', h[last_table_end:])) <= 60:
- connect_flag1 = True
- # 找第一个表格
- first_table_start, first_table_end = None, None
- match = re.finditer('<table', h)
- for m in match:
- first_table_start = m.span()[0]
- break
- # 第一个表格前有无内容
- connect_flag2 = False
- if first_table_start is not None and first_table_start == 0:
- connect_flag2 = True
- # 有内容但是是页眉
- if not connect_flag2:
- tables = soup.findAll('table')
- if tables:
- first_table = tables[0]
- rows = first_table.findAll('tr')
- if rows:
- first_row = rows[0]
- col_text_len_list = [len(x.text) for x in first_row]
- col_text_list = [x.text for x in first_row]
- # 文字大于60且第一个为空
- if not connect_flag2 and len(h[:first_table_start]) <= 60 and col_text_len_list[0] == 0 and max(col_text_len_list) >= 30:
- connect_flag2 = True
- # 有文字格子数占一半一下且第一个格子为空
- if not connect_flag2 and col_text_len_list.count(0) >= len(col_text_len_list) / 2 and col_text_len_list[0] == 0:
- connect_flag2 = True
- # 表格前最多只有一行且第一个格子为纯数字
- if not connect_flag2 and len(col_text_list) > 0 and \
- len(re.findall('<div>', h[:first_table_start])) <= 0 and \
- len(re.findall('\d', col_text_list[0])) == len(col_text_list[0]):
- connect_flag2 = True
- # if not connect_flag2 and len(re.findall('<div>', h[:first_table_start])) <= 0 and len(re.findall('[\u4e00-\u9fff]', h[:first_table_start])) <= 25:
- # connect_flag2 = True
- connect_flag_list.append([i, connect_flag2, connect_flag1])
- print('connect_flag_list', connect_flag_list)
- # 根据条件1合并需连接页码,形成组
- connect_pages_list = []
- if connect_flag_list:
- temp_list = [connect_flag_list[0]]
- for i in range(1, len(connect_flag_list)):
- c = connect_flag_list[i]
- if c[1] and temp_list[-1][2]:
- temp_list.append(c)
- else:
- if temp_list:
- connect_pages_list.append(temp_list)
- temp_list = [c]
- # connect_pages_list.append([c])
- if temp_list:
- connect_pages_list.append(temp_list)
- print('connect_pages_list', connect_pages_list)
- # 判断后续条件:判断组内列数是否相同
- connect_pages_list2 = []
- for c_list in connect_pages_list:
- if len(c_list) == 1:
- connect_pages_list2.append(c_list)
- else:
- col_cnt_list = []
- # 单元格可能被复制了,相同的合并当做一列
- merge_col_cnt_list = []
- for c in c_list:
- soup = soup_list[c[0]]
- table1 = soup.findAll('table')[-1]
- table2 = soup.findAll('table')[0]
- tr1 = table1.findAll('tr')
- tr2 = table2.findAll('tr')
- td1 = tr1[-1].findAll('td')
- td2 = tr2[0].findAll('td')
- col_cnt_list.append([len(td2), len(td1)])
- # # 计算合并重复文本格子后的列数
- # last_text = td1[0].text
- # merge_td1 = [last_text]
- # for td in td1:
- # if td.text == last_text:
- # continue
- # else:
- # merge_td1.append(td.text)
- # last_text = td.text
- # last_text = td2[0].text
- # merge_td2 = [last_text]
- # for td in td2:
- # if td.text == last_text:
- # continue
- # else:
- # merge_td2.append(td.text)
- # last_text = td.text
- # merge_col_cnt_list.append([len(merge_td2), len(merge_td1)])
- # 判断
- new_c_list = [c_list[0]]
- # print('col_cnt_list', col_cnt_list)
- for i in range(len(col_cnt_list) - 1):
- if col_cnt_list[i][1] != col_cnt_list[i + 1][0]:
- # and merge_col_cnt_list[i][1] != merge_col_cnt_list[i + 1][0]:
- connect_pages_list2.append(new_c_list)
- new_c_list = [c_list[i + 1]]
- else:
- new_c_list.append(c_list[i + 1])
- if new_c_list:
- connect_pages_list2.append(new_c_list)
- print('connect_pages_list2', connect_pages_list2)
- # 符合连接条件的拼接表格
- new_html_list = []
- for c_list in connect_pages_list2:
- if len(c_list) == 1:
- new_html_list.append(html_list[c_list[0][0]])
- continue
- new_html = ''
- for c in c_list:
- match = re.finditer('</table>', new_html)
- last_table_index = None
- for m in match:
- last_table_index = m.span()[0]
- new_html += html_list[c[0]]
- # print('html_list[c[0]]', html_list[c[0]])
- if last_table_index is None:
- continue
- match = re.finditer('<table border="1">', new_html[last_table_index:])
- first_table_index = None
- for m in match:
- first_table_index = last_table_index + m.span()[1]
- break
- if first_table_index is None:
- continue
- # print('re', re.findall('</table>.*?<table border="1">', new_html[last_table_index:first_table_index]))
- # 非贪婪匹配
- new_html_sub = re.sub('</table>.*?<table border="1">',
- '<tr><td>#@#@#</td></tr>',
- new_html[last_table_index:first_table_index])
- new_html = new_html[:last_table_index] + new_html_sub + new_html[first_table_index:]
- # print('new_html', new_html)
- # new_html = new_html[:-5]
- # ([-/第页0-9]|<div>|</div>)*
- # 非贪婪匹配
- # match = re.finditer('</table>.*?<table border="1">', new_html)
- # for m in match:
- # if '#@#@#' in m.group():
- #
- # new_html = re.sub('</table>.*#@#@#.*?<table border="1">',
- # '<tr><td>#@#@#</td></tr>',
- # new_html)
- # print('new_html', new_html)
- soup = BeautifulSoup(new_html, 'lxml')
- trs = soup.findAll('tr')
- decompose_trs = []
- for i in range(len(trs)):
- if trs[i].get_text() == '#@#@#':
- td1 = trs[i - 1].findAll('td')
- td2 = trs[i + 1].findAll('td')
- if td2[0].get_text() == '':
- # 解决连续多页是一行表格,该行会被去掉问题
- find_father = False
- for father, son in decompose_trs:
- # print('son', son)
- # print('td1', trs[i - 1])
- if father != '' and son == trs[i - 1]:
- td_father = father.findAll('td')
- for j in range(len(td_father)):
- # print('td_father[j].string3', td_father[j].string)
- td_father[j].string = td_father[j].get_text() + td2[j].get_text()
- # print('td_father[j].string4', td_father[j].string)
- find_father = True
- decompose_trs.append([father, trs[i + 1]])
- break
- if not find_father:
- for j in range(len(td1)):
- # print('td1[j].string1', td1[j].string)
- td1[j].string = td1[j].get_text() + td2[j].get_text()
- # print('td1[j].string2', td1[j].string)
- decompose_trs.append([trs[i - 1], trs[i + 1]])
- # print('trs[i + 1]', trs[i + 1])
- # trs[i + 1].decompose()
- # print('trs[i-1]', trs[i-1])
- # trs[i].decompose()
- decompose_trs.append(['', trs[i]])
- # print('decompose_trs', decompose_trs)
- # for father, son in decompose_trs:
- # print('father', father)
- # print('son', son)
- # print('len(decompose_trs)', len(decompose_trs))
- for father, son in decompose_trs:
- for tr in trs:
- if tr == son:
- tr.decompose()
- break
- new_html = str(soup)
- new_html_list.append(new_html)
- html_str = ''
- for h in new_html_list:
- html_str += h
- return [html_str]
- def get_html(self):
- if self._doc.error_code is not None:
- return self._doc.error_code
- self.convert()
- if self._doc.error_code is not None:
- return self._doc.error_code
- html = self._doc.get_html(return_list=True)
- # 表格连接
- try:
- html = self.connect_table(html)
- except:
- traceback.print_exc()
- return [-12]
- return html
- def delete_water_mark(self, lt_text_list, page_bbox, times=5):
- # 删除过多重复字句,为水印
- duplicate_dict = {}
- for _obj in lt_text_list:
- t = _obj.get_text()
- if t in duplicate_dict.keys():
- duplicate_dict[t][0] += 1
- duplicate_dict[t][1].append(_obj)
- else:
- duplicate_dict[t] = [1, [_obj]]
- delete_text = []
- for t in duplicate_dict.keys():
- if duplicate_dict[t][0] >= times:
- obj_list = duplicate_dict[t][1]
- obj_list.sort(key=lambda x: x.bbox[3])
- obj_distance_h = abs(obj_list[-1].bbox[3] - obj_list[0].bbox[1])
- obj_list.sort(key=lambda x: x.bbox[2])
- obj_distance_w = abs(obj_list[-1].bbox[2] - obj_list[0].bbox[0])
- if obj_distance_h >= abs(page_bbox[1] - page_bbox[3]) * 0.7 \
- and obj_distance_w >= abs(page_bbox[0] - page_bbox[2]) * 0.7:
- delete_text.append(t)
- temp_text_list = []
- for _obj in lt_text_list:
- t = _obj.get_text()
- if t not in delete_text:
- temp_text_list.append(_obj)
- return temp_text_list
- def resize_image(self, img_path, max_size=2000):
- _img = cv2.imread(img_path)
- if _img.shape[0] <= max_size or _img.shape[1] <= max_size:
- return
- else:
- resize_axis = 0 if _img.shape[0] >= _img.shape[1] else 1
- ratio = max_size / _img.shape[resize_axis]
- new_shape = [0, 0]
- new_shape[resize_axis] = max_size
- new_shape[1 - resize_axis] = int(_img.shape[1 - resize_axis] * ratio)
- _img = cv2.resize(_img, (new_shape[1], new_shape[0]))
- cv2.imwrite(img_path, _img)
- def get_single_pdf(self, path, page_no):
- log("into get_single_pdf")
- try:
- pdf_origin = copy.deepcopy(self.doc_pypdf2)
- pdf_new = copy.deepcopy(self.doc_pypdf2_new)
- pdf_new.addPage(pdf_origin.getPage(page_no))
- path_new = path.split(".")[0] + "_split.pdf"
- with open(path_new, "wb") as ff:
- pdf_new.write(ff)
- return path_new
- except PyPDF2.utils.PdfReadError as e:
- return [-3]
- except Exception as e:
- log("get_single_pdf error! page " + str(page_no))
- return [-3]
- def get_text_font():
- def flags_decomposer(flags):
- """Make font flags human readable."""
- l = []
- if flags & 2 ** 0:
- l.append("superscript")
- if flags & 2 ** 1:
- l.append("italic")
- if flags & 2 ** 2:
- l.append("serifed")
- else:
- l.append("sans")
- if flags & 2 ** 3:
- l.append("monospaced")
- else:
- l.append("proportional")
- if flags & 2 ** 4:
- l.append("bold")
- return ", ".join(l)
- def get_underlined_textLines(page):
- """
- 获取某页pdf上的所有下划线文本信息
- :param page: fitz中的一页
- :return: list of tuples,每个tuple都是一个完整的下划线覆盖的整体:[(下划线句, 所在blk_no, 所在line_no), ...]
- """
- paths = page.get_drawings() # get drawings on the current page
- # 获取该页内所有的height很小的bbox。因为下划线其实大多是这种矩形
- # subselect things we may regard as lines
- lines = []
- for p in paths:
- for item in p["items"]:
- if item[0] == "l": # an actual line
- p1, p2 = item[1:]
- if p1.y == p2.y:
- lines.append((p1, p2))
- elif item[0] == "re": # a rectangle: check if height is small
- r = item[1]
- if r.width > r.height and r.height <= 2:
- lines.append((r.tl, r.tr)) # take top left / right points
- # 获取该页的`max_lineheight`,用于下面比较距离使用
- blocks = page.get_text("dict", flags=11)["blocks"]
- max_lineheight = 0
- for b in blocks:
- for l in b["lines"]:
- bbox = fitz.Rect(l["bbox"])
- if bbox.height > max_lineheight:
- max_lineheight = bbox.height
- underlined_res = []
- # 开始对下划线内容进行查询
- # make a list of words
- words = page.get_text("words")
- # if underlined, the bottom left / right of a word
- # should not be too far away from left / right end of some line:
- for wdx, w in enumerate(words): # w[4] is the actual word string
- r = fitz.Rect(w[:4]) # first 4 items are the word bbox
- for p1, p2 in lines: # check distances for start / end points
- if abs(r.bl - p1) <= max_lineheight: # 当前word的左下满足下划线左下
- if abs(r.br - p2) <= max_lineheight: # 当前word的右下满足下划线右下(单个词,无空格)
- print(f"Word '{w[4]}' is underlined! Its block-line number is {w[-3], w[-2]}")
- underlined_res.append((w[4], w[-3], w[-2])) # 分别是(下划线词,所在blk_no,所在line_no)
- break # don't check more lines
- else: # 继续寻找同line右侧的有缘人,因为有些下划线覆盖的词包含多个词,多个词之间有空格
- curr_line_num = w[-2] # line nunmber
- for right_wdx in range(wdx + 1, len(words), 1):
- _next_w = words[right_wdx]
- if _next_w[-2] != curr_line_num: # 当前遍历到的右侧word已经不是当前行的了(跨行是不行的)
- break
- _r_right = fitz.Rect(_next_w[:4]) # 获取当前同行右侧某word的方框4点
- if abs(_r_right.br - p2) <= max_lineheight: # 用此word右下点和p2(目标下划线右上点)算距离,距离要小于max_lineheight
- print(
- f"Word '{' '.join([_one_word[4] for _one_word in words[wdx:right_wdx + 1]])}' is underlined! " +
- f"Its block-line number is {w[-3], w[-2]}")
- underlined_res.append(
- (' '.join([_one_word[4] for _one_word in words[wdx:right_wdx + 1]]),
- w[-3], w[-2])
- ) # 分别是(下划线词,所在blk_no,所在line_no)
- break # don't check more lines
- return underlined_res
- _p = r'C:\Users\Administrator\Desktop\test_pdf\error2-2.pdf'
- doc_pymupdf = read_pymupdf(_p)
- page = doc_pymupdf[0]
- blocks = page.get_text("dict", flags=11)["blocks"]
- for b in blocks: # iterate through the text blocks
- for l in b["lines"]: # iterate through the text lines
- for s in l["spans"]: # iterate through the text spans
- print("")
- font_properties = "Font: '%s' (%s), size %g, color #%06x" % (
- s["font"], # font name
- flags_decomposer(s["flags"]), # readable font flags
- s["size"], # font size
- s["color"], # font color
- )
- print(s)
- print("Text: '%s'" % s["text"]) # simple print of text
- print(font_properties)
- get_underlined_textLines(page)
- # 以下为现成pdf单页解析接口
- class ParseSentence:
- def __init__(self, bbox, fontname, fontsize, _text, _title, title_text, _pattern, title_degree, is_outline,
- outline_location, page_no):
- (x0, y0, x1, y1) = bbox
- self.x0 = x0
- self.y0 = y0
- self.x1 = x1
- self.y1 = y1
- self.bbox = bbox
- self.fontname = fontname
- self.fontsize = fontsize
- self.text = _text
- self.title = _title
- self.title_text = title_text
- self.groups = _pattern
- self.title_degree = title_degree
- self.is_outline = is_outline
- self.outline_location = outline_location
- self.page_no = page_no
- def __repr__(self):
- return "%s,%s,%s,%d,%s" % (self.text, self.title, self.is_outline, self.outline_location, str(self.bbox))
- class ParseUtils:
- @staticmethod
- def getFontinfo(_page):
- for _obj in _page._objs:
- if isinstance(_obj, (LTTextBoxHorizontal, LTTextBoxVertical)):
- for textline in _obj._objs:
- done = False
- for lchar in textline._objs:
- if isinstance(lchar, (LTChar)):
- _obj.fontname = lchar.fontname
- _obj.fontsize = lchar.size
- done = True
- break
- if done:
- break
- @staticmethod
- def recognize_sentences(list_textbox, filter_objs, page_bbox, page_no,
- remove_space=True, sourceP_LB=True):
- list_textbox.sort(key=lambda x: x.bbox[0])
- list_textbox.sort(key=lambda x: x.bbox[3], reverse=sourceP_LB)
- cluster_textbox = []
- for _textbox in list_textbox:
- if _textbox in filter_objs:
- continue
- _find = False
- for _ct in cluster_textbox:
- if abs(_ct["y"] - _textbox.bbox[1]) < 5:
- _find = True
- _ct["textbox"].append(_textbox)
- if not _find:
- cluster_textbox.append({"y": _textbox.bbox[1], "textbox": [_textbox]})
- cluster_textbox.sort(key=lambda x: x["y"], reverse=sourceP_LB)
- list_sentences = []
- for _line in cluster_textbox:
- _textboxs = _line["textbox"]
- _textboxs.sort(key=lambda x: x.bbox[0])
- _linetext = _textboxs[0].get_text()
- for _i in range(1, len(_textboxs)):
- if abs(_textboxs[_i].bbox[0] - _textboxs[_i - 1].bbox[2]) > 60:
- if _linetext and _linetext[-1] not in (",", ",", "。", ".", "、", ";"):
- _linetext += "=,="
- _linetext += _textboxs[_i].get_text()
- _linetext = re.sub("[\s\r\n]", "", _linetext)
- _bbox = (_textboxs[0].bbox[0], _textboxs[0].bbox[1],
- _textboxs[-1].bbox[2], _textboxs[-1].bbox[3])
- _title = None
- _pattern_groups = None
- title_text = ""
- if not _title:
- _groups = ParseUtils.find_title_by_pattern(_textboxs[0].get_text())
- if _groups:
- _title = _groups[0][0]
- title_text = _groups[0][1]
- _pattern_groups = _groups
- if not _title:
- _groups = ParseUtils.find_title_by_pattern(_linetext)
- if _groups:
- _title = _groups[0][0]
- title_text = _groups[0][1]
- _pattern_groups = _groups
- if not _title:
- _title = ParseUtils.rec_incenter(_bbox, page_bbox)
- title_degree = 2
- if not _title:
- _linetext = _linetext.replace("=,=", ",")
- else:
- _linetext = _linetext.replace("=,=", "")
- title_degree = int(_title.split("_")[1])
- # 页码
- if ParseUtils.rec_incenter(_bbox, page_bbox) and re.search("^\d+$", _linetext) is not None:
- continue
- if _linetext == "" or re.search("^,+$", _linetext) is not None:
- continue
- is_outline = False
- outline_location = -1
- _search = re.search("(?P<text>.+?)\.{5,}(?P<nums>\d+)$", _linetext)
- if _search is not None:
- is_outline = True
- _linetext = _search.group("text")
- outline_location = int(_search.group("nums"))
- list_sentences.append(
- ParseSentence(_bbox, _textboxs[-1].__dict__.get("fontname"), _textboxs[-1].__dict__.get("fontsize"),
- _linetext, _title, title_text, _pattern_groups, title_degree, is_outline,
- outline_location, page_no))
- # for _sen in list_sentences:
- # print(_sen.__dict__)
- return list_sentences
- @staticmethod
- def find_title_by_pattern(_text,
- _pattern="(?P<title_1>(?P<title_1_index_0_0>^第?)(?P<title_1_index_1_1>[一二三四五六七八九十ⅠⅡⅢⅣⅤⅥⅦⅧⅨⅩⅪⅫ]+)(?P<title_1_index_2_0>[、章]))|" \
- "(?P<title_3>^(?P<title_3_index_0_1>[ⅠⅡⅢⅣⅤⅥⅦⅧⅨⅩⅪⅫ]+))|" \
- "(?P<title_4>^(?P<title_4_index_0_0>第?)(?P<title_4_index_1_1>[一二三四五六七八九十]+)(?P<title_4_index_2_0>[节]))|" \
- "(?P<title_11>^(?P<title_11_index_0_0>\d{1,2}[\..、\s\-]\d{1,2}[\..、\s\-]\d{1,2}[\..、\s\-])(?P<title_11_index_1_1>\d{1,2})(?P<title_11_index_2_0>[\..、\s\-]))|" \
- "(?P<title_10>^(?P<title_10_index_0_0>\d{1,2}[\..、\s\-]\d{1,2}[\..、\s\-])(?P<title_10_index_1_1>\d{1,2})(?P<title_10_index_2_0>[\..、\s\-]))|" \
- "(?P<title_7>^(?P<title_7_index_0_0>\d{1,2}[\..、\s\-])(?P<title_7_index_1_1>\d{1,2})(?P<title_7_index_2_0>[\..、\s\-]))|" \
- "(?P<title_6>^(?P<title_6_index_0_1>\d{1,2})(?P<title_6_index_1_0>[\..、\s\-]))|" \
- "(?P<title_15>^(?P<title_15_index_0_0>(?)(?P<title_15_index_1_1>\d{1,2})(?P<title_15_index_2_0>)))|" \
- "(?P<title_17>^(?P<title_17_index_0_0>(?)(?P<title_17_index_1_1>[a-zA-Z]+)(?P<title_17_index_2_0>)))|"
- "(?P<title_19>^(?P<title_19_index_0_0>(?)(?P<title_19_index_1_1>[一二三四五六七八九十ⅠⅡⅢⅣⅤⅥⅦⅧⅨⅩⅪⅫ]+)(?P<title_19_index_2_0>)))|" \
- ):
- _se = re.search(_pattern, _text)
- groups = []
- if _se is not None:
- _gd = _se.groupdict()
- for k, v in _gd.items():
- if v is not None:
- groups.append((k, v))
- if len(groups):
- groups.sort(key=lambda x: x[0])
- return groups
- return None
- @staticmethod
- def rec_incenter(o_bbox, p_bbox):
- p_width = p_bbox[2] - p_bbox[0]
- l_space = (o_bbox[0] - p_bbox[0]) / p_width
- r_space = (p_bbox[2] - o_bbox[2]) / p_width
- if abs((l_space - r_space)) < 0.1 and l_space > 0.2:
- return "title_2"
- @staticmethod
- def is_first_title(_title):
- if _title is None:
- return False
- if re.search("^\d+$", _title) is not None:
- if int(_title) == 1:
- return True
- return False
- if re.search("^[一二三四五六七八九十百]+$", _title) is not None:
- if _title == "一":
- return True
- return False
- if re.search("^[a-z]+$", _title) is not None:
- if _title == "a":
- return True
- return False
- if re.search("^[A-Z]+$", _title) is not None:
- if _title == "A":
- return True
- return False
- if re.search("^[ⅠⅡⅢⅣⅤⅥⅦⅧⅨⅩⅪⅫ]$", _title) is not None:
- if _title == "Ⅰ":
- return True
- return False
- return False
- @staticmethod
- def get_next_title(_title):
- if re.search("^\d+$", _title) is not None:
- return str(int(_title) + 1)
- if re.search("^[一二三四五六七八九十百]+$", _title) is not None:
- _next_title = ParseUtils.make_increase(['一', '二', '三', '四', '五', '六', '七', '八', '九', '十'],
- re.sub("[十百]", '', _title))
- _next_title = list(_next_title)
- _next_title.reverse()
- if _next_title[-1] != "十":
- if len(_next_title) >= 2:
- _next_title.insert(-1, '十')
- if len(_next_title) >= 4:
- _next_title.insert(-3, '百')
- if _title[0] == "十":
- if _next_title == "十":
- _next_title = ["二", "十"]
- _next_title.insert(0, "十")
- _next_title = "".join(_next_title)
- return _next_title
- if re.search("^[a-z]+$", _title) is not None:
- _next_title = ParseUtils.make_increase([chr(i + ord('a')) for i in range(26)], _title)
- _next_title = list(_next_title)
- _next_title.reverse()
- return "".join(_next_title)
- if re.search("^[A-Z]+$", _title) is not None:
- _next_title = ParseUtils.make_increase([chr(i + ord('A')) for i in range(26)], _title)
- _next_title = list(_next_title)
- _next_title.reverse()
- return "".join(_next_title)
- if re.search("^[ⅠⅡⅢⅣⅤⅥⅦⅧⅨⅩⅪⅫ]$", _title) is not None:
- _sort = ["Ⅰ", "Ⅱ", "Ⅲ", "Ⅳ", "Ⅴ", "Ⅵ", "Ⅶ", "Ⅷ", "Ⅸ", "Ⅹ", "Ⅺ", "Ⅻ"]
- _index = _sort.index(_title)
- if _index < len(_sort) - 1:
- return _sort[_index + 1]
- return None
- @staticmethod
- def make_increase(_sort, _title, _add=1):
- if len(_title) == 0 and _add == 0:
- return ""
- if len(_title) == 0 and _add == 1:
- return _sort[0]
- _index = _sort.index(_title[-1])
- next_index = (_index + _add) % len(_sort)
- next_chr = _sort[next_index]
- if _index == len(_sort) - 1:
- _add = 1
- else:
- _add = 0
- return next_chr + ParseUtils.make_increase(_sort, _title[:-1], _add)
- @staticmethod
- def rec_serial(_text, o_bbox, p_bbox, fontname, _pattern="(?P<title_1>^[一二三四五六七八九十]+[、])|" \
- "(?P<title_2>^\d+[\.、\s])|" \
- "(?P<title_3>^\d+\.\d+[\.、\s])|" \
- "(?P<title_4>^\d+\.\d+\.\d+[\.、\s])|" \
- "(?P<title_5>^\d+\.\d+\.\d+\.\d+[\.、\s])"):
- # todo :recog the serial of the sentence
- _se = re.search(_pattern, _text)
- if _se is not None:
- _gd = _se.groupdict()
- for k, v in _gd.items():
- if v is not None:
- return k
- return None
- if __name__ == '__main__':
- # get_text_font()
- PDFConvert(r"C:/Users/Administrator/Downloads/1651896704621.pdf", "C:/Users/Administrator/Downloads/1").get_html()
- # print(b'\x10')
|