Try allowing include of templates relative to parent template

SUNET · May 7, 2021 · 6580403 · 6580403
1 parent 76a1331
commit 6580403
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/cnaas_nms/confpush/nornir_helper.py b/src/cnaas_nms/confpush/nornir_helper.py
@@ -19,11 +19,16 @@ class NornirJobResult(JobResult):
     change_score: Optional[float] = None
 
 
-cnaas_jinja_env = JinjaEnvironment(
+class RelativeJinjaEnvironment(JinjaEnvironment):
+    """Enable relative template paths"""
+    def join_path(self, template, parent):
+        return os.path.join(os.path.dirname(parent), template)
+
+
+cnaas_jinja_env = RelativeJinjaEnvironment(
     trim_blocks=True,
     lstrip_blocks=True,
-    keep_trailing_newline=True,
-    cache_size=0
+    keep_trailing_newline=True
 )