-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgenerate_readme.py
32 lines (25 loc) · 1 KB
/
generate_readme.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import os
def write_file_to_readme(readme, file_path, file_name):
with open(file_path, "r") as f:
content = f.read()
readme.write(f"<details><summary>{file_name}</summary>\n\n")
readme.write("```py\n")
readme.write(content)
readme.write("\n```\n")
readme.write("</details>\n\n")
directory = "./src"
readme_file = "./README.md"
with open(readme_file, "w") as readme:
with open("./readme/prefix.md", "r") as prefix:
readme.write(prefix.read())
for dirpath, dirnames, filenames in os.walk(directory):
if "pycache" in dirpath:
continue
# If we're not in the root directory, write the directory name
if dirpath != directory:
dirname = os.path.basename(dirpath)
readme.write(f"### {dirname}\n")
for filename in filenames:
if filename.endswith(".py"):
file_path = os.path.join(dirpath, filename)
write_file_to_readme(readme, file_path, filename)