Upload app.py

app.py

@@ -1,13 +1,16 @@
 from __future__ import annotations
 
 import asyncio
-from datetime import date, datetime, time
+import hmac
+import os
+from datetime import date, datetime, time, timedelta
 
 import sys
 from pathlib import Path
 
-from fastapi import FastAPI, Query
+from fastapi import Body, Depends, Header, HTTPException, Query
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI
 
 sys.path.insert(0, str(Path(__file__).resolve().parent))
 from nifty_backend.runtime import dashboard_payload, latest_saved_prediction, refresh_daily_data, refresh_first5_prediction, seconds_until_next_ist_run, IST
@@ -24,12 +27,136 @@ app.add_middleware(
 
 
 market_status = "Waiting for next session"
+MARKET_OPEN = time(9, 15)
+FIRST5_READY = time(9, 20)
+MARKET_CLOSE = time(15, 30)
+
+
+def configured_pin() -> str:
+    return os.getenv("DASHBOARD_PIN", "1979")
+
+
+def is_trading_day(day: date) -> bool:
+    holidays = {
+        d.strip()
+        for d in os.getenv("NSE_HOLIDAYS", "").split(",")
+        if d.strip()
+    }
+    return day.weekday() < 5 and day.isoformat() not in holidays
+
+
+def next_trading_day(start: date) -> date:
+    day = start
+    while not is_trading_day(day):
+        day += timedelta(days=1)
+    return day
+
+
+def latest_prediction_date(payload: dict | None = None) -> date | None:
+    try:
+        latest = payload if payload is not None else latest_saved_prediction()
+        raw = latest.get("input_date")
+        return date.fromisoformat(str(raw)) if raw else None
+    except Exception:
+        return None
+
+
+def current_market_state(now: datetime | None = None) -> dict:
+    global market_status
+    now = now or datetime.now(IST)
+    today = now.date()
+    current_time = now.time()
+    trading_day = is_trading_day(today)
+    latest_date = latest_prediction_date()
+    has_live_first5 = trading_day and latest_date == today and FIRST5_READY <= current_time <= MARKET_CLOSE
+
+    if not trading_day:
+        next_day = next_trading_day(today + timedelta(days=1))
+        status = "Market Closed"
+        detail = f"Next trading session is {next_day.isoformat()}."
+    elif current_time < time(9, 0):
+        status = "Waiting for 9:00 AM"
+        detail = "Market has not entered pre-open yet."
+    elif current_time < MARKET_OPEN:
+        status = "Market Pre-Open"
+        detail = "Market opens at 9:15 AM IST."
+    elif current_time < FIRST5_READY:
+        status = "Market Officially Opened"
+        detail = "Waiting for the first 5 one-minute bars."
+    elif current_time <= MARKET_CLOSE:
+        if market_status in {"Fetching T+5 Prediction Data...", "Prediction Failed"}:
+            status = market_status
+            detail = "The first-five-minute prediction job is still resolving."
+        elif has_live_first5:
+            status = "Prediction Ready"
+            detail = "Today's first-five-minute prediction is available."
+        else:
+            status = "Prediction Pending"
+            detail = "No current-session prediction has been generated yet."
+    else:
+        status = "Market Closed"
+        detail = "Trading session has ended."
+
+    return {
+        "market_status": status,
+        "market_detail": detail,
+        "is_trading_day": trading_day,
+        "session_date": today.isoformat(),
+        "latest_prediction_date": latest_date.isoformat() if latest_date else None,
+        "t5_available": has_live_first5,
+    }
+
+
+def attach_market_state(payload: dict) -> dict:
+    state = current_market_state()
+    payload.setdefault("data_status", {})
+    payload["data_status"].update(state)
+
+    latest = payload.get("latest") or {}
+    t5_available = bool(state["t5_available"] and latest.get("prediction"))
+    market_closed = state["market_status"] == "Market Closed"
+    unavailable_reason = "Market Closed" if market_closed else state["market_status"]
+    payload["predictions"] = {
+        "tomorrow": {
+            "available": False,
+            "status": "Market Closed" if market_closed else "Pending",
+            "reason": "No next-session model is generated by this backend.",
+        },
+        "t5": {
+            "available": t5_available,
+            "status": "Ready" if t5_available else unavailable_reason,
+            "reason": None if t5_available else state["market_detail"],
+            "input_date": latest.get("input_date"),
+            "prediction": latest.get("prediction") if t5_available else None,
+            "prob_up": latest.get("prob_up") if t5_available else None,
+            "confidence": latest.get("confidence") if t5_available else None,
+        },
+    }
+    return payload
+
+
+def verify_pin(pin: str | None) -> None:
+    if not pin or not hmac.compare_digest(pin, configured_pin()):
+        raise HTTPException(status_code=401, detail="Invalid PIN")
+
+
+def verify_pin_header(
+    x_dashboard_pin: str | None = Header(default=None),
+    authorization: str | None = Header(default=None),
+) -> None:
+    pin = x_dashboard_pin
+    if not pin and authorization and authorization.lower().startswith("bearer "):
+        pin = authorization[7:].strip()
+    verify_pin(pin)
 
 async def daily_ist_refresh_loop() -> None:
     global market_status
     while True:
         # Wait until 9:00 AM IST
         await asyncio.sleep(seconds_until_next_ist_run(time(9, 0)))
+        if not is_trading_day(datetime.now(IST).date()):
+            market_status = "Market Closed"
+            continue
         market_status = "Market Pre-Open"
         print("[scheduler] 9:00 AM IST - Market Pre-Open", flush=True)
 
@@ -60,7 +187,10 @@ async def start_scheduler() -> None:
     global market_status
     # Initialize correct status on startup based on current time
     now = datetime.now(IST).time()
-    if now < time(9, 0):
+    today = datetime.now(IST).date()
+    if not is_trading_day(today):
+        market_status = "Market Closed"
+    elif now < time(9, 0):
         market_status = "Waiting for 9:00 AM"
     elif now < time(9, 15):
         market_status = "Market Pre-Open"
@@ -83,22 +213,23 @@ def root() -> dict[str, str]:
 
 
 @app.get("/dashboard")
-def dashboard() -> dict:
-    global market_status
-    payload = dashboard_payload()
-    if "data_status" not in payload:
-        payload["data_status"] = {}
-    payload["data_status"]["market_status"] = market_status
-    return payload
+def dashboard(_: None = Depends(verify_pin_header)) -> dict:
+    return attach_market_state(dashboard_payload())
+
+
+@app.post("/dashboard/unlock")
+def dashboard_unlock(body: dict = Body(default_factory=dict)) -> dict:
+    verify_pin(str(body.get("pin", "")))
+    return attach_market_state(dashboard_payload())
 
 
 @app.get("/cron/keepalive")
 def cron_keepalive() -> dict:
-    return {"status": "awake", "latest": latest_saved_prediction()}
+    return {"status": "awake", "market": current_market_state()}
 
 
 @app.get("/prediction/latest")
-def prediction_latest() -> dict:
+def prediction_latest(_: None = Depends(verify_pin_header)) -> dict:
     return latest_saved_prediction()