Markdown TOC Maker

md_toc.py

import re
import sys
from pathlib import Path

def parse_markdown_for_toc(markdown_content):
    heading_pattern = re.compile(r'^(#{1,})\s+(.+?)(\s*)$', re.MULTILINE)

    toc = []

    for match in heading_pattern.finditer(markdown_content):
        hashes, heading, _ = match.groups()
        level = len(hashes)

        anchor = re.sub(r'[^\w\s-]', '', heading).replace(' ', '-').lower()
        toc.append((level, heading.strip(), anchor))

    toc_str = ""

    for level, heading, anchor in toc:
        toc_str += "    " * (level - 1) + f"* [{heading}](#{anchor})\n"

    return toc_str

def create_toc_from_markdown_file(file_path):
    markdown_path = Path(file_path)
    markdown_content = markdown_path.read_text(encoding='utf-8')

    toc = parse_markdown_for_toc(markdown_content)
    return toc

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print(f"Usage: python {sys.argv[0]} <path_to_markdown_file>")
        sys.exit(1)

    file_path = sys.argv[1]
    toc = create_toc_from_markdown_file(file_path)
    print(toc)