1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
import sys
from bs4 import BeautifulSoup
import djvu
from djvu.decode import Context
from itertools import chain
import collections
from PIL import Image
def parse_page(page, html=False):
s, page_size = page.text.sexpr, page.size[1]
def aux(s, html):
if type(s) is djvu.sexpr.ListExpression:
if len(s) == 0:
pass
if str(s[0].value) == "word":
if html:
coords = (s[1].value, page_size - s[2].value,
s[3].value, page_size - s[4].value)
coords = ",".join(map(str,coords))
else:
coords = [s[i].value for i in xrange(1, 5)]
word = s[5].value
yield (word, coords)
else:
for c in chain.from_iterable(aux(child, html) for child in s[5:]):
yield c
else:
pass
return aux(s, html) if s else None
def get_document(djvufile):
c = Context()
document = c.new_document(djvu.decode.FileURI(djvufile))
document.decoding_job.wait()
return document
def parse_book(djvubook, page=None, html=False):
"""
returns the list of words and coordinates from a djvu book.
if page is None, returns the whole book.
if html is True, coordinates are computed from the bottom of the page
"""
document = get_document(djvubook)
if type(page) is int:
toparse = [document.pages[page - 1]]
elif isinstance(page, collections.Iterable):
toparse = [document.pages[p - 1] for p in page]
else:
toparse = document.pages
return [parse_page(page, html=html) for page in toparse]
def image_from_book(djvubook, page):
document = get_document(djvubook)
mode = djvu.decode.RENDER_COLOR
djvu_pixel_format = djvu.decode.PixelFormatRgb()
page = document.pages[page-1]
page_job = page.decode(wait=True)
width, height = page_job.size
rect = (0, 0, width, height)
buf = page_job.render(mode, rect, rect, djvu_pixel_format)
return Image.frombuffer("RGB", (width, height), buf, 'raw', 'RGB', 0, -1)
if __name__ == "__main__":
book = parse_book(sys.argv[1], page=[10,11], html=True)
im = image_from_book(sys.argv[1], 11)
im.save("test.webp")
|
