FakeImageStore for tests

[paperdoorknob] / args.py

diff --git a/args.py b/args.py
index fb36083b28b9bcf852d37963f8b53c33760819d3..f966ba92ddcd0d6f2044556d9323d2508c296f43 100644 (file)
--- a/args.py
+++ b/args.py
@@ -16,6 +16,7 @@ from xdg_base_dirs import xdg_cache_home
 from domfilter import ApplyDOMFilters, DOMFilters
 from fetch import CachingFetcher
 from htmlfilter import ApplyHTMLFilters, HTMLFilters
+from images import DiskImageStore
 from spec import Spec
 from texify import PandocTexifier
 
@@ -31,6 +32,13 @@ def _command_line_parser() -> ArgumentParser:
         '--domfilters',
         help='Which DOM filters to use (default: %(default)s)',
         default=','.join(f[0] for f in DOMFilters))
+    parser.add_argument(
+        '--geometry',
+        help='''Page size and margin control
+See https://faculty.bard.edu/bloch/geometry.pdf for details
+(default: %(default)s)''',
+        default='paperwidth=5.5in,paperheight=8.5in,nohead,' +
+        'tmargin=15mm,hmargin=15mm,bmargin=17mm,foot=4mm')
     parser.add_argument(
         '--htmlfilters',
         help='Which HTML filters to use (default: %(default)s)',
@@ -59,7 +67,9 @@ def spec_from_commandline_args() -> Iterator[Spec]:
             yield Spec(
                 args.url,
                 fetcher,
+                DiskImageStore(args.out + '_images', fetcher),
                 lambda x: ApplyHTMLFilters(args.htmlfilters, x),
                 lambda x: ApplyDOMFilters(args.domfilters, x),
                 PandocTexifier(args.pandoc or 'pandoc'),
+                args.geometry,
                 texout)