diff options
| -rw-r--r-- | parsepdftext.py | 26 |
1 files changed, 17 insertions, 9 deletions
diff --git a/parsepdftext.py b/parsepdftext.py index 51e237c..8521d7a 100644 --- a/parsepdftext.py +++ b/parsepdftext.py @@ -1,12 +1,20 @@ import sys from xml.etree import ElementTree as ET -document = ET.parse(sys.argv[1]) -ns = 'http://www.w3.org/1999/xhtml' -corners = ['xMin', 'xMax', 'yMin', 'yMax'] -# coordinates are in dpi, and computed from the top left corner -words = [] -coords = [] -for i, page in enumerate(document.findall('.//{{{0}}}page'.format(ns))): - words.append([word.text for word in page.getchildren()]) - coords.append([tuple([float(word.attrib[c]) for c in corners]) for word in page.getchildren()]) +def parse_coords(word): + # coordinates are in dpi, and computed from the top left corner + return tuple([word.attrib[c] for c in ['xMin', 'xMax', 'yMin', 'yMax']]) + +def parse_book(book): + document = ET.parse(book) + ns = 'http://www.w3.org/1999/xhtml' + + words = [] + coords = [] + for i, page in enumerate(document.findall('.//{{{0}}}page'.format(ns))): + words.append([word.text for word in page.getchildren()]) + coords.append([parse_coords(word) for word in page.getchildren()]) + return {"words": words, "coords": coords} + +if __name__=="__main__": + book = parse_book(sys.argv[1]) |