Upload 42 files

app.py

@@ -211,12 +211,24 @@ def current_market_state(now: datetime | None = None) -> dict:
     }
 
 
-def attach_market_state(payload: dict) -> dict:
-    state = current_market_state()
-    payload.setdefault("data_status", {})
-    payload["data_status"].update(state)
-
-    t5_latest = payload.get("latest") or {}
+def attach_market_state(payload: dict) -> dict:
+    state = current_market_state()
+    payload.setdefault("data_status", {})
+    payload["data_status"].update(state)
+    try:
+        payload["nifty_quote"] = kotak_neo_manager.fetch_nifty50_quote()
+        payload["nifty_quote_error"] = None
+    except KotakNeoSessionRequired as exc:
+        payload["nifty_quote"] = None
+        payload["nifty_quote_error"] = {"status": 401, "message": str(exc)}
+    except KotakNeoConfigError as exc:
+        payload["nifty_quote"] = None
+        payload["nifty_quote_error"] = {"status": 503, "message": str(exc)}
+    except KotakNeoError as exc:
+        payload["nifty_quote"] = None
+        payload["nifty_quote_error"] = {"status": 502, "message": str(exc)}
+
+    t5_latest = payload.get("latest") or {}
     tomorrow_latest = payload.get("tomorrow_latest") or {}
     tplus1_latest = payload.get("tplus1_latest") or {}
     t5_available = bool(state["t5_available"] and t5_latest.get("prediction"))

kotak_neo.py

@@ -29,6 +29,8 @@ TOTP_LOGIN_PATH = "login/1.0/tradeApiLogin"
 TOTP_VALIDATE_PATH = "login/1.0/tradeApiValidate"
 DEFAULT_TIMEOUT_SECONDS = 20
 ACCOUNT_TIMEOUT_SECONDS = 7
+NIFTY_QUOTE_TIMEOUT_SECONDS = 2.5
+NIFTY_QUOTE_CACHE_SECONDS = 1.0
 DATA_DIR = Path(__file__).resolve().parent / "data"
 KOTAK_ACTIVITY_LOG_PATH = DATA_DIR / "kotak_activity_log.txt"
 NIFTY_1M_PATH = DATA_DIR / "nifty50_1m.parquet"
@@ -472,7 +474,7 @@ class KotakNeoManager:
         with self._lock:
             if not force_refresh:
                 cached = self._quote_cache.get(cache_key)
-                if cached and (monotonic() - float(cached.get("stored_at_monotonic") or 0.0)) < 0.25:
+                if cached and (monotonic() - float(cached.get("stored_at_monotonic") or 0.0)) < NIFTY_QUOTE_CACHE_SECONDS:
                     return dict(cached["payload"])
             context = self._context_locked()
 
@@ -485,7 +487,7 @@ class KotakNeoManager:
                     "instrument_token": str(reference["quote_instrument_token"]),
                 }
             ],
-            timeout=ACCOUNT_TIMEOUT_SECONDS,
+            timeout=NIFTY_QUOTE_TIMEOUT_SECONDS,
         )
         items = _extract_items(quote_payload)
         if not items: