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']
            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}')