feature: download svgs instead of building them
This commit is contained in:
+10
-40
@@ -1,9 +1,5 @@
|
||||
import json
|
||||
import multiprocessing
|
||||
import os
|
||||
import shutil
|
||||
|
||||
from joblib import Memory, Parallel, delayed
|
||||
from tqdm import tqdm
|
||||
|
||||
from api.directus import DirectusAPI
|
||||
@@ -97,50 +93,24 @@ class Download:
|
||||
|
||||
for photo in tqdm(photos, desc="Downloading photos"):
|
||||
data = self.api.get_file("assets/" + photo["photo"])
|
||||
with (self.config.p_data_photos / f"{photo['id']}.jpg").open("wb") as file:
|
||||
with (self.config.p_data_photos / f"{photo["id"]}.jpg").open("wb") as file:
|
||||
file.write(data)
|
||||
|
||||
def download_pictures(self):
|
||||
self.config.p_data_pictures.mkdir(parents=True, exist_ok=True)
|
||||
# Symlink photos, so LaTeX can access these assets to render into graphics.
|
||||
foto_link = self.config.p_data_pictures / "foto"
|
||||
if not foto_link.is_symlink():
|
||||
foto_link.symlink_to(self.config.p_data_photos)
|
||||
|
||||
memory = Memory("./cache", verbose=0)
|
||||
pictures = self.api.get("items/pictures", { "limit": -1 })
|
||||
|
||||
@memory.cache
|
||||
def build_picture(picture):
|
||||
pid = picture["id"]
|
||||
for picture in tqdm(pictures, desc="Downloading pictures") :
|
||||
|
||||
if (
|
||||
"\\begin{document}" in picture["latex"] or "\\begin{table}" in picture["latex"]
|
||||
): # Workaround, if the picture contains a document class it should be ignored!
|
||||
return
|
||||
# Download LaTeX Sources:
|
||||
with (self.config.p_data_pictures / f"{picture["id"]}include.tex").open("w") as file:
|
||||
file.write(picture["latex"])
|
||||
|
||||
if self.config.no_latex:
|
||||
shutil.copyfile("./assets/images/50ohm_gray.svg", self.config.p_data_pictures / f"{pid}.svg")
|
||||
else:
|
||||
with (self.config.p_data_pictures / f"img/{pid}include.tex").open("w") as file:
|
||||
file.write(picture["latex"])
|
||||
|
||||
aux_file = self.config.p_data_photos / f"{pid}.tex"
|
||||
with aux_file.open("w", encoding="utf-8") as file:
|
||||
# Write the auxilary LaTeX file:
|
||||
file.write(f"\\documentclass{{BNetzA-Fragenkatalog}}\\DARCimageOnly{{9cm}}{{{pid}include}}")
|
||||
# Build the picture:
|
||||
os.system(f"latexmk -lualatex -f -interaction=nonstopmode {aux_file} > /dev/null 2>&1")
|
||||
# Convert the PDF to SVG:
|
||||
os.system(
|
||||
"pdftocairo -svg"
|
||||
f"{self.config.p_data_pictures / f'{pid}.pdf'} - > {self.config.p_data_pictures / f'{pid}.svg'}"
|
||||
)
|
||||
|
||||
pictures = self.api.get("items/pictures", params={"limit": -1})
|
||||
|
||||
Parallel(n_jobs=multiprocessing.cpu_count())(
|
||||
delayed(build_picture)(picture) for picture in tqdm(pictures, desc="Build All Pictures")
|
||||
)
|
||||
# Download SVGs:
|
||||
data = self.api.get_file(f"assets/{picture["picture"]}")
|
||||
with (self.config.p_data_pictures / f"{picture["id"]}.svg").open("wb") as file:
|
||||
file.write(data)
|
||||
|
||||
def download_includes(self):
|
||||
includes = self.api.get("items/include", params={"limit": -1})
|
||||
|
||||
Reference in New Issue
Block a user