Support builtin and third extensions

tankywoo · Jan 21, 2020 · 7517e74 · 7517e74
1 parent a365e54
commit 7517e74
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/simiki/generators.py b/simiki/generators.py
@@ -250,6 +250,7 @@ def _set_markdown_extensions(self):
             "nl2br": {},
             "toc": {"title": "Table of Contents"},
             "extra": {},
+            "markdown_checklist.extension": {},
         }
         # Handle pygments
         if self.site_config["pygments"]:
@@ -264,15 +265,17 @@ def _set_markdown_extensions(self):
         markdown_extensions = []
         # add builtin markdown extensions
         for k, v in markdown_extensions_config.items():
-            ext = import_string("markdown.extensions." + k).makeExtension()
+            if '.' not in k:
+                # builtin extensions
+                ext = import_string("markdown.extensions." + k).makeExtension()
+            else:
+                # third extensions, use module path
+                ext = import_string(k).makeExtension()
             if v:
                 for i, j in v.items():
                     ext.setConfig(i, j)
             markdown_extensions.append(ext)
 
-        # add 3rd markdown extensions
-        markdown_extensions.append('markdown_checklist.extension')
-
         return markdown_extensions
 
     def get_relation(self):