Upload 42 files

app.py

@@ -452,6 +452,18 @@ def kotak_account() -> dict:
         raise HTTPException(status_code=502, detail=str(exc)) from exc
 
 
+@app.get("/kotak/quote/nifty50")
+def kotak_nifty50_quote() -> dict:
+    try:
+        return kotak_neo_manager.fetch_nifty50_quote()
+    except KotakNeoConfigError as exc:
+        raise HTTPException(status_code=503, detail=str(exc)) from exc
+    except KotakNeoSessionRequired as exc:
+        raise HTTPException(status_code=401, detail=str(exc)) from exc
+    except KotakNeoError as exc:
+        raise HTTPException(status_code=502, detail=str(exc)) from exc
+
+
 @app.get("/kotak/activity-log")
 def kotak_activity_log() -> dict:
     try:

kotak_neo.py

@@ -6,6 +6,7 @@ from csv import DictReader
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from datetime import datetime, timezone
 from pathlib import Path
+from time import monotonic
 from typing import Any
 from urllib.parse import quote
 
@@ -133,6 +134,7 @@ class KotakNeoManager:
         self.activity_log_path.parent.mkdir(parents=True, exist_ok=True)
         self._seen_activity_keys: set[str] = set()
         self._scrip_cache: dict[str, list[dict[str, str]]] = {}
+        self._quote_cache: dict[str, dict[str, Any]] = {}
         self._load_existing_activity_keys()
         self._clear_session_locked()
 
@@ -376,6 +378,67 @@ class KotakNeoManager:
             "quotes": list(quote_map.values()),
         }
 
+    def fetch_nifty50_quote(self, *, force_refresh: bool = False) -> dict[str, Any]:
+        cache_key = "nifty50_quote"
+        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)) < 1.0:
+                    return dict(cached["payload"])
+            context = self._context_locked()
+
+        reference = self._resolve_nifty50_reference(context)
+        quote_payload = self._fetch_quotes_with_context(
+            context,
+            [
+                {
+                    "exchange_segment": "nse_cm",
+                    "instrument_token": str(reference["quote_instrument_token"]),
+                }
+            ],
+            timeout=ACCOUNT_TIMEOUT_SECONDS,
+        )
+        items = _extract_items(quote_payload)
+        if not items:
+            raise KotakNeoError("Kotak Neo did not return a NIFTY 50 quote.")
+
+        item = items[0]
+        payload = {
+            "symbol": "NIFTY 50",
+            "exchange_segment": "nse_cm",
+            "instrument_token": _first_text(
+                item.get("instrument_token"),
+                item.get("instrumentToken"),
+                item.get("tk"),
+                reference.get("master_instrument_token"),
+                reference["quote_instrument_token"],
+            ),
+            "display_name": _first_text(item.get("trading_symbol"), item.get("ts"), item.get("name"), "NIFTY 50"),
+            "last_traded_price": _first_number(item.get("last_traded_price"), item.get("ltp"), item.get("iv")),
+            "close": _first_number(item.get("close"), item.get("c"), item.get("ic")),
+            "change": _first_number(item.get("change"), item.get("cng")),
+            "change_pct": _first_number(item.get("net_change_percentage"), item.get("nc")),
+            "open": _first_number(item.get("openingPrice"), item.get("open"), item.get("o")),
+            "high": _first_number(item.get("highPrice"), item.get("high"), item.get("h")),
+            "low": _first_number(item.get("lowPrice"), item.get("low"), item.get("l")),
+            "exchange_feed_time": _first_text(item.get("tvalue"), item.get("updRecvTm"), item.get("hsUpTm")),
+            "as_of": _utc_now_iso(),
+            "source": {
+                "quote_api": QUOTE_PATH_TEMPLATE,
+                "master_scrip_verified": bool(reference.get("master_record_found")),
+                "instrument_lookup": reference.get("lookup_mode"),
+                "master_symbol_name": reference.get("master_symbol_name"),
+                "master_trading_symbol": reference.get("master_trading_symbol"),
+            },
+        }
+
+        with self._lock:
+            self._quote_cache[cache_key] = {
+                "stored_at_monotonic": monotonic(),
+                "payload": payload,
+            }
+        return payload
+
     def _ensure_authenticated_locked(self) -> None:
         if not self.edit_token or not self.edit_sid or not self.base_url:
             raise KotakNeoSessionRequired("Kotak Neo session is not authenticated.")
@@ -710,6 +773,34 @@ class KotakNeoManager:
         self._scrip_cache[key] = rows
         return rows
 
+    def _resolve_nifty50_reference(self, context: dict[str, str]) -> dict[str, Any]:
+        candidates = self._load_scrip_candidates(context, "nse_cm")
+        match = None
+        aliases = {
+            "NIFTY 50",
+            "NIFTY50",
+            "NIFTY 50 INDEX",
+            "NIFTY",
+        }
+        for item in candidates:
+            candidate_values = {
+                str(item.get("pSymbolName") or "").strip().upper(),
+                str(item.get("pTrdSymbol") or "").strip().upper(),
+                str(item.get("pSymbol") or "").strip().upper(),
+            }
+            if aliases & candidate_values:
+                match = item
+                break
+
+        return {
+            "quote_instrument_token": "Nifty 50",
+            "lookup_mode": "index-name-direct" if match is None else "master-scrip-verified-index-name-direct",
+            "master_record_found": match is not None,
+            "master_instrument_token": _first_text(match.get("pSymbol")) if match else None,
+            "master_symbol_name": _first_text(match.get("pSymbolName")) if match else None,
+            "master_trading_symbol": _first_text(match.get("pTrdSymbol")) if match else None,
+        }
+
     def _build_quote_map(self, payload: dict[str, Any]) -> dict[str, dict[str, Any]]:
         items = _extract_items(payload)
         quote_map: dict[str, dict[str, Any]] = {}