chore: jinja

src/py/mat3ra/utils/extra/jinja.py

@@ -2,17 +2,33 @@
 from jinja2 import Environment, FileSystemLoader
 
 
-def render_template_file(template_file: str, **kwargs):
+def render_template_file(template_file_path: str, **kwargs):
     """
     Renders a given template file.
 
     Args:
-        template_file (str): template file path
+        template_file_path (str): template file path
         kwargs: variables passed to the template
 
     Returns:
         str
     """
-    env = Environment(loader=FileSystemLoader(os.path.dirname(template_file)))
-    template_file = env.get_template(os.path.basename(template_file))
-    return template_file.render(**kwargs)
+    env = Environment(loader=FileSystemLoader(os.path.dirname(template_file_path)))
+    template_file_path = env.get_template(os.path.basename(template_file_path))
+    return template_file_path.render(**kwargs)
+
+
+def render_template_string(template_string: str, **kwargs):
+    """
+    Renders a given template string.
+
+    Args:
+        template_string (str): template string
+        kwargs: variables passed to the template
+
+    Returns:
+        str
+    """
+    env = Environment()
+    template = env.from_string(template_string)
+    return template.render(**kwargs)

tests/py/unit/test_extra_jinja.py

@@ -3,10 +3,13 @@
 
 FILE_JINJA_TEMPLATE_PATH = Path(__file__).parent / "./fixtures/file_jinja_template.jinja"
 
+
 def test_render_template():
-    """
-    test_camel_to_snake should be converted to testCamelToSnake
-    """
     context = {"name": "World"}
     rendered = utils.render_template_file(FILE_JINJA_TEMPLATE_PATH, **context)
     assert rendered == "Hello, World!"
+
+def test_render_template_string():
+    context = {"name": "World"}
+    rendered = utils.render_template_string("Hello, {{ name }}!", **context)
+    assert rendered == "Hello, World!"