From 7b4b681229f07e431e0945b41b956a28fa1258e4 Mon Sep 17 00:00:00 2001 From: Scott Worley Date: Fri, 5 Jan 2024 17:24:46 -0800 Subject: [PATCH] Show name of thread being processed --- paperdoorknob.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/paperdoorknob.py b/paperdoorknob.py index 86b8766..1350784 100644 --- a/paperdoorknob.py +++ b/paperdoorknob.py @@ -7,6 +7,7 @@ from typing import Any, Iterable from bs4 import BeautifulSoup +from bs4.element import Tag from args import spec_from_commandline_args from glowfic import chunkDOMs, flatURL, makeChunk, renderChunk @@ -21,6 +22,13 @@ def ilen(it: Iterable[Any]) -> int: return sum(1 for _ in it) +def get_title(dom: BeautifulSoup) -> str | None: + span = dom.findChild("span", id="post-title") + if not isinstance(span, Tag): + return None + return span.text + + def process(spec: Spec) -> None: spec.texout.write(br'''\documentclass{article} \usepackage{booktabs} @@ -51,9 +59,10 @@ def process(spec: Spec) -> None: dom = parse(spec.htmlfilter(html)) spec.log('Counting chunks...\r') num_chunks = ilen(chunkDOMs(dom)) + title = get_title(dom) or "chunk" for i, r in enumerate(chunkDOMs(dom)): percent = 100.0 * i / num_chunks - spec.log(f'Processing chunk {i} of {num_chunks} ({percent:.1f}%)\r') + spec.log(f'Processing {title} {i} of {num_chunks} ({percent:.1f}%)\r') spec.domfilter(r) chunk = makeChunk(r, spec.images) spec.texout.write(spec.texfilter(renderChunk(spec.texifier, chunk))) -- 2.44.1