✅ 完全可以！**不依赖任何外部配置文件（如 `launch.json`、`tasks.json`），也不需要命令行参数** —— 我们直接在 `extract_homeagent.py` 脚本内部「硬编码」你的路径和输出格式，实现 **双击/一键运行即出结果**。 > ✅ 效果： > 你只需把修改后的 `extract_homeagent.py` 放在任意位置 → 双击它（或运行 `python extract_homeagent.py`）→ 自动扫描 `D:\bixby3-data\...\SRCB` → 生成 `output.json`，全程无交互、无报错静默完成（或带清晰错误提示）。 --- ### ✅ 修改方案（仅需改 3 处，5 分钟搞定） #### 🔧 步骤 1：打开 `extract_homeagent.py`，找到 `if __name__ == "__main__":` 块 （通常在文件最底部） #### 🔧 步骤 2：**完全替换原有 `main()` 调用逻辑**（删除旧的 `argparse` 解析部分，用固定参数替代） > ⚠️ 注意：以下代码 **兼容 Python 3.6+，零依赖，自动处理 Windows 路径、中文、JSON 编码、错误捕获**： ```python if __name__ == "__main__": # ==================== 🔒 硬编码配置区（按需修改下方3行） ==================== INPUT_DIR = r"D:\bixby3-data\bixby3-data\llm-finetuning-data\zh-CN\train\SRCB" # ← 你的输入路径（原始字符串，安全！） OUTPUT_FORMAT = "json" # ← "csv" 或 "json" OUTPUT_FILE = "output.json" # ← 输出文件名（自动保存到当前目录） # ============================================================================== # ✅ 自动校验路径是否存在 import os if not os.path.isdir(INPUT_DIR): print(f"❌ 错误：输入目录不存在！\n 请检查路径：{INPUT_DIR}") input("按回车键退出...") exit(1) # ✅ 自动搜索所有 JSON 文件（递归） import glob json_files = glob.glob(os.path.join(INPUT_DIR, "**", "*.json"), recursive=True) if not json_files: print(f"⚠️ 警告：在 {INPUT_DIR} 下未找到任何 .json 文件！") print(" 请确认路径是否正确，或 JSON 是否在子文件夹中。") input("按回车键退出...") exit(1) print(f"🔍 开始处理 {len(json_files)} 个 JSON 文件...") # ✅ 导入主逻辑（假设原脚本有 extract_data() 或类似函数） # ↓↓↓ 如果你原脚本的主函数叫别的名字（如 `run()`, `process()`），请在此处修改 ↓↓↓ try: # 方案 A：如果原脚本有独立函数（推荐！例如 def extract_homeagent_data(...):） # from extract_homeagent import extract_homeagent_data # results = extract_homeagent_data(json_files) # 方案 B：如果原逻辑全在 if __name__ == "__main__": 下，我们直接复刻核心逻辑： import json results = [] for i, fp in enumerate(json_files): try: with open(fp, "r", encoding="utf-8") as f: data = json.load(f) # 👇 核心提取逻辑（按你原脚本的规则匹配 HomeAgent） # 示例：遍历 turns → invocations → agent_id == "HomeAgent" if isinstance(data, dict) and "turns" in data: for turn in data["turns"]: if isinstance(turn, dict) and "invocations" in turn: for inv in turn["invocations"]: if isinstance(inv, dict) and inv.get("agent_id") == "HomeAgent": # 提取你需要的字段（示例） record = { "file": os.path.relpath(fp, INPUT_DIR), "request": turn.get("request", ""), "request_en": turn.get("request_en", ""), "nlg": inv.get("response", {}).get("nlg", ""), } results.append(record) except Exception as e: print(f"⚠️ 跳过文件 {fp}（解析失败）：{e}") # ✅ 写出结果 if OUTPUT_FORMAT == "json": with open(OUTPUT_FILE, "w", encoding="utf-8") as f: json.dump(results, f, ensure_ascii=False, indent=2) print(f"✅ JSON 已保存：{os.path.abspath(OUTPUT_FILE)}（共 {len(results)} 条）") elif OUTPUT_FORMAT == "csv": import csv if results: with open(OUTPUT_FILE, "w", newline="", encoding="utf-8-sig") as f: writer = csv.DictWriter(f, fieldnames=results[0].keys()) writer.writeheader() writer.writerows(results) print(f"✅ CSV 已保存：{os.path.abspath(OUTPUT_FILE)}（共 {len(results)} 行）") else: print("⚠️ 无数据可导出（CSV）") else: print(f"❌ 不支持的格式：{OUTPUT_FORMAT}") except Exception as e: print(f"💥 运行时严重错误：{e}") import traceback traceback.print_exc() input("\n按回车键退出...") ``` --- ### ✅ 修改说明（为什么这能解决你所有问题？） | 问题 | 如何解决 | 代码体现 | |------|-----------|-----------| | **路径 `\t` 转义崩溃** | 使用 `r"D:\..."` 原始字符串，`\` 不被解释 | `INPUT_DIR = r"D:\bixby3-data\..."` | | **无输出/静默失败** | 全程 `print()` + `input()` 阻塞 + `traceback` 捕获 | `print("❌ 错误：...")` + `traceback.print_exc()` | | **JSON 编码乱码** | 强制 `encoding="utf-8"` 读写，CSV 用 `utf-8-sig`（Excel 友好） | `open(..., encoding="utf-8")` | | **空目录不提示** | `if not json_files:` 显式警告 | `print("⚠️ 警告：在 ... 下未找到任何 .json 文件！")` | | **双击运行闪退** | 结尾 `input("按回车键退出...")` 保持窗口 | 最后一行 | --- ### 📌 使用流程（3 步） 1. **复制上面完整代码块** 2. **打开 `extract_homeagent.py`** → 删除从 `if __name__ == "__main__":` 开始的所有内容 → **粘贴进去** 3. **修改三处变量**（就改这 3 行！）： ```python INPUT_DIR = r"D:\bixby3-data\bixby3-data\llm-finetuning-data\zh-CN\train\SRCB" # ✅ 改成你的路径 OUTPUT_FORMAT = "json" # ✅ 改成 "csv" 也行 OUTPUT_FILE = "my_result.json" # ✅ 改成你喜欢的文件名 ``` 4. **保存文件** → 双击它（或终端运行 `python extract_homeagent.py`）→ 看结果！ > ✅ 成功时你会看到： > ``` > 🔍 开始处理 127 个 JSON 文件... > ✅ JSON 已保存：C:\your\path\output.json（共 89 条） > > 按回车键退出... > ``` --- ### 💡 进阶建议（可选但强烈推荐） #### ✅ 让双击更友好：生成 `.bat` 批处理（点击即运行，错误弹窗） 新建文件 `run.bat`，内容： ```bat @echo off python extract_homeagent.py pause ``` → 双击 `run.bat`，即使报错也会暂停显示。 #### ✅ 自动打开结果文件（Windows） 在 `print(f"✅ JSON 已保存：...")` 后加一行： ```python os.startfile(OUTPUT_FILE) # 自动用默认程序打开 output.json ``` #### ✅ 支持拖拽文件夹到脚本上运行（高级） 把 `INPUT_DIR = r"..."` 替换为： ```python import sys INPUT_DIR = sys.argv[1] if len(sys.argv) > 1 else r"D:\..." ``` → 直接把 `SRCB` 文件夹拖到 `extract_homeagent.py` 图标上即可运行！ --- 需要我帮你： ✅ **根据你原 `extract_homeagent.py` 的真实代码，定制化这段硬编码逻辑**（发我前 30 行，我精准适配） ✅ **生成一个带图标、无黑框的 `.exe` 可执行文件**（PyInstaller 打包，双击即用） ✅ **写一个「智能路径探测」版本**：自动向上查找 `SRCB` 文件夹（适合项目结构不固定） 随时发我 👇