]> git.scottworley.com Git - paperdoorknob/blob - paperdoorknob.py
86b8766e659601b6d05a1c2dcd28e6c7f2fbb868
[paperdoorknob] / paperdoorknob.py
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
7 from typing import Any, Iterable
8
9 from bs4 import BeautifulSoup
10
11 from args import spec_from_commandline_args
12 from glowfic import chunkDOMs, flatURL, makeChunk, renderChunk
13 from spec import Spec
14
15
16 def parse(content: bytes) -> BeautifulSoup:
17 return BeautifulSoup(content, 'html.parser')
18
19
20 def ilen(it: Iterable[Any]) -> int:
21 return sum(1 for _ in it)
22
23
24 def process(spec: Spec) -> None:
25 spec.texout.write(br'''\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(br'\usepackage[' +
35 spec.geometry.encode('UTF-8') +
36 b']{geometry}\n')
37 spec.texout.write(br'''\begin{document}
38 \newcommand{\href}[2]{#2\footnote{\detokenize{#1}}}
39 \def\glowiconsize{%fmm}
40 \newcommand{\glowicon}[1]{\includegraphics[
41 width=\glowiconsize,height=\glowiconsize,keepaspectratio
42 ]{#1}}
43 \newcommand{\ifnotempty}[2]{\expandafter\ifx\expandafter\relax
44 \detokenize{#1}\relax\else #2\fi}
45 %s
46 ''' % (spec.icon_size, spec.layout))
47 url = flatURL(spec.url)
48 spec.log('Fetching HTML...\r')
49 html = spec.fetcher.fetch(url)
50 spec.log('Parsing HTML...\r')
51 dom = parse(spec.htmlfilter(html))
52 spec.log('Counting chunks...\r')
53 num_chunks = ilen(chunkDOMs(dom))
54 for i, r in enumerate(chunkDOMs(dom)):
55 percent = 100.0 * i / num_chunks
56 spec.log(f'Processing chunk {i} of {num_chunks} ({percent:.1f}%)\r')
57 spec.domfilter(r)
58 chunk = makeChunk(r, spec.images)
59 spec.texout.write(spec.texfilter(renderChunk(spec.texifier, chunk)))
60 spec.log('')
61 spec.texout.write(b'\\end{document}\n')
62
63
64 def main() -> None:
65 with spec_from_commandline_args() as spec:
66 process(spec)
67
68
69 if __name__ == '__main__':
70 main()