X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/ba3b7c52144a19a83892638415d2bfdb667888f5..372211762c57033dfef33075bd593ec5d426f3c7:/paperdoorknob.py?ds=sidebyside diff --git a/paperdoorknob.py b/paperdoorknob.py index 3aae538..4e7cdf6 100644 --- a/paperdoorknob.py +++ b/paperdoorknob.py @@ -4,38 +4,53 @@ # under the terms of the GNU General Public License as published by the # Free Software Foundation, version 3. - -from argparse import ArgumentParser -import os.path -import requests -import requests_cache -from xdg_base_dirs import xdg_cache_home - - -def command_line_parser() -> ArgumentParser: - parser = ArgumentParser(prog='paperdoorknob', description='Print glowfic') - parser.add_argument( - '--cache_path', - metavar='PATH', - help='Where to keep the http cache (instead of %(default)s)', - default=os.path.join(xdg_cache_home(), "paperdoorknob")) - parser.add_argument( - '--timeout', - help='How long to wait for HTTP requests, in seconds', - default=30) - parser.add_argument('url', help='URL to retrieve') - return parser - - -def fetch(url: str, session: requests.Session, timeout: int) -> None: - with session.get(url, timeout=timeout) as r: - r.raise_for_status() +from typing import Any, Iterable + +from bs4 import BeautifulSoup + +from args import spec_from_commandline_args +from glowfic import chunkDOMs, flatURL, makeChunk +from spec import Spec + + +def parse(content: bytes) -> BeautifulSoup: + return BeautifulSoup(content, 'html.parser') + + +def ilen(it: Iterable[Any]) -> int: + return sum(1 for _ in it) + + +def process(spec: Spec) -> None: + spec.texout.write(b'''\\documentclass{article} +\\usepackage{booktabs} +\\usepackage{graphicx} +\\usepackage{longtable} +\\usepackage{soul} +\\usepackage{varwidth} +\\usepackage{wrapstuff} +''') + if spec.geometry is not None: + spec.texout.write(b'\\usepackage[' + + spec.geometry.encode('UTF-8') + + b']{geometry}\n') + spec.texout.write(b'\\begin{document}\n') + url = flatURL(spec.url) + html = parse(spec.htmlfilter(spec.fetcher.fetch(url))) + num_chunks = ilen(chunkDOMs(html)) + for i, r in enumerate(chunkDOMs(html)): + percent = 100.0 * i / num_chunks + spec.log(f'Processing chunk {i} of {num_chunks} ({percent:.1f}%)\r') + spec.domfilter(r) + chunk = makeChunk(r, spec.images) + spec.texout.write(spec.texfilter(spec.layout.renderChunk(chunk))) + spec.log('') + spec.texout.write(b'\\end{document}\n') def main() -> None: - args = command_line_parser().parse_args() - with requests_cache.CachedSession(args.cache_path) as session: - fetch(args.url, session, args.timeout) + with spec_from_commandline_args() as spec: + process(spec) if __name__ == '__main__':