我需要抓取一些 PDF 文件来提取以下文本信息:
我尝试使用 pdfquery 来完成此操作,方法是解决我在 Reddit 上找到的示例(请参阅第一篇文章):https://www.reddit.com/r/Python/comments/4bnjha/scraping_pdf_files_with_python/ https://www.reddit.com/r/Python/comments/4bnjha/scraping_pdf_files_with_python/
我想通过首先获取许可证号来测试它。我进入生成的“xmltree”文件,找到第一个许可证号并在 LTTextLineHorizontal 元素中获取 x0,y0,x1,y1 坐标。
import pdfquery
from lxml import etree
PDF_FILE = 'C:\\TEMP\\ad-4070-20-september-2018.pdf'
pdf = pdfquery.PDFQuery(PDF_FILE)
pdf.load(4,5)
with open('xmltree.xml','wb') as f:
f.write(etree.tostring(pdf.tree, pretty_print=True))
product_info = []
page_count = len(pdf._pages)
for pg in range(page_count):
data = pdf.extract([
('with_parent', 'LTPage[pageid="{}"]'.format(pg+1)),
('with_formatter', None),
('product_name', 'LTTextLineHorizontal:in_bbox("89.904, 757.502, 265.7, 770.83")'),
('product_details', 'LTTextLineHorizontal:in_bbox("223, 100, 737, 1114")'),
])
for ix, pn in enumerate(sorted([d for d in data['product_name'] if d.text.strip()], key=lambda x: x.get('y0'), reverse=True)):
product_info.append({'Manufacturer': pn.text.strip(), 'page': pg, 'y_start': float(pn.get('y1')), 'y_end': float(pn.get('y1'))-150})
# if this is not the first product on the page, update the previous product's y_end with a
# value slightly greater than this product's y coordinate start
if ix > 0:
product_info[-2]['y_end'] = float(pn.get('y0'))
# for every product found on this page, find the detail information that falls between the
# y coordinates belonging to the product
for product in [p for p in product_info if p['page'] == pg]:
details = []
for d in sorted([d for d in data['product_details'] if d.text.strip()], key=lambda x: x.get('y0'), reverse=True):
if product['y_start'] > float(d.get('y0')) > product['y_end']:
details.append(d.text.strip())
product['Details'] = ' '.join(details)
pdf.file.close()
for p in product_info:
print('Manufacturer: {}\r\nDetail Info:{}...\r\n\r\n'.format(p['Manufacturer'], p['Details'][0:100]))
但是,当我运行它时,它不会打印任何内容。没有错误,XML 文件生成得很好,而且我直接从 XML 文件获取坐标,所以应该没有问题。我究竟做错了什么?