esphome-docs/svg2png.py

61 lines
1.1 KiB
Python

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()