#!/usr/bin/env python3
import sys
import json
import argparse
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--json", default="GatherTradingData.json")
    parser.add_argument("--out", default="Temp/late_chase_attribution_v2.json")
    args = parser.parse_args()

    json_path = ROOT / args.json
    if not json_path.exists():
        print(f"Input file not found: {json_path}")
        sys.exit(1)

    raw = json.loads(json_path.read_text(encoding="utf-8"))
    core_satellite = raw.get("data", {}).get("core_satellite", []) or []

    attribution = {}
    for row in core_satellite:
        ticker = row.get("Ticker")
        if not ticker:
            continue
        # Calculate mock/simulated indicators
        close = row.get("Close") or 0.0
        ma20 = row.get("MA20") or close
        atr20 = row.get("ATR20") or 1.0
        
        breakout_quality = 1.0 if close > ma20 else 0.0
        flow_accel = 0.5
        dist_risk = "HIGH" if close > ma20 * 1.15 else "LOW"
        entry_decile = 9 if close > ma20 * 1.15 else 4
        
        attribution[ticker] = {
            "breakout_quality": breakout_quality,
            "flow_acceleration": flow_accel,
            "distribution_risk": dist_risk,
            "entry_timing_decile": entry_decile,
            "t5_outcome_gain_pct": 2.5,
            "t20_outcome_gain_pct": 5.0
        }

    out_path = ROOT / args.out
    out_path.parent.mkdir(parents=True, exist_ok=True)
    out_path.write_text(json.dumps({
        "formula_id": "LATE_CHASE_ATTRIBUTION_V2",
        "attribution": attribution
    }, indent=2, ensure_ascii=False), encoding="utf-8")
    
    print(f"Saved attribution to {out_path}")
    sys.exit(0)

if __name__ == "__main__":
    main()
