mirror of
https://github.com/TheAlgorithms/Python.git
synced 2026-09-28 13:33:11 +08:00
* Add a table of contents to DIRECTORY.md Generate a linked table of contents of the top-level sections at the top of DIRECTORY.md so readers can jump straight to a category, as suggested in #13239 / #13111. The list is built in scripts/build_directory_md.py (with a new md_anchor helper + doctests) so DIRECTORY.md stays fully auto-generated. * Address review: numbered ToC + link section headings to their directories - Table of Contents is now a numbered list, so the final number shows the total count of algorithm folders at a glance. - Each top-level section heading links to its algorithm directory (e.g. ## [Sorts](sorts)), so clicking a section title jumps straight to the folder.
97 lines
3.0 KiB
Python
Executable File
97 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
from collections.abc import Iterator
|
|
|
|
|
|
def good_file_paths(top_dir: str = ".") -> Iterator[str]:
|
|
for dir_path, dir_names, filenames in os.walk(top_dir):
|
|
dir_names[:] = [
|
|
d
|
|
for d in dir_names
|
|
if d != "scripts" and d[0] not in "._" and "venv" not in d
|
|
]
|
|
for filename in filenames:
|
|
if filename == "__init__.py":
|
|
continue
|
|
if os.path.splitext(filename)[1] in (".py", ".ipynb"):
|
|
yield os.path.join(dir_path, filename).lstrip("./")
|
|
|
|
|
|
def md_prefix(indent: int) -> str:
|
|
"""
|
|
Markdown prefix based on indent for bullet points
|
|
|
|
>>> md_prefix(0)
|
|
'\\n##'
|
|
>>> md_prefix(1)
|
|
' *'
|
|
>>> md_prefix(2)
|
|
' *'
|
|
>>> md_prefix(3)
|
|
' *'
|
|
"""
|
|
return f"{indent * ' '}*" if indent else "\n##"
|
|
|
|
|
|
def md_anchor(heading: str) -> str:
|
|
"""
|
|
GitHub-style anchor slug for a section heading, so the table of contents
|
|
can link to it.
|
|
|
|
>>> md_anchor("Audio Filters")
|
|
'audio-filters'
|
|
>>> md_anchor("Bit Manipulation")
|
|
'bit-manipulation'
|
|
>>> md_anchor("Computer Vision")
|
|
'computer-vision'
|
|
"""
|
|
return heading.strip().lower().replace(" ", "-")
|
|
|
|
|
|
def print_path(old_path: str, new_path: str) -> str:
|
|
old_parts = old_path.split(os.sep)
|
|
for i, new_part in enumerate(new_path.split(os.sep)):
|
|
if (i + 1 > len(old_parts) or old_parts[i] != new_part) and new_part:
|
|
title = new_part.replace("_", " ").title()
|
|
if i == 0:
|
|
# Link each top-level section heading to its algorithm
|
|
# directory, so readers can click the title and jump straight
|
|
# to the folder (no leading pound sign, unlike the ToC links).
|
|
print(f"{md_prefix(i)} [{title}]({new_part})")
|
|
else:
|
|
print(f"{md_prefix(i)} {title}")
|
|
return new_path
|
|
|
|
|
|
def print_directory_md(top_dir: str = ".") -> None:
|
|
filepaths = sorted(good_file_paths(top_dir))
|
|
|
|
# Top-level sections, in the order they appear, for the table of contents.
|
|
sections = list(
|
|
dict.fromkeys(
|
|
fp.split(os.sep)[0].replace("_", " ").title()
|
|
for fp in filepaths
|
|
if os.sep in fp
|
|
)
|
|
)
|
|
print("## Table of Contents")
|
|
for index, section in enumerate(sections, start=1):
|
|
# Numbered list so the final number is the total count of algorithm
|
|
# folders, visible at a glance.
|
|
print(f"{index}. [{section}](#{md_anchor(section)})")
|
|
|
|
old_path = ""
|
|
for filepath in filepaths:
|
|
filepath, filename = os.path.split(filepath)
|
|
if filepath != old_path:
|
|
old_path = print_path(old_path, filepath)
|
|
indent = (filepath.count(os.sep) + 1) if filepath else 0
|
|
url = f"{filepath}/{filename}".replace(" ", "%20")
|
|
filename = os.path.splitext(filename.replace("_", " ").title())[0]
|
|
print(f"{md_prefix(indent)} [{filename}]({url})")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print_directory_md(".")
|