o _j @sdZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z mZmZmZer6ddlmZedZdd dZde dfdddZde dfdddZGdddZgdZdS)umHTTP session management — creation, warm-up, and proxy rotation. Key improvements over original: - Session is created ONCE per fetch operation, not once per retry attempt. - Warm-up failures are non-fatal and don't create new sessions. - Proxy rotation is deterministic (round-robin) not purely random. - Sessions are properly closed when context manager exits. ) annotationsN) TYPE_CHECKINGAny)requests)DEFAULT_WARMUP_TIMEOUT_S NSE_BASE_URLNSE_WARMUP_URLS USER_AGENTS)Settingszindiaopt.session proxy_urls list[str]returndict[str, str] | NonecCs|sdSt|}||dS)zReturn a randomly selected proxy dict, or ``None`` if no proxies configured. Args: proxy_urls: List of proxy URL strings. Returns: ``{"http": url, "https": url}`` or ``None``. N)httphttps)randomchoice)r urlrP/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/indiaopt/http/session.py pick_proxy s   rTlist[str] | Nonewarmup_timeoutfloatwarmupboolrc Cstjdd}|jttdddtddddd |r)t|}|r)|j ||r`t D]2}z|j ||d }t d ||jWntyY}z t d ||WYd }~nd }~wwtdq-|S)aCreate and warm up an NSE-compatible curl_cffi session. Performs two GET requests to NSE's landing pages so that the session accumulates the cookies NSE requires for API calls. Args: proxy_urls: Optional list of proxy URLs to rotate over. warmup_timeout: Timeout for each warm-up request in seconds. warmup: Set to ``False`` to skip warm-up (e.g. in tests). Returns: A ready-to-use :class:`curl_cffi.requests.Session`. chrome124 impersonate!application/json, text/plain, */*en-US,en;q=0.9zgzip, deflate, brz /option-chain keep-alivezno-cache) User-AgentAcceptAccept-LanguagezAccept-EncodingReferer Connectionz Cache-ControlPragmatimeoutuWarm-up GET %s → HTTP %dz,Warm-up request to %s failed (non-fatal): %sNg333333?) cffi_requestsSessionheadersupdaterrr rrproxiesrgetloggerdebug status_code Exceptiontimesleep)r rrsessionr.rrespexcrrrcreate_nse_session/s:    r9c Csddlm}tjdd}|jttdd|dd|r)t |}|r)|j ||rWz|j ||d }t d ||jW|StyV}z t d |WYd }~|Sd }~ww|S) a)Create a BSE-compatible curl_cffi session. Args: proxy_urls: Optional list of proxy URLs. warmup_timeout: Timeout for warm-up requests. warmup: Set to ``False`` to skip warm-up. Returns: A ready-to-use :class:`curl_cffi.requests.Session`. r) BSE_BASE_URLrrrr r!)r"r#r$r%r&r(uBSE warm-up GET %s → HTTP %dz"BSE warm-up failed (non-fatal): %sN)indiaopt.constantsr:r*r+r,r-rrr rr.r/r0r1r2r3)r rrr:r6r.r7r8rrrcreate_bse_sessioncs2    r<c@sNeZdZdZdddZddd Zdd d ZeddddZeddddZ dS)SessionManagerzContext manager that owns a session lifetime. Ensures the session is properly closed even if an exception occurs:: with SessionManager.for_nse(settings) as session: resp = session.get(url, timeout=15) r6rr NonecCs ||_dSN_session)selfr6rrr__init__s zSessionManager.__init__cCs|jSr?r@)rBrrr __enter__szSessionManager.__enter___objectcGs&z|jWdStyYdSwr?)rAcloser3)rBrErrr__exit__s  zSessionManager.__exit__NsettingsSettings | NonecC0|r|jng}|r |jnt}t||d}||S)z)r r)rErFr r>r?)rIrJr r=) __name__ __module__ __qualname____doc__rCrDrH classmethodrOrQrrrrr=s    r=)r=r<r9r)r r r r)r rrrrrr r)rU __future__rloggingrr4typingrr curl_cffirr*r;rrrr indiaopt.config.settingsr getLoggerr0rr9r<r=__all__rrrrs,      5 + ,