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