29 lines
1.3 KiB
Python
29 lines
1.3 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def generate_markdown_from_py_files(directory, output_file):
|
||
|
|
with open(output_file, 'w', encoding='utf-8') as md_file:
|
||
|
|
for root, dirs, files in os.walk(directory):
|
||
|
|
# 排除 venv 目录
|
||
|
|
dirs[:] = [d for d in dirs if d != '.venv']
|
||
|
|
dirs[:] = [d for d in dirs if d != '.vscode']
|
||
|
|
dirs[:] = [d for d in dirs if d != 'scripts']
|
||
|
|
dirs[:] = [d for d in dirs if d != 'build']
|
||
|
|
dirs[:] = [d for d in dirs if d != 'lib']
|
||
|
|
for file in files:
|
||
|
|
if file.endswith('.py') or file.endswith('.rs'):
|
||
|
|
file_path = os.path.join(root, file)
|
||
|
|
md_file.write(f'`{file_path}`\n')
|
||
|
|
md_file.write('```python\n')
|
||
|
|
with open(file_path, 'r', encoding='utf-8') as py_file:
|
||
|
|
md_file.write(py_file.read())
|
||
|
|
md_file.write('\n```\n\n')
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# 指定目录和输出文件名
|
||
|
|
target_directory = sys.argv[1] # 替换为你的目标目录
|
||
|
|
output_markdown_file = 'output.md' # 输出的 Markdown 文件名
|
||
|
|
|
||
|
|
generate_markdown_from_py_files(target_directory, output_markdown_file)
|
||
|
|
print(f'Markdown 文件已生成：{output_markdown_file}')
|