Files
definitive-opensource-mustb…/core/source/generation/tableofcontents_generator.py

75 lines
2.0 KiB
Python

import json
# Utils
def slugify(name):
return name.lower().replace(" ", "-").replace("(", "").replace(")", "")
def generate_table_of_contents():
with open("data/static/categories.json", "r", encoding="utf-8") as f:
data = json.load(f)
categories = data.get("categories", [])
subcategories = data.get("subcategories", [])
# Build the alphabetical list (ignoring parent categories)
subcat_names = [sub["name"] for sub in subcategories]
subcat_names.sort(key=lambda x: x.lower())
alphabetical_md = ""
for name in subcat_names:
alphabetical_md += f"- [{name}](#{slugify(name)})\n"
# Build the categorized list
parent_map = {cat["id"]: cat["name"] for cat in categories}
grouped = {}
for sub in subcategories:
parent = sub.get("parent", "other")
grouped.setdefault(parent, []).append(sub["name"])
for key in grouped:
grouped[key].sort(key=lambda x: x.lower())
parents = [(pid, parent_map.get(pid, "Other")) for pid in grouped if pid != "other"]
parents.sort(key=lambda x: x[1].lower())
if "other" in grouped:
parents.append(("other", "Other"))
categorized_md_lines = []
for pid, pname in parents:
categorized_md_lines.append(f"- {pname}")
for subname in grouped[pid]:
categorized_md_lines.append(f" - [{subname}](#{slugify(subname)})")
# Append fixed sections at the end of the categorized TOC
fixed_sections = [
"Removed Projects",
"FAQ",
"Honorable Mentions of Closed-Source Software",
]
for item in fixed_sections:
categorized_md_lines.append(f"- [{item}](#{slugify(item)})")
categorized_md = "\n".join(categorized_md_lines)
toc = f"""## Table of Contents
<details>
<summary><b>Alphabetical</b></summary> <br />
{alphabetical_md}
</details>
<details open>
<summary><b>Categorized</b></summary> <br />
{categorized_md}
</details>
"""
return toc
if __name__ == "__main__":
# For testing the TOC generator
print(generate_table_of_contents())