#!/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/distribution_risk_score_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 []

    scores = {}
    for row in core_satellite:
        ticker = row.get("Ticker")
        if not ticker:
            continue
        close = row.get("Close") or 0.0
        ma20 = row.get("MA20") or close
        
        # Calculate distribution risk score: 0 to 100
        score = min(100, max(0, int((close - ma20) / ma20 * 200)))
        scores[ticker] = {
            "distribution_risk_score": score,
            "status": "HIGH" if score >= 50 else "NORMAL"
        }

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

if __name__ == "__main__":
    main()
