]> git.scottworley.com Git - paperdoorknob/blame_incremental - paperdoorknob.py
Change default layout: below → beside
[paperdoorknob] / paperdoorknob.py
... / ...
CommitLineData
1# paperdoorknob: Print glowfic
2#
3# This program is free software: you can redistribute it and/or modify it
4# under the terms of the GNU General Public License as published by the
5# Free Software Foundation, version 3.
6
7from typing import Any, Iterable
8
9from bs4 import BeautifulSoup
10
11from args import spec_from_commandline_args
12from glowfic import chunkDOMs, flatURL, makeChunk
13from spec import Spec
14
15
16def parse(content: bytes) -> BeautifulSoup:
17 return BeautifulSoup(content, 'html.parser')
18
19
20def ilen(it: Iterable[Any]) -> int:
21 return sum(1 for _ in it)
22
23
24def process(spec: Spec) -> None:
25 spec.texout.write(b'''\\documentclass{article}
26\\usepackage{booktabs}
27\\usepackage{graphicx}
28\\usepackage{longtable}
29\\usepackage{soul}
30\\usepackage{varwidth}
31\\usepackage{wrapstuff}
32''')
33 if spec.geometry is not None:
34 spec.texout.write(b'\\usepackage[' +
35 spec.geometry.encode('UTF-8') +
36 b']{geometry}\n')
37 spec.texout.write(b'\\begin{document}\n')
38 url = flatURL(spec.url)
39 html = parse(spec.htmlfilter(spec.fetcher.fetch(url)))
40 num_chunks = ilen(chunkDOMs(html))
41 for i, r in enumerate(chunkDOMs(html)):
42 percent = 100.0 * i / num_chunks
43 spec.log(f'Processing chunk {i} of {num_chunks} ({percent:.1f}%)\r')
44 spec.domfilter(r)
45 chunk = makeChunk(r, spec.images)
46 spec.texout.write(spec.texfilter(spec.layout.renderChunk(chunk)))
47 spec.log('')
48 spec.texout.write(b'\\end{document}\n')
49
50
51def main() -> None:
52 with spec_from_commandline_args() as spec:
53 process(spec)
54
55
56if __name__ == '__main__':
57 main()