mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-15 23:31:05 +00:00
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
"""批量提取 .py 文件的 import/from 之前的注释"""
|
||
|
||
from pathlib import Path
|
||
|
||
|
||
def extract_leading_comments(file_path: Path) -> str:
|
||
"""
|
||
从 .py 文件中提取 import/from 之前的全部内容（包括注释、docstring、空行等）
|
||
"""
|
||
try:
|
||
with open(file_path, "r", encoding="utf-8") as f:
|
||
lines = f.readlines()
|
||
except (UnicodeDecodeError, OSError) as e:
|
||
print(f"⚠️ 无法读取文件 {file_path}: {e}")
|
||
return ""
|
||
|
||
# 找到第一个非空且以 import 或 from 开头的行（忽略前面的空白和注释）
|
||
for i, line in enumerate(lines):
|
||
stripped = line.strip()
|
||
if not stripped:
|
||
continue
|
||
if stripped.startswith(("import ", "from ")):
|
||
# 找到了第一个 import/from，返回之前的所有内容
|
||
return "".join(lines[:i]).rstrip() + "\n\n"
|
||
|
||
# 如果整个文件都没有 import/from，则返回全部内容（可能是纯脚本或空文件）
|
||
content = "".join(lines).rstrip()
|
||
return content + "\n\n" if content else ""
|
||
|
||
|
||
def main(root_dir: str, output_file: str = "merged_comments.txt"):
|
||
"""批量提取 .py 文件的 import/from 之前的注释"""
|
||
root = Path(root_dir)
|
||
if not root.exists() or not root.is_dir():
|
||
raise ValueError(f"目录不存在: {root_dir}")
|
||
|
||
all_content = []
|
||
py_files = list(root.rglob("*.py"))
|
||
|
||
print(f"🔍 找到 {len(py_files)} 个 .py 文件，开始提取前导注释...")
|
||
|
||
for py_file in py_files:
|
||
comment = extract_leading_comments(py_file)
|
||
if comment.strip(): # 只保留非空内容
|
||
header = f"\n{'=' * 60}\n# 文件: {py_file.relative_to(root)}\n{'=' * 60}\n"
|
||
all_content.append(header + comment)
|
||
|
||
# 写入合并文件
|
||
with open(output_file, "w", encoding="utf-8") as out:
|
||
out.write("# 合并自所有 .py 文件的前导注释（import 之前的内容）\n")
|
||
out.write("=" * 70 + "\n\n")
|
||
out.writelines(all_content)
|
||
|
||
print(f"✅ 合并完成！结果已保存至: {output_file}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 👇 修改为你自己的目标目录
|
||
target_directory = "/Users/yuli/workspace/ali_openclaw/src" # 例如："/home/user/myproject" 或 r"C:\myproject"
|
||
main(target_directory)
|