template_parser.py

import re
from datetime import datetime

from jinja2 import Environment, FileSystemLoader, Template

from common import project_root

continuous_bl: re.Pattern = re.compile(r"\n\n+")
template_dir = project_root.path("wizard_common/resources/prompt_templates")

env = Environment(loader=FileSystemLoader(template_dir))


class TemplateParser:
    def __init__(self, base_dir: str):
        self.base_dir: str = base_dir
        self.env: Environment = Environment(loader=FileSystemLoader(self.base_dir))

    def get_template(self, template_name: str) -> Template:
        return self.env.get_template(template_name)

    @staticmethod
    def render_template(template: Template, now: str = None, **kwargs) -> str:
        now: str = now or datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        render_result = template.render({**kwargs, "now": now})
        render_result = continuous_bl.sub("\n\n", render_result.strip())
        return render_result.strip()