mirror of
https://github.com/jsvine/pdfplumber.git
synced 2026-08-30 09:00:24 +08:00
Jeremy Singer-Vine
f8fbd572ce
- Should make equality comparisons more precise - PDF class now accepts a "precision" argument, used to quantize floats - This commit also adds some changes/improvements to JSON output
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
from pdfplumber.container import Container
|
|
from pdfplumber.page import Page
|
|
from pdfplumber.utils import decode_text
|
|
|
|
from pdfminer.pdfparser import PDFParser
|
|
from pdfminer.pdfdocument import PDFDocument
|
|
from pdfminer.pdfpage import PDFPage
|
|
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
|
|
from pdfminer.layout import LAParams
|
|
from pdfminer.converter import PDFPageAggregator
|
|
|
|
import atexit
|
|
|
|
class PDF(Container):
|
|
cached_properties = Container.cached_properties + [ "_pages" ]
|
|
|
|
def __init__(self, stream, pages=None, laparams=None, precision=0.001):
|
|
self.laparams = None if laparams == None else LAParams(**laparams)
|
|
self.stream = stream
|
|
self.pages_to_parse = pages
|
|
self.precision = precision
|
|
rsrcmgr = PDFResourceManager()
|
|
self.doc = PDFDocument(PDFParser(stream))
|
|
self.metadata = {}
|
|
for info in self.doc.info:
|
|
self.metadata.update(info)
|
|
for k, v in self.metadata.items():
|
|
if hasattr(v, "resolve"):
|
|
v = v.resolve()
|
|
self.metadata[k] = decode_text(v)
|
|
self.device = PDFPageAggregator(rsrcmgr, laparams=self.laparams)
|
|
self.interpreter = PDFPageInterpreter(rsrcmgr, self.device)
|
|
atexit.register(self.close)
|
|
|
|
def process_page(self, page):
|
|
self.interpreter.process_page(page)
|
|
return self.device.get_result()
|
|
|
|
@property
|
|
def pages(self):
|
|
if hasattr(self, "_pages"): return self._pages
|
|
|
|
doctop = 0
|
|
pp = self.pages_to_parse
|
|
self._pages = []
|
|
for i, page in enumerate(PDFPage.create_pages(self.doc)):
|
|
if pp != None and i+1 not in pp: continue
|
|
p = Page(self, page, initial_doctop=doctop)
|
|
self._pages.append(p)
|
|
doctop += p.height
|
|
return self._pages
|
|
|
|
def close(self):
|
|
self.stream.close()
|
|
|
|
@property
|
|
def objects(self):
|
|
if hasattr(self, "_objects"): return self._objects
|
|
all_objects = {}
|
|
for p in self.pages:
|
|
for kind in p.objects.keys():
|
|
all_objects[kind] = all_objects.get(kind, []) + p.objects[kind]
|
|
self._objects = all_objects
|
|
return self._objects
|