markdown库依赖过低和修改默认加载toc

requirements.txt

@@ -1,4 +1,4 @@
-Markdown==2.6.8
+Markdown==3.0.1
 Pygments==1.6
 Jinja2==2.7.3
 PyYAML==3.11

simiki/generators.py

@@ -142,6 +142,14 @@ def get_meta_and_content(self, do_render=True):
 
         return meta, content
 
+    def get_toc(self):
+        with io.open(self._src_file, "rt", encoding="utf-8") as fd:
+            cont_str=fd.read()
+        if cont_str:
+            md = markdown.Markdown(extensions=['markdown.extensions.toc'])
+            md.convert(cont_str)
+        return md.toc_tokens if cont_str else []
+
     def get_layout(self, meta):
         """Get layout config in meta, default is `page'"""
         if "layout" in meta:
@@ -170,7 +178,7 @@ def get_template_vars(self, meta, content):
         page = {"content": content}
         page.update(meta)
         page.update({'relation': self.get_relation()})
-
+        page.update({"toc":self.get_toc()})
         template_vars.update({'page': page})
 
         return template_vars

tests/test_generators.py

@@ -93,7 +93,8 @@ def test_get_template_vars(self):
                 u'date': '2013-10-17 00:03',
                 u'layout': 'page',
                 u'relation': [],
-                u'title': 'Foo Page 2'
+                u'title': 'Foo Page 2',
+                u'toc': []
             },
             u'site': get_default_config()
         }