Source code for pychum.engine.eon._renderer

import os

from jinja2 import Environment, FileSystemLoader

from pychum.engine.eon._dataclasses import NWChemSocketConfig



[docs]
class NWChemRenderer:
    def __init__(self, config: NWChemSocketConfig):

[docs]
        self.config = config

        package_path = os.path.dirname(os.path.abspath(__file__))
        template_dir = os.path.join(package_path, "_blocks")

[docs]
        self.env = Environment(loader=FileSystemLoader(template_dir), autoescape=True)

        self.env.trim_blocks = True
        self.env.lstrip_blocks = True


[docs]
    def render(self, template_name: str):
        template = self.env.get_template(template_name)
        context = {
            "config": self.config,
        }
        return template.render(context)