git.scottworley.com Git - paperdoorknob/blob - 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
   8 from bs4 import BeautifulSoup
   9
  10 from args import spec_from_commandline_args
  11 from glowfic import chunkDOMs
  12 from spec import Spec
  13
  14
  15 def parse(content: bytes) -> BeautifulSoup:
  16     return BeautifulSoup(content, 'html.parser')
  17
  18
  19 def process(spec: Spec) -> None:
  20     spec.texout.write(b'\\documentclass{article}\n\\usepackage{wrapfig}\n')
  21     if spec.geometry is not None:
  22         spec.texout.write(b'\\usepackage[' +
  23                           spec.geometry.encode('UTF-8') +
  24                           b']{geometry}\n')
  25     spec.texout.write(b'\\begin{document}\n')
  26     html = parse(spec.htmlfilter(spec.fetcher.fetch(spec.url)))
  27     for r in chunkDOMs(html):
  28         spec.domfilter(r)
  29         spec.texout.write(spec.texifier.texify(r))
  30     spec.texout.write(b'\\end{document}\n')
  31
  32
  33 def main() -> None:
  34     with spec_from_commandline_args() as spec:
  35         process(spec)
  36
  37
  38 if __name__ == '__main__':
  39     main()