o _jg(@sUdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZedZGd d d e ZGd d d e ZeGd ddZGdddZdaded<eZ  ddddZgdZdS)uThread-safe circuit breaker. The circuit breaker prevents cascading failures by stopping fetch attempts when a symbol repeatedly fails, allowing the exchange to recover. States: * **CLOSED** — normal operation; requests pass through. * **OPEN** — failure threshold exceeded; requests are rejected immediately with :class:`~indiaopt.exceptions.CircuitOpenError`. * **HALF-OPEN** — backoff period elapsed; one probe request is allowed through. Success → CLOSED, failure → OPEN (with reset backoff). Each (exchange, symbol) pair has its own independent circuit state. Example:: breaker = CircuitBreaker(failure_threshold=5, backoff_seconds=300) # Inside your fetch loop: if breaker.is_open("NSE", "NIFTY"): raise CircuitOpenError(...) try: result = do_fetch() breaker.record_success("NSE", "NIFTY") except Exception: breaker.record_failure("NSE", "NIFTY") raise ) annotationsN) dataclass)Enumauto) NamedTuple)CircuitOpenErrorzindiaopt.circuitc@s"eZdZdZeZeZeZdS) CircuitStatezPossible states of a circuit.N)__name__ __module__ __qualname____doc__rCLOSEDOPEN HALF_OPENrrK/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/indiaopt/circuit.pyr.s  rc@seZdZUded<ded<dS) _CircuitKeystrexchangesymbolN)r r r __annotations__rrrrr6s  rc@sJeZdZUejZded<dZded<dZded<dZ ded <dd d Z dS) _CircuitEntryrstaterintfailuresfloat open_untillast_failure_timereturnNonecCstj|_d|_d|_dS)Nrr)rr rrrselfrrrresetBs z_CircuitEntry.resetNrr ) r r r rr rrrrrr#rrrrr;s    rc@szeZdZdZ  d(d)d d Zd*ddZd+ddZd,ddZd-ddZd-ddZ d-ddZ d.d d!Z d-d"d#Z d/d%d&Z d'S)0CircuitBreakeraThread-safe circuit breaker with CLOSED / OPEN / HALF-OPEN states. Args: failure_threshold: Consecutive failures before opening the circuit. backoff_seconds: How long (seconds) the circuit stays open before moving to HALF-OPEN for a probe attempt. Usage:: breaker = CircuitBreaker(failure_threshold=5, backoff_seconds=300) key = ("NSE", "NIFTY") if breaker.is_open(*key): raise CircuitOpenError("NIFTY", open_until=..., failures=...) try: result = do_fetch() breaker.record_success(*key) except Exception: breaker.record_failure(*key) raise r@failure_thresholdrbackoff_secondsrrr cCs@|dkrtd|dkrtd||_||_i|_t|_dS)Nzfailure_threshold must be >= 1rzbackoff_seconds must be > 0) ValueError_failure_threshold_backoff_seconds _circuits threadingLock_lock)r"r(r)rrr__init__`szCircuitBreaker.__init__keyrrcCs ||jvr t|j|<|j|S)zBReturn (or create) the entry for *key*. Must be called under lock.)r.r)r"r3rrr _get_entryps   zCircuitBreaker._get_entryrrrboolcCst||}|jJ||}|jtjkr! WddS|jtjkrMt }||j krDtj |_t d|| WddS WddS WddS1sYwYdS)zReturn ``True`` if the circuit is open for *(exchange, symbol)*. When the backoff period has elapsed, transitions the circuit to HALF-OPEN and returns ``False`` so one probe attempt can proceed. NF7Circuit HALF-OPEN for %s/%s — allowing probe attempt.T)rupperr1r4rrr rtime monotonicrrloggerinfor"rrr3entrynowrrris_openxs*    $zCircuitBreaker.is_openrcCsHt||}|j||jWdS1swYdS)zBReturn the current :class:`CircuitState` for *(exchange, symbol)*.N)rr7r1r4rr"rrr3rrr get_states $zCircuitBreaker.get_statecCsxt||}|j&||}|jdks|jtjkr&t d|||j| WddS1s5wYdS)z}Record a successful fetch for *(exchange, symbol)*. Resets the circuit to CLOSED regardless of prior state. rzBCircuit CLOSED for %s/%s after successful fetch (had %d failures).N) rr7r1r4rrrr r:r;r#r"rrr3r=rrrrecord_successs  "zCircuitBreaker.record_successcCst||}|jc||}|jd7_t|_|jt j kr:t j |_t|j |_ td|||j n-|j|jkr_t j |_t|j |_ td|||j|j WddSWddSWddS1srwYdS)zwRecord a fetch failure for *(exchange, symbol)*. Opens the circuit when failures reach the threshold. r*zGCircuit re-OPENED for %s/%s after probe failure. Backing off for %.0fs.zBCircuit OPENED for %s/%s after %d failures. Backing off for %.0fs.N)rr7r1r4rr8r9rrrrrr-rr:warningr,rBrrrrecord_failures:    "zCircuitBreaker.record_failurecCsbt||}|j||jvr|j|Wdn1s#wYtd||dS)zManually reset the circuit for *(exchange, symbol)* to CLOSED. Useful in tests or operator-driven recovery scenarios. Nz!Circuit manually reset for %s/%s.)rr7r1r.r#r:r;r@rrrr#s zCircuitBreaker.resetcCsL|j|jD]}|q Wdn1swYtddS)z=Reset ALL circuits to CLOSED. Use with caution in production.NzAll circuits manually reset.)r1r.valuesr#r:r;)r"r=rrr reset_alls  zCircuitBreaker.reset_allcCst||}|j:||}|jtjkr>t}||j kr+t ||j |j |dtj |_t d||WddSWddS1sIwYdS)zRaise :class:`~indiaopt.exceptions.CircuitOpenError` if circuit is open. Convenience wrapper for the common check-and-raise pattern:: breaker.raise_if_open("NSE", "NIFTY") # raises or passes through )rrrrr6N)rr7r1r4rrrr8r9rrrrr:r;r<rrr raise_if_opens*   "zCircuitBreaker.raise_if_opendict[str, dict[str, object]]cCs>|jdd|jDWdS1swYdS)zReturn a snapshot of all circuit states for monitoring / metrics. Returns: Dict keyed by ``"EXCHANGE/SYMBOL"`` with state info. cSs4i|]\}}|jd|j|jj|j|jdqS)/)rrr)rrrnamerr).0r3r=rrr sz(CircuitBreaker.stats..N)r1r.itemsr!rrrstatss $zCircuitBreaker.statsNr&r')r(rr)rrr )r3rrr)rrrrrr5)rrrrrr)rrrrrr r$)rrI)r r r r r2r4r?rArCrEr#rGrHrOrrrrr%Hs      ! r%zCircuitBreaker | None_default_breakerr&r'r(rr)rrcCsNttdurt||daWdtSWdtS1s wYtS)zReturn the module-level shared :class:`CircuitBreaker`. Creates it on first call with the given parameters. Subsequent calls return the same instance regardless of parameters. Nr(r))_default_breaker_lockrQr%rRrrrget_default_breakers   rT)r%rrTrP)r(rr)rrr%)r __future__rloggingr/r8 dataclassesrenumrrtypingrindiaopt.exceptionsr getLoggerr:rrrr%rQrr0rSrT__all__rrrrs,       C