mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-28 16:20:34 +08:00
Gareth Davidson
23 lines
688 B
Python
23 lines
688 B
Python
"""Message templates for the discord bot."""
|
|
import typing
|
|
|
|
import jinja2
|
|
from loguru import logger
|
|
|
|
|
|
class MessageTemplates:
|
|
"""Create message templates for the discord bot."""
|
|
|
|
def __init__(self, template_dir: str = "./templates"):
|
|
self.env = jinja2.Environment( # noqa: S701
|
|
loader=jinja2.FileSystemLoader(template_dir),
|
|
autoescape=jinja2.select_autoescape(disabled_extensions=("msg",), default=False, default_for_string=False),
|
|
)
|
|
|
|
def render(self, template_name: str, **kwargs: typing.Any):
|
|
template = self.env.get_template(template_name)
|
|
txt = template.render(kwargs)
|
|
logger.debug(txt)
|
|
|
|
return txt
|