#!/usr/bin/env python3
from __future__ import annotations

import argparse
import json
from pathlib import Path


ROOT = Path(__file__).resolve().parents[1]
DEFAULT_JSON = ROOT / "Temp" / "fundamental_raw_evidence_v4.json"


def main() -> int:
    ap = argparse.ArgumentParser()
    ap.add_argument("--json", default=str(DEFAULT_JSON))
    args = ap.parse_args()
    path = Path(args.json)
    data = json.loads(path.read_text(encoding="utf-8")) if path.exists() else {}
    exposure = data.get("imputed_data_exposure") if isinstance(data.get("imputed_data_exposure"), dict) else {}
    has_provenance = bool(data.get("raw_fundamental_value_provenance") is True)
    has_fields = isinstance(exposure.get("missing_count"), (int, float)) and isinstance(exposure.get("coverage_pct"), (int, float))
    ok = has_provenance and has_fields
    payload = {
        "formula_id": "IMPUTED_DATA_EXPOSURE_V1",
        "gate": "PASS" if ok else "FAIL",
        "missing_count": int(exposure.get("missing_count") or 0),
        "coverage_pct": float(exposure.get("coverage_pct") or 0.0),
        "has_provenance": has_provenance,
    }
    print(json.dumps(payload, ensure_ascii=True, indent=2))
    return 0 if ok else 1


if __name__ == "__main__":
    raise SystemExit(main())
