45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from website_generator.fields import PathMatcherField, PathField
|
|
from website_generator.config import ActionConfig
|
|
from website_generator.actions import Action
|
|
from jinja2 import Environment, FilesystemLoader, select_autoescape
|
|
import typesystem
|
|
|
|
|
|
class JinjaActionConfig(ActionConfig):
|
|
template = PathMatcherField()
|
|
dest = PathField(allow_null=True, must_exist=False, allow_folders=True)
|
|
environment = typesystem.Object(additional_properties=True, default=dict)
|
|
|
|
|
|
class JinjaAction(Action):
|
|
type = 'jinja2'
|
|
|
|
def __init__(self, **params):
|
|
config = JinjaActionConfig.validate(params)
|
|
params['template'] = config.template
|
|
params['dest'] = config.dest
|
|
super().__init__(**params)
|
|
|
|
def build_environment(self):
|
|
"""
|
|
Build a Jinja2 environment for templates.
|
|
"""
|
|
return Environment(
|
|
loader=FilesystemLoader(self.main_config.content_root),
|
|
)
|
|
|
|
def render_template(self, env, template_path, dest_path):
|
|
template = env.get_template(template_path)
|
|
|
|
def __call__(self):
|
|
env = self.build_environment()
|
|
if self.dest.is_dir() or str(self.dest).endswith('/'):
|
|
self.dest.mkdir(parents=True, exist_ok=True)
|
|
for template_path in self.src.all():
|
|
self.render_template(env, template_path, self.dest / template_path.name)
|
|
else:
|
|
template_path =
|
|
template = env.get_template(template_path)
|
|
|
|
|