o o]b(@s8ddlmZddlmZddlmZddlmZddlmZddlmZmZddl m Z m Z ddl m Z mZdd lmZmZzdd lmZWneyUdd lmZYnwzefZWn eyheefZYnwd Zd Zd ZddZddZGdddeZ Gddde!Z"GdddeZ#ddddde fddZ$dS))unicode_literals)division)print_function)absolute_import)wraps)datetime timedelta)isgeneratorfunctionisclass)AnyStrIterable)ceilfloor) monotonicclosedopenZ half_opencsfdd}|S)zOBuild a predicate function that checks if an exception is a subtype from a listcs t|SN) issubclass)Z thrown_type_ exc_types2usr/lib/python3.10/site-packages/circuitbreaker.py matches_types  z(in_exception_list..matches_typesr)rrrrrin_exception_lists rcCsxt|rt|trt|}|Szt|t|trtdt|}W|Sty;t |r2t|r6td|}Y|Sw)az Build a failure predicate_function. The returned function has the signature (Type[Exception], Exception) -> bool. Return value True indicates a failure in the underlying function. :param expected_exception: either an type of Exception, iterable of Exception types, or a predicate function. If an Exception type or iterable of Exception types, the failure predicate will return True when a thrown exception type matches one of the provided types. If a predicate function, it will just be returned as is. :return: callable (Type[Exception], Exception) -> bool z9expected_exception cannot be a string. Did you mean name?z1expected_exception does not look like a predicate) r r Exceptionriter isinstance STRING_TYPES ValueError TypeErrorcallable)expected_exceptionZfailure_predicaterrrbuild_failure_predicate&s   r$c@seZdZdZdZeZdZ     d*ddZddZ dd Z d d Z d d Z ddZ ddZddZddZeddZeddZeddZeddZeddZed d!Zed"d#Zed$d%Zed&d'Zd(d)ZdS)+CircuitBreakerNcCsd|_d|_|p |j|_|p|j|_|s*z t|jd}Wn ty)t j }Ynwt ||_ |p3|j |_||_t|_t|_dS)a Construct a circuit breaker. :param failure_threshold: break open after this many failures :param recovery_timeout: close after this many seconds :param expected_exception: either an type of Exception, iterable of Exception types, or a predicate function. :param name: name for this circuitbreaker :param fallback_function: called when the circuit is opened :return: Circuitbreaker instance :rtype: Circuitbreaker NrEXPECTED_EXCEPTION) _last_failure_failure_countFAILURE_THRESHOLD_failure_thresholdRECOVERY_TIMEOUT_recovery_timeouttype__dict__KeyErrorr%r(r$ is_failureFALLBACK_FUNCTION_fallback_function_name STATE_CLOSED_stater_opened)selffailure_thresholdrecovery_timeoutr#namefallback_functionrrr__init__Ns       zCircuitBreaker.__init__cCs ||Sr)decorate)r9wrappedrrr__call__yrzCircuitBreaker.__call__cCsdSrrr9rrr __enter__|szCircuitBreaker.__enter__cCs.|r|||r||_|dS|dS)NF)r2r)_CircuitBreaker__call_failed_CircuitBreaker__call_succeeded)r9exc_type exc_value _tracebackrrr__exit__s zCircuitBreaker.__exit__csnjdurzj_Wn tyj_Ynwttr&jnjt fdd}|S)z; Applies the circuit breaker to a function Ncs:jrjrj|i|Stg|Ri|Sr)openedr=CircuitBreakerError)argskwargscallfunctionr9rrwrappers z(CircuitBreaker.decorate..wrapper) r5 __qualname__AttributeError__name__CircuitBreakerMonitorregisterr call_generatorrOr)r9rPrQrrNrr?s     zCircuitBreaker.decoratecOs6|||i|WdS1swYdS)z Calls the decorated function and applies the circuit breaker rules on success or failure :param func: Decorated function Nr)r9funcrLrMrrrrOs $zCircuitBreaker.callcosF|||i|D]}|Vq WddS1swYdS)z Calls the decorated generator function and applies the circuit breaker rules on success or failure :param func: Decorated generator function Nr)r9rXrLrMelrrrrWs "zCircuitBreaker.call_generatorcCst|_d|_d|_dS)zR Close circuit after successful execution and reset failure count Nr)r6r7r)r*rBrrrZ__call_succeededs zCircuitBreaker.__call_succeededcCs0|jd7_|j|jkrt|_t|_dSdS)zO Count failure and open circuit, if threshold has been reached N)r*r, STATE_OPENr7rr8rBrrrZ __call_faileds   zCircuitBreaker.__call_failedcCs|jtkr |jdkr tS|jSNr)r7r[open_remainingSTATE_HALF_OPENrBrrrstateszCircuitBreaker.statecCstt|jdS)zq The approximate datetime when the circuit breaker will try to recover :return: datetime )seconds)rutcnowrr]rBrrr open_untilszCircuitBreaker.open_untilcCs*|j|jt}|dkrt|St|S)zk Number of seconds remaining, the circuit breaker stays in OPEN state :return: int r)r8r.rr r)r9Zremainrrrr]szCircuitBreaker.open_remainingcC|jSr)r*rBrrr failure_countzCircuitBreaker.failure_countcC |jtkSr)r_r6rBrrrr zCircuitBreaker.closedcCrfr)r_r[rBrrrrJrgzCircuitBreaker.openedcCrcrr5rBrrrr<rezCircuitBreaker.namecCrcr)r)rBrrr last_failurerezCircuitBreaker.last_failurecCrcr)r4rBrrrr=rez CircuitBreaker.fallback_functioncOrcrrhr9rLrMrrr__str__szCircuitBreaker.__str__)NNNNN)rT __module__rRr+r-rr(r3r>rArCrIr?rOrWrErDpropertyr_rbr]rdrrJr<rir=rkrrrrr%HsL +             r%cs$eZdZfddZddZZS)rKcs tt|j|i|||_dS)zf :param circuit_breaker: :param args: :param kwargs: :return: N)superrKr>_circuit_breaker)r9circuit_breakerrLrM __class__rrr>s zCircuitBreakerError.__init__cOs*d|jj|jj|jjt|jj|jjfS)NzMCircuit "%s" OPEN until %s (%d failures, %d sec remaining) (last_failure: %r))ror<rbrdroundr]rirjrrrrks zCircuitBreakerError.__str__)rTrlrRr>rk __classcell__rrrqrrKs  rKc@sXeZdZiZeddZeddZeddZeddZed d Z ed d Z d S)rUcCs||j|j<dSr)circuit_breakersr<)clsrprrrrVszCircuitBreakerMonitor.registercCstt|dkSr\)lenlistget_openrvrrr all_closedsz CircuitBreakerMonitor.all_closedcCs |jSr)ruvaluesrzrrr get_circuitss z"CircuitBreakerMonitor.get_circuitscCs |j|Sr)ruget)rvr<rrrr~ s zCircuitBreakerMonitor.getcc |D]}|jr |VqdSr)r}rJrvcircuitrrrry%  zCircuitBreakerMonitor.get_openccrr)r}rrrrr get_closed,rz CircuitBreakerMonitor.get_closedN) rTrlrRru classmethodrVr{r}r~ryrrrrrrUs     rUNcCs&t|r ||S||||||dS)N)r:r;r#r<r=)r"r?)r:r;r#r<r=rvrrrr4s r)% __future__rrrr functoolsrrrinspectr r typingr r mathr rtimer ImportErrorZ basestringr NameErrorbytesstrr6r[r^rr$objectr%rrKrUrrrrrsF          "3%