X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/929db57622c6b8756ed341d7ae2862126a7c638f..refs/heads/main:/paperdoorknob.py?ds=inline diff --git a/paperdoorknob.py b/paperdoorknob.py index 34223e4..5ad341f 100644 --- a/paperdoorknob.py +++ b/paperdoorknob.py @@ -4,51 +4,36 @@ # under the terms of the GNU General Public License as published by the # Free Software Foundation, version 3. - -import itertools - -from typing import Iterable - -from bs4 import BeautifulSoup -from bs4.element import Tag - from args import spec_from_commandline_args +from glowfic import Thread from spec import Spec -def parse(content: bytes) -> BeautifulSoup: - return BeautifulSoup(content, 'html.parser') - - -def clean(html: BeautifulSoup) -> BeautifulSoup: - for eb in html.find_all("div", class_="post-edit-box"): - eb.decompose() - for footer in html.find_all("div", class_="post-footer"): - footer.decompose() - return html - - -def replies(html: BeautifulSoup) -> Iterable[Tag]: - def text() -> Tag: - body = html.body - assert body - text = body.find_next("div", class_="post-post") - assert isinstance(text, Tag) - return text - - def the_replies() -> Iterable[Tag]: - rs = html.find_all("div", class_="post-reply") - assert all(isinstance(r, Tag) for r in rs) - return rs - - return itertools.chain([text()], the_replies()) - - def process(spec: Spec) -> None: - spec.texout.write(b'\\documentclass{article}\n\\begin{document}\n') - html = clean(parse(spec.htmlfilter(spec.fetcher.fetch(spec.url)))) - for r in replies(html): - spec.texout.write(spec.texifier.texify(r)) + spec.texout.write(br'''\documentclass{article} +\usepackage{amssymb} +\usepackage{booktabs} +\usepackage{graphicx} +\usepackage{longtable} +\usepackage{soul} +\usepackage{varwidth} +\usepackage{wrapstuff} +''') + if spec.geometry is not None: + spec.texout.write(br'\usepackage[' + + spec.geometry.encode('UTF-8') + + b']{geometry}\n') + 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(spec).emit() spec.texout.write(b'\\end{document}\n')