mirror of
https://github.com/jsvine/pdfplumber.git
synced 2026-08-29 16:40:24 +08:00
Jeremy Singer-Vine
100 lines
2.7 KiB
Python
Executable File
100 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import pdfplumber
|
|
import argparse
|
|
from itertools import chain
|
|
from decimal import Decimal, ROUND_HALF_UP
|
|
import unicodecsv
|
|
import codecs
|
|
import json
|
|
import sys
|
|
|
|
class DecimalEncoder(json.JSONEncoder):
|
|
def default(self, o):
|
|
if isinstance(o, Decimal):
|
|
return float(o.quantize(Decimal('.0001'), rounding=ROUND_HALF_UP))
|
|
return super(DecimalEncoder, self).default(o)
|
|
|
|
def parse_page_spec(p_str):
|
|
if "-" in p_str:
|
|
return list(range(*map(int, p_str.split("-"))))
|
|
else: return [ int(p_str) ]
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser("pdfplumber")
|
|
|
|
stdin = (sys.stdin.buffer if sys.version_info[0] >= 3 else sys.stdin)
|
|
parser.add_argument("infile", nargs="?",
|
|
type=argparse.FileType("rb"),
|
|
default=stdin)
|
|
|
|
parser.add_argument("--format",
|
|
choices=["csv", "json"],
|
|
default="csv")
|
|
|
|
parser.add_argument("--encoding",
|
|
default="utf-8")
|
|
|
|
parser.add_argument("--types", nargs="+",
|
|
choices=[ "char", "anno", "line", "rect", "rect_edge" ],
|
|
default=[ "char", "anno", "line", "rect" ])
|
|
|
|
parser.add_argument("--pages", nargs="+",
|
|
type=parse_page_spec)
|
|
|
|
args = parser.parse_args()
|
|
if args.pages != None:
|
|
args.pages = list(chain(*args.pages))
|
|
return args
|
|
|
|
def to_csv(pdf, types, encoding):
|
|
objs = []
|
|
fields = set()
|
|
for t in types:
|
|
new_objs = getattr(pdf, t + "s")
|
|
if len(new_objs):
|
|
objs += new_objs
|
|
fields = fields.union(set(new_objs[0].keys()))
|
|
|
|
first_columns = [
|
|
"object_type", "pageid",
|
|
"x0", "x1", "y0", "y1",
|
|
"doctop", "top", "bottom",
|
|
"width", "height"
|
|
]
|
|
|
|
cols = first_columns + list(sorted(set(fields) - set(first_columns)))
|
|
|
|
w = unicodecsv.DictWriter(sys.stdout.buffer,
|
|
fieldnames=cols, encoding=encoding)
|
|
w.writeheader()
|
|
w.writerows(objs)
|
|
|
|
def to_json(pdf, types, encoding):
|
|
data = { "metadata": pdf.metadata }
|
|
|
|
def get_page_data(page):
|
|
d = dict((t + "s", getattr(page, t + "s"))
|
|
for t in types)
|
|
d["width"] = page.width
|
|
d["height"] = page.height
|
|
return d
|
|
|
|
data["pages"] = list(map(get_page_data, pdf.pages))
|
|
|
|
if hasattr(sys.stdout, "buffer"):
|
|
sys.stdout = codecs.getwriter("utf-8")(sys.stdout.buffer, "strict")
|
|
json.dump(data, sys.stdout, cls=DecimalEncoder)
|
|
else:
|
|
json.dump(data, sys.stdout, cls=DecimalEncoder, encoding=encoding)
|
|
|
|
def main():
|
|
args = parse_args()
|
|
pdf = pdfplumber.load(args.infile, pages=args.pages)
|
|
if args.format == "csv":
|
|
to_csv(pdf, args.types, args.encoding)
|
|
else:
|
|
to_json(pdf, args.types, args.encoding)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|