feat: create generate worker
This commit is contained in:
parent
08411bd42d
commit
76a033cf43
|
@ -0,0 +1,13 @@
|
|||
from jinja2.environment import Template
|
||||
|
||||
|
||||
def generate(args, deps, output):
|
||||
env = args["jinja2"]["environment"]
|
||||
template = env.get_template(deps[0])
|
||||
with open(output, "w") as out:
|
||||
out.write(tpl2tex(template, metas=args))
|
||||
yield f"GENERATE - {deps[0]} to {output}"
|
||||
|
||||
|
||||
def tpl2tex(template: Template, metas: dict = {}) -> str:
|
||||
return template.render(metas)
|
|
@ -1,28 +0,0 @@
|
|||
""" A worker consumes tasks """
|
||||
def generate():
|
||||
pass
|
||||
|
||||
|
||||
def compile():
|
||||
pass
|
||||
|
||||
|
||||
def activate_corr():
|
||||
pass
|
||||
|
||||
|
||||
def join_pdf():
|
||||
pass
|
||||
|
||||
|
||||
def clean():
|
||||
pass
|
||||
|
||||
|
||||
WORKERS = {
|
||||
"GENERATE": generate,
|
||||
"COMPILE": compile,
|
||||
"ACTIVATE_CORR": activate_corr,
|
||||
"JOIN_PDF": join_pdf,
|
||||
"CLEAN": clean,
|
||||
}
|
|
@ -0,0 +1,45 @@
|
|||
import os
|
||||
import jinja2
|
||||
from pathlib import Path
|
||||
from bopytex.worker.generate import tpl2tex, generate
|
||||
import pytest
|
||||
|
||||
|
||||
def test_tpl2tex():
|
||||
tpl = "Plop {{a}}"
|
||||
jinja2_tpl = jinja2.Template(tpl)
|
||||
fed = tpl2tex(jinja2_tpl, metas={"a": 1})
|
||||
assert fed == "Plop 1"
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def template_path(tmp_path):
|
||||
template = tmp_path / "template.j2"
|
||||
with open(template, "w") as tpl:
|
||||
tpl.write("Plop {{ a }}")
|
||||
return template
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def jinja2_env(tmp_path):
|
||||
templateEnv = jinja2.Environment(loader=jinja2.FileSystemLoader(tmp_path))
|
||||
return templateEnv
|
||||
|
||||
|
||||
def test_generate(template_path, jinja2_env):
|
||||
tmp_path = template_path.parent
|
||||
os.chdir(tmp_path)
|
||||
|
||||
assert template_path.exists
|
||||
template = str(template_path.name)
|
||||
output = "output"
|
||||
|
||||
result = next(generate(
|
||||
args={"a": 2, "jinja2": {"environment": jinja2_env}},
|
||||
deps=[template],
|
||||
output=output,
|
||||
))
|
||||
assert result == "GENERATE - template.j2 to output"
|
||||
with open(output, "r") as out:
|
||||
lines = out.readlines()
|
||||
assert lines == ["Plop 2"]
|
Loading…
Reference in New Issue