o _j3<@sdZddlmZddlZddlZddlmZddlmZddl m Z m Z ddl m Z mZddlmZmZmZmZdd lmZmZmZmZmZdd lmZdd lmZdd lmZdd l m!Z"e#dZ$dZ%GdddZ&dgZ'dS)uNSEClient — high-level async client for NSE option chain data. Usage:: import asyncio from indiaopt import NSEClient async def main(): async with NSEClient() as client: result = await client.fetch_option_chain("NIFTY", is_index=True) print(f"Spot: {result.spot_price}, ATM: {result.atm_strike}") for row in result.atm_window(n=5): print(f"{row.strike:>8} CE OI: {row.call_oi:>10} PE OI: {row.put_oi:>10}") asyncio.run(main()) ) annotationsN)ThreadPoolExecutor)Any)CircuitBreakerget_default_breaker)Settings get_settings) NSE_BASE_URLNSE_CONTRACT_INFO_PATHNSE_OPTION_CHAIN_PATHRATE_LIMIT_STATUS_CODES) FetchErrorFetchTimeoutError NetworkError ParseErrorRateLimitError) RetryPolicy)create_nse_session)OptionChainResult)nsezindiaopt.exchanges.nseNSEc@seZdZdZ  d3d4d d Zd5d d Zd6ddZd7ddZdddd8ddZdddd9dd Z d:d"d#Z d;d$d%Z dd/d0Zd:d1d2ZdS)? NSEClientaZAsync client for NSE option chain and market data. Manages its own session lifetime, thread pool, circuit breaker, and retry policy. Supports both context-manager and direct usage. Args: settings: Optional :class:`~indiaopt.config.settings.Settings` override. Uses global settings (from env/``.env``) when not provided. breaker: Optional :class:`~indiaopt.circuit.CircuitBreaker` override. Shares the global default when not provided. Context manager (recommended):: async with NSEClient() as client: result = await client.fetch_option_chain("NIFTY") Direct usage (must call :meth:`close` manually):: client = NSEClient() try: result = await client.fetch_option_chain("NIFTY") finally: await client.close() NsettingsSettings | NonebreakerCircuitBreaker | NonereturnNonecCsP|pt|_|pt|jj|jjd|_t|j|_t |jj dd|_ d|_ dS)N)failure_thresholdbackoff_secondsz indiaopt-nse) max_workersthread_name_prefix) r _settingsrcircuit_failure_thresholdcircuit_backoff_seconds_breakerr from_settings _retry_policyrthread_pool_workers_pool_session)selfrrr,Q/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/indiaopt/exchanges/nse.py__init__Ls  zNSEClient.__init__cs|SNr,r+r,r,r- __aenter___szNSEClient.__aenter___objectcs|IdHdSr/)close)r+r2r,r,r- __aexit__bszNSEClient.__aexit__csF|jdurz|jWn tyYnwd|_|jjdddS)z-Release all resources (session, thread pool).NF)wait)r*r4 Exceptionr)shutdownr0r,r,r-r4es  zNSEClient.closeT)is_indexexpirysymbolstrr9boolr: str | Nonedict[str, Any]cs|}|jt|t}ztj||j |j ||||j j dIdHWStj y@|jt|t||j j tdw)z=Fetch the unparsed raw JSON dictionary for *symbol* from NSE.timeoutNr; timeout_sexchange)upperstripr% raise_if_openEXCHANGEasyncioget_running_loopwait_forrun_in_executorr) _fetch_syncr" async_timeout TimeoutErrorrecord_failurer)r+r;r9r:loopr,r,r-fetch_raw_option_chainqs.  z NSEClient.fetch_raw_option_chainrcs|}|jt|t}ztj||j |j ||||j j dIdH}Wntj yA|jt|t||j j tdwt||S)uFetch the full option chain for *symbol*. Args: symbol: NSE trading symbol (e.g. ``"NIFTY"``, ``"RELIANCE"``). is_index: ``True`` for index instruments (NIFTY, BANKNIFTY, …), ``False`` for equity derivatives. expiry: Specific expiry date string (``"27-Jun-2024"`` format). Uses the nearest expiry when not provided. Returns: :class:`~indiaopt.models.option_chain.OptionChainResult` Raises: :class:`~indiaopt.exceptions.CircuitOpenError`: Circuit is open. :class:`~indiaopt.exceptions.FetchTimeoutError`: Exceeded async timeout. :class:`~indiaopt.exceptions.FetchError`: All retries exhausted. r@NrB)rErFr%rGrHrIrJrKrLr)rMr"rNrOrPr nse_parserparse_option_chain)r+r;r9r:rQrawr,r,r-fetch_option_chains0   zNSEClient.fetch_option_chain list[str]cs\|}t}ztj||j|j||jj dIdH}W|Stj y-gYSw)zReturn available expiry dates for *symbol*. Args: symbol: NSE trading symbol. Returns: List of expiry date strings in ``"DD-Mon-YYYY"`` format. r@N) rErFrIrJrKrLr)_fetch_expiry_dates_syncr"rNrO)r+r;rQresultr,r,r-get_expiry_datess   zNSEClient.get_expiry_datescCs|jt|dS)zManually reset the circuit breaker for *symbol*. Use after manually resolving an issue (e.g. IP unblock, network fix). N)r%resetrH)r+r;r,r,r- reset_circuitszNSEClient.reset_circuitdict[str, dict[str, object]]cCs |jS)z?Return a snapshot of all circuit breaker states for monitoring.)r%statsr0r,r,r- circuit_statss zNSEClient.circuit_statsrcCs&|jdurt|jj|jjd|_|jS)z.Return existing session or create a fresh one.N) proxy_urlswarmup_timeout)r*rr"r`rar0r,r,r-_get_or_create_sessions z NSEClient._get_or_create_sessionc Cs<|}|rdnd}|dur|||}ttd|d|d|}d}t|jjD]V}z|j||jj d} | j dkrz| } Wn;t y~} z/t d ||d | d|_|}td | |t|d t| d ru| jddd| d d| d} ~ ww|jt|t dt|||d | WS| j tvrd} | jdp| jd} | rzt| } Wn tyYnwtd| j d||t| j |d | d}t d||d |jj| j |}||jjd kr|j|Wq+td| j d||t| j |d dttfy tyR}z:td|d||t|d dd}t d||d |jj|d|_|}||jjd krH|j|WYd}~q+d}~wt y}z$|}t d||d |jj|||jjd krx|j|WYd}~q+d}~ww|j t|td|jjd ||t|jjd!d|)"zvSynchronous fetch with retry, circuit breaker, and session reuse. This runs in a thread pool worker. IndicesEquitiesNz?type=z&symbol=z&expiry=r@z?JSON decode failed for %s (attempt %d): %s. Refreshing session.z.Invalid JSON from NSE (likely cookie expiry): text)r;rDattempts raw_previewz-Fetched %s/%s expiry=%s HTTP 200 (attempt %d)z Retry-Afterz retry-afterzHTTP z from NSE for )r;rD status_coderi retry_afterz:Rate-limited for %s (attempt %d/%d, HTTP %d). Backing off.zUnexpected HTTP z for )r;rDrkrizNetwork error for z: z3Check network connectivity and proxy configuration.)r;rDrirecoveryz(Network error for %s (attempt %d/%d): %sz+Unexpected error for %s (attempt %d/%d): %szAll z attempts failed for zrr?)r;r<r9r=r:r>rr)r;r<rrW)r;r<rr)rr])rr)r~rr;r<rr<)__name__ __module__ __qualname____doc__r.r1r5r4rRrVrZr\r_rbrMrnrXr,r,r,r-r2s,   # 1     r)(r __future__rrIloggingconcurrent.futuresrtypingrindiaopt.circuitrrindiaopt.config.settingsrrindiaopt.constantsr r r r indiaopt.exceptionsr rrrrindiaopt.http.retryrindiaopt.http.sessionrindiaopt.models.option_chainrindiaopt.parsersrrS getLoggerrtrHr__all__r,r,r,r-s(         n