]>
Commit | Line | Data |
---|---|---|
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 | ||
8 | from argparse import ArgumentParser | |
9 | from contextlib import contextmanager | |
10 | import os.path | |
11 | ||
12 | from typing import Iterator | |
13 | ||
14 | from xdg_base_dirs import xdg_cache_home | |
15 | ||
16 | from domfilter import ApplyDOMFilters, DOMFilters | |
17 | from fetch import CachingFetcher | |
18 | from htmlfilter import ApplyHTMLFilters, HTMLFilters | |
19 | from images import DiskImageStore | |
20 | from spec import Spec | |
21 | from texify import PandocTexifier | |
22 | ||
23 | ||
24 | def _command_line_parser() -> ArgumentParser: | |
25 | parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic') | |
26 | parser.add_argument( | |
27 | '--cache_path', | |
28 | metavar='PATH', | |
29 | help='Where to keep the http cache (instead of %(default)s)', | |
30 | default=os.path.join(xdg_cache_home(), "paperdoorknob")) | |
31 | parser.add_argument( | |
32 | '--domfilters', | |
33 | help='Which DOM filters to use (default: %(default)s)', | |
34 | default=','.join(f[0] for f in DOMFilters)) | |
35 | parser.add_argument( | |
36 | '--geometry', | |
37 | help='''Page size and margin control | |
38 | See https://faculty.bard.edu/bloch/geometry.pdf for details | |
39 | (default: %(default)s)''', | |
40 | default='paperwidth=5.5in,paperheight=8.5in,nohead,' + | |
41 | 'tmargin=15mm,hmargin=15mm,bmargin=17mm,foot=4mm') | |
42 | parser.add_argument( | |
43 | '--htmlfilters', | |
44 | help='Which HTML filters to use (default: %(default)s)', | |
45 | default=','.join(f[0] for f in HTMLFilters)) | |
46 | parser.add_argument( | |
47 | '--out', | |
48 | help='The filename stem at which to write output ' + | |
49 | '(eg: "%(default)s" produces %(default)s.tex, %(default)s.pdf, etc.)', | |
50 | default='book') | |
51 | parser.add_argument('--pandoc', help='Location of the pandoc executable') | |
52 | parser.add_argument( | |
53 | '--timeout', | |
54 | help='How long to wait for HTTP requests, in seconds', | |
55 | default=30) | |
56 | parser.add_argument( | |
57 | 'url', | |
58 | help='URL to retrieve (example: https://www.projectlawful.com/posts/4582 )') | |
59 | return parser | |
60 | ||
61 | ||
62 | @contextmanager | |
63 | def spec_from_commandline_args() -> Iterator[Spec]: | |
64 | args = _command_line_parser().parse_args() | |
65 | with CachingFetcher(args.cache_path, args.timeout) as fetcher: | |
66 | with open(args.out + '.tex', 'wb') as texout: | |
67 | yield Spec( | |
68 | args.url, | |
69 | fetcher, | |
70 | DiskImageStore(args.out + '_images', fetcher), | |
71 | lambda x: ApplyHTMLFilters(args.htmlfilters, x), | |
72 | lambda x: ApplyDOMFilters(args.domfilters, x), | |
73 | PandocTexifier(args.pandoc or 'pandoc'), | |
74 | args.geometry, | |
75 | texout) |