TheAlgorithms-Python

1
#!/usr/bin/env python3
2

3
import os
4
from collections.abc import Iterator
5

6

7
def good_file_paths(top_dir: str = ".") -> Iterator[str]:
8
    for dir_path, dir_names, filenames in os.walk(top_dir):
9
        dir_names[:] = [
10
            d
11
            for d in dir_names
12
            if d != "scripts" and d[0] not in "._" and "venv" not in d
13
        ]
14
        for filename in filenames:
15
            if filename == "__init__.py":
16
                continue
17
            if os.path.splitext(filename)[1] in (".py", ".ipynb"):
18
                yield os.path.join(dir_path, filename).lstrip("./")
19

20

21
def md_prefix(i):
22
    return f"{i * '  '}*" if i else "\n##"
23

24

25
def print_path(old_path: str, new_path: str) -> str:
26
    old_parts = old_path.split(os.sep)
27
    for i, new_part in enumerate(new_path.split(os.sep)):
28
        if (i + 1 > len(old_parts) or old_parts[i] != new_part) and new_part:
29
            print(f"{md_prefix(i)} {new_part.replace('_', ' ').title()}")
30
    return new_path
31

32

33
def print_directory_md(top_dir: str = ".") -> None:
34
    old_path = ""
35
    for filepath in sorted(good_file_paths(top_dir)):
36
        filepath, filename = os.path.split(filepath)
37
        if filepath != old_path:
38
            old_path = print_path(old_path, filepath)
39
        indent = (filepath.count(os.sep) + 1) if filepath else 0
40
        url = f"{filepath}/{filename}".replace(" ", "%20")
41
        filename = os.path.splitext(filename.replace("_", " ").title())[0]
42
        print(f"{md_prefix(indent)} [{filename}]({url})")
43

44

45
if __name__ == "__main__":
46
    print_directory_md(".")
47