Progress indicator

[paperdoorknob] / paperdoorknob.py

# paperdoorknob: Print glowfic
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation, version 3.

from sys import stderr

from typing import Any, Iterable

from bs4 import BeautifulSoup

from args import spec_from_commandline_args
from glowfic import chunkDOMs, flatURL, makeChunk
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}
\\usepackage{booktabs}
\\usepackage{graphicx}
\\usepackage{longtable}
\\usepackage{soul}
\\usepackage{varwidth}
\\usepackage{wrapstuff}
''')
    if spec.geometry is not None:
        spec.texout.write(b'\\usepackage[' +
                          spec.geometry.encode('UTF-8') +
                          b']{geometry}\n')
    spec.texout.write(b'\\begin{document}\n')
    url = flatURL(spec.url)
    html = parse(spec.htmlfilter(spec.fetcher.fetch(url)))
    num_chunks = ilen(chunkDOMs(html))
    for i, r in enumerate(chunkDOMs(html)):
        percent = 100.0 * i / num_chunks
        print(
            f'Processing chunk {i} of {num_chunks} ({percent:.1f}%)',
            end='\r',
            file=stderr)
        spec.domfilter(r)
        chunk = makeChunk(r, spec.images)
        spec.texout.write(spec.texfilter(spec.layout.renderChunk(chunk)))
    spec.texout.write(b'\\end{document}\n')


def main() -> None:
    with spec_from_commandline_args() as spec:
        process(spec)


if __name__ == '__main__':
    main()