from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent


def find_python_files(dir_, file):
    with open(file, "w", encoding="utf-8") as f:
        for py_file in dir_.rglob("*.py"):  # 递归查找所有 .py 文件
            f.write(str(py_file) + "\n")  # 写入文件路径


# 设置源目录和输出文件
source_directory = ROOT.joinpath("source")  # 源目录
output_file = "py_files.txt"  # 输出文件名

find_python_files(source_directory, output_file)
print(f"所有 .py 文件路径已保存到 {output_file}")