黑盒，白盒处理合并

2026-08-24 18:34:17 +08:00
parent 5a968b4e37
commit cbaf3deb36
75 changed files with 10020 additions and 107 deletions
@@ -0,0 +1,86 @@
+#!/usr/bin/env python
+"""COBOL 迁移验证平台全流程入口：先白盒 cobol_testgen 再黑盒 LLM 数据生成。
+
+用法:
+  python run.py --design <詳細設計書.md> --source <程序.cbl> \
+      --file-db-md <COPY句定義書.md> --cpy <COPYBOOK目录> \
+      --db-md <DB定義書.md> --output <输出目录>
+
+步骤:
+  1) cobol_testgen  (白盒静态分析 + 测试数据生成)   → python -m cobol_testgen --gcov <source> <output>
+  2) black-box-data-create (DeepSeek LLM 数据生成) → 透传全部参数给 black-box-data-create/main.py
+
+任一步失败即停止，返回该步退出码。
+"""
+import argparse
+import os
+import subprocess
+import sys
+
+ROOT = os.path.dirname(os.path.abspath(__file__))
+BLACKBOX_MAIN = os.path.join(ROOT, "black-box-data-create", "main.py")
+
+
+def build_parser():
+    p = argparse.ArgumentParser(
+        description="COBOL 迁移验证平台：先跑白盒 cobol_testgen，再跑黑盒 LLM 数据生成")
+    p.add_argument("--design", required=True, help="詳細設計書 .md のパス")
+    p.add_argument("--source", required=True, help="COBOL ソース .cbl のパス")
+    p.add_argument("--file-db-md", required=True, help="ファイル/DB 構造 .md のパス")
+    p.add_argument("--cpy", required=True, help="COPYBOOK 格納ディレクトリ")
+    p.add_argument("--db-md", required=True, help="DB 定義書 .md のパス")
+    p.add_argument("--output", default="output", help="出力ディレクトリ")
+    p.add_argument("--api-key", help="DeepSeek API Key（透传给黑盒）")
+    p.add_argument("--model", help="API モデル名（透传给黑盒）")
+    p.add_argument("--rules", help="ルール格納ディレクトリ（透传给黑盒）")
+    p.add_argument("--max-tokens", type=int, help="API 生成トークン上限（透传给黑盒）")
+    p.add_argument("--dry-run", action="store_true", help="只打印要执行的命令，不真正执行")
+    return p
+
+
+def _run(cmd, cwd, label, dry_run=False):
+    print(f"\n== {label} ==")
+    print(f"  $ {' '.join(cmd)}")
+    if dry_run:
+        print("  [dry-run] 跳过执行")
+        return 0
+    r = subprocess.run(cmd, cwd=cwd)
+    return r.returncode
+
+
+def main():
+    args = build_parser().parse_args()
+
+    rc = _run(
+        [sys.executable, "-m", "cobol_testgen", "--gcov", args.source, args.output],
+        cwd=ROOT,
+        label="步骤1: cobol_testgen 白盒数据生成",
+        dry_run=args.dry_run,
+    )
+    if rc != 0:
+        return rc
+
+    bb_cmd = [
+        sys.executable, BLACKBOX_MAIN,
+        "--design", args.design,
+        "--source", args.source,
+        "--file-db-md", args.file_db_md,
+        "--cpy", args.cpy,
+        "--db-md", args.db_md,
+        "--output", args.output,
+    ]
+    for opt in ("--api-key", "--model", "--rules", "--max-tokens"):
+        v = getattr(args, opt.lstrip("-").replace("-", "_"))
+        if v is not None:
+            bb_cmd.extend([opt, str(v)])
+
+    return _run(
+        bb_cmd,
+        cwd=ROOT,
+        label="步骤2: black-box-data-create LLM 数据生成",
+        dry_run=args.dry_run,
+    )
+
+
+if __name__ == "__main__":
+    sys.exit(main())