core: fixed an error when loading templates

alploskov · Aug 6, 2022 · 3c72e38 · 3c72e38
1 parent 957f39a
commit 3c72e38
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/kithon/core.py b/kithon/core.py
@@ -144,10 +144,12 @@ def add_templ(self, name, template):
         elif isinstance(template, dict):
             for field, value in template.items():
                 keywords = [
-                    'tmp', 'type', 'import_code',
+                    'type', 'import_code',
                     'code', 'alt_name', 'ret_type',
                     'meta', 'decorate', 'args'
                 ]
+                if field == 'tmp':
+                    self.templates[name].update({'tmp': Template(value)})
                 if field in keywords:
                     self.templates[name].update({field: value})
                 else:

diff --git a/kithon/node.py b/kithon/node.py
@@ -31,7 +31,7 @@ def __init__(self, env=None, tmp=None, name=None, parts=None, type=None, nl=1, o
             self.tmp = env.templates[tmp].get('tmp', '')
         else:
             self.name = name or 'unknown'
-            self.tmp = Template(tmp)
+            self.tmp = Template(tmp or '')
         self.parts = parts
         self.type = type or 'None'
         self.env = env