X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/8d583d5bdb7c8c2c6f7dfc3336c48a9ac04393ca..19ed28f37b59e480cebe8f122e07ce18ce536db8:/paperdoorknob.py?ds=sidebyside diff --git a/paperdoorknob.py b/paperdoorknob.py index 160f0db..d545f61 100644 --- a/paperdoorknob.py +++ b/paperdoorknob.py @@ -4,29 +4,51 @@ # under the terms of the GNU General Public License as published by the # Free Software Foundation, version 3. - -from bs4 import BeautifulSoup +from typing import Any, Iterable from args import spec_from_commandline_args -from glowfic import chunkDOMs +from glowfic import makeChunk, renderChunk, Thread 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}\n\\usepackage{wrapfig}\n') + spec.texout.write(br'''\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.texout.write(br'\usepackage[' + spec.geometry.encode('UTF-8') + b']{geometry}\n') - spec.texout.write(b'\\begin{document}\n') - html = parse(spec.htmlfilter(spec.fetcher.fetch(spec.url))) - for r in chunkDOMs(html): + spec.texout.write(br'''\begin{document} +\newcommand{\href}[2]{#2\footnote{\detokenize{#1}}} +\def\glowiconsize{%fmm} +\newcommand{\glowicon}[1]{\includegraphics[ + width=\glowiconsize,height=\glowiconsize,keepaspectratio +]{#1}} +\newcommand{\ifnotempty}[2]{\expandafter\ifx\expandafter\relax + \detokenize{#1}\relax\else #2\fi} +%s +''' % (spec.icon_size, spec.layout)) + thread = Thread(spec) + spec.log('Counting chunks...\r') + num_chunks = ilen(thread.chunkDOMs()) + title = thread.title() or "chunk" + for i, r in enumerate(thread.chunkDOMs()): + percent = 100.0 * i / num_chunks + spec.log(f'Processing {title} {i} of {num_chunks} ({percent:.1f}%)\r') spec.domfilter(r) - spec.texout.write(spec.texifier.texify(r)) + chunk = makeChunk(r, spec.images) + spec.texout.write(spec.texfilter(renderChunk(spec.texifier, chunk))) + spec.log('') spec.texout.write(b'\\end{document}\n')