diff --git a/src/templates/__init__.py b/src/templates/__init__.py index 8f4c03f..6fdd5bf 100644 --- a/src/templates/__init__.py +++ b/src/templates/__init__.py @@ -1 +1,23 @@ -# Templates for documentation generation +"""Template module for documentation generation.""" + +import os + +from jinja2 import Environment, FileSystemLoader + +TEMPLATES_DIR = os.path.join(os.path.dirname(__file__)) + +def startswith(s, prefix): + return s.startswith(prefix) if s else False + +env = Environment( + loader=FileSystemLoader(TEMPLATES_DIR), + trim_blocks=True, + lstrip_blocks=True, +) +env.filters["startswith"] = startswith + +HTML_TEMPLATE = env.get_template("html_template.html") +MARKDOWN_TEMPLATE = env.get_template("markdown_template.md") +JSON_TEMPLATE = env.get_template("json_template.json") + +__all__ = ["HTML_TEMPLATE", "MARKDOWN_TEMPLATE", "JSON_TEMPLATE", "env"]