esphome-docs/svg2png.py

from pathlib import Path
import re
import shlex
import subprocess
import threading
import queue
import sys


to_p = Path("svg2png")
to_p.mkdir(exist_ok=True)

images = [
    f
    for f in Path("_build/html/_images/").glob("*.svg")
    if not re.match(r"^seg[0-9A-F]{2}$", f.stem)
]
q = queue.Queue()


def worker():
    while True:
        item = q.get()
        if item is None:
            break

        to = to_p / item.with_suffix(".png").name
        if to.is_file():
            q.task_done()
            continue
        args = [
            "inkscape",
            f"--export-filename={to.absolute()}",
            "-w",
            "800",
            "--export-background=white",
            str(item.absolute()),
        ]
        print("Running:  {}".format(" ".join(shlex.quote(x) for x in args)))
        subprocess.check_call(args)

        q.task_done()


NUM_THREADS = 8
threads = []
for i in range(NUM_THREADS):
    t = threading.Thread(target=worker)
    t.start()
    threads.append(t)

for img in sorted(images):
    q.put(img)

q.join()

for i in range(NUM_THREADS):
    q.put(None)
for t in threads:
    t.join()