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
+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 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}
\detokenize{#1}\relax\else #2\fi}
%s
''' % (spec.icon_size, spec.layout))
- url = flatURL(spec.url)
- spec.log('Fetching HTML...\r')
- html = spec.fetcher.fetch(url)
- spec.log('Parsing HTML...\r')
- dom = parse(spec.htmlfilter(html))
+ thread = Thread(spec)
spec.log('Counting chunks...\r')
- num_chunks = ilen(chunkDOMs(dom))
- title = get_title(dom) or "chunk"
- for i, r in enumerate(chunkDOMs(dom)):
+ 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)