Fixes #26. (#27)

rosscdh · Nov 2, 2020 · b816b32 · b816b32
1 parent 9af2957
commit b816b32
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/markdownextradata/plugin.py b/markdownextradata/plugin.py
@@ -7,7 +7,7 @@
 from mkdocs.plugins import BasePlugin
 from mkdocs.utils import warning_filter
 
-from jinja2 import Template
+import jinja2
 from pathlib import Path
 from itertools import chain
 
@@ -91,5 +91,6 @@ def on_pre_build(self, config, **kwargs):
     def on_page_markdown(self, markdown, config, **kwargs):
         context = {key: config.get(key) for key in CONFIG_KEYS if key in config}
         context.update(config.get("extra", {}))
-        md_template = Template(markdown)
+        env = jinja2.Environment(undefined=jinja2.DebugUndefined)
+        md_template = env.from_string(markdown)
         return md_template.render(**config.get("extra"))