TOC extension now attaches toc to Markdown instance (Markdown.toc), but only if a marker was not found in the document.

Author: Waylan Limberg

markdown/extensions/toc.py

@@ -23,6 +23,8 @@ def iterparent(self, root):
  yield parent, child
 
  def run(self, doc):
+ marker_found = False
+
  div = etree.Element("div")
  div.attrib["class"] = "toc"
  last_li = None
@@ -61,6 +63,7 @@ def run(self, doc):
  if p[i] == c:
  p[i] = div
  break
+ marker_found = True
 
  if header_rgx.match(c.tag):
  tag_level = int(c.tag[-1])
@@ -106,6 +109,14 @@ def run(self, doc):
  c.append(anchor)
 
  list_stack[-1].append(last_li)
+ if not marker_found:
+ # searialize and attach to markdown instance.
+ prettify = self.markdown.treeprocessors.get('prettify')
+ if prettify: prettify.run(div)
+ toc = self.markdown.serializer(div)
+ for pp in self.markdown.postprocessors.values():
+ toc = pp.run(toc)
+ self.markdown.toc = toc
 
 class TocExtension(markdown.Extension):
  def __init__(self, configs):