]> git.scottworley.com Git - paperdoorknob/blobdiff - paperdoorknob.py
Use raw strings for less escaping
[paperdoorknob] / paperdoorknob.py
index 05b068f6c6993015df844e7a6b7b3a3f40dc7387..a19045cbfde8ba813d8f549650592ce27c746bf1 100644 (file)
@@ -22,23 +22,27 @@ def ilen(it: Iterable[Any]) -> int:
 
 
 def process(spec: Spec) -> None:
-    spec.texout.write(b'''\\documentclass{article}
-\\usepackage{booktabs}
-\\usepackage{graphicx}
-\\usepackage{longtable}
-\\usepackage{soul}
-\\usepackage{varwidth}
-\\usepackage{wrapstuff}
+    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}
-\\newcommand{\\href}[2]{#2\\footnote{#1}}
+    spec.texout.write(br'''\begin{document}
+\newcommand{\href}[2]{#2\footnote{\detokenize{#1}}}
 ''')
     url = flatURL(spec.url)
-    dom = parse(spec.htmlfilter(spec.fetcher.fetch(url)))
+    spec.log('Fetching HTML...\r')
+    html = spec.fetcher.fetch(url)
+    spec.log('Parsing HTML...\r')
+    dom = parse(spec.htmlfilter(html))
+    spec.log('Counting chunks...\r')
     num_chunks = ilen(chunkDOMs(dom))
     for i, r in enumerate(chunkDOMs(dom)):
         percent = 100.0 * i / num_chunks