From f8e216b7eb31a37698fb88ce7883ce94c261e7ac Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Sat, 19 Oct 2024 23:21:00 -0600
Subject: add toc to posts

---
 generate_me.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'generate_me.py')

diff --git a/generate_me.py b/generate_me.py
index 4e8ca58..1b1f5d6 100644
--- a/generate_me.py
+++ b/generate_me.py
@@ -1,4 +1,4 @@
-from markdown3 import Markdown
+from markdown2 import Markdown, UnicodeWithAttrs
 import os
 from jinja2 import Environment, FileSystemLoader
 from distutils.dir_util import copy_tree
@@ -25,13 +25,14 @@ image_text_font = "fonts/futura_light.ttf"
 
 md = Markdown(
     extras=[
+        "toc",
         "fenced-code-blocks",
         "metadata",
         "task_list",
         "tables",
         "target-blank-links",
         "header-ids",
-        "latex",
+        "latex"
     ]
 )
 
@@ -101,6 +102,12 @@ for x in os.walk(src_folder):
                         _post["image_link"] = "/images/opengraph" + fpath.replace(
                             src_folder, ""
                         ).replace("md", "png")
+                        toc_html = md._toc_html
+                        position = _html.find('</h1>')
+                        if position != -1:
+                            metadata_copy = _html.metadata
+                            _html = UnicodeWithAttrs(_html[:position+5] + toc_html + _html[position+5:])
+                            _html.metadata = metadata_copy
 
                         to_write_path = "./Resources" + _post["image_link"]
 
-- 
cgit v1.2.3