#!/usr/bin/env python3
"""
prune_branches.py — 分支剪枝引擎（每日）

裁掉低效分支：trigger_count ≥ 3 且 success_rate < 30%
被剪的分支从 strategy_tree.branches 移除，归档到 strategy_tree.pruned_branches

Dad说"每周"太低频 → 改为每日16:30（收盘后）
"""

import json, sys
from datetime import datetime

DECISIONS_PATH = "/home/hmo/web-dashboard/data/decisions.json"

# 剪枝阈值：触发≥3次且成功率<30%
TRIGGER_MIN = 3
SUCCESS_MAX = 30


def prune():
    try:
        with open(DECISIONS_PATH) as f:
            data = json.load(f)
    except Exception as e:
        print(f"[错误] 读 decisions.json 失败: {e}", file=sys.stderr)
        return 1

    decisions = data.get("decisions", [])
    total_pruned = 0
    results = []

    for entry in decisions:
        code = entry.get("code", "")
        tree = entry.get("strategy_tree")
        if not tree:
            continue
        branches = tree.get("branches", [])
        if not branches:
            continue

        pruned_branches = tree.get("pruned_branches", [])
        kept = []
        for br in branches:
            tc = br.get("trigger_count", 0)
            sr = br.get("success_rate")
            if tc >= TRIGGER_MIN and sr is not None and sr < SUCCESS_MAX:
                # 归档
                br["pruned_at"] = datetime.now().isoformat()
                pruned_branches.append(br)
                total_pruned += 1
                results.append({
                    "code": code,
                    "branch_id": br.get("id", "?"),
                    "trigger_count": tc,
                    "success_rate": sr,
                    "rationale": br.get("rationale", ""),
                })
            else:
                kept.append(br)

        tree["branches"] = kept
        tree["pruned_branches"] = pruned_branches
        tree["last_pruned"] = datetime.now().isoformat() if total_pruned > 0 else tree.get("last_pruned", "")

    with open(DECISIONS_PATH, "w") as f:
        json.dump(data, f, indent=2, ensure_ascii=False)

    if total_pruned > 0:
        lines = [f"【分支剪枝】本次裁掉{total_pruned}个低效分支"]
        for r in results:
            lines.append(f"  ✂ {r['code']}/{r['branch_id']}(触发{r['trigger_count']}次/成功率{r['success_rate']}%)")
            lines.append(f"    理由: {r['rationale']}")
        print("\n".join(lines))
        return 0
    else:
        # 静默
        print(f"【分支剪枝】无需剪枝（所有分支均未达到触发{TRIGGER_MIN}次且成功率<{SUCCESS_MAX}%的阈值）")
        return 0


if __name__ == "__main__":
    sys.exit(prune())