/
usr
/
lib
/
python3.9
/
site-packages
/
__pycache__
/
Upload File
HOME
a ��e3/ � @ s� d dl mZ d dlmZmZ d dlmZ d dlmZmZm Z d dl m Z m Z d dl mZ d dlmZmZ eefZdZd Zd Zd d � Zd d� ZG dd� de�ZG dd� de �Z G dd� de�Z dddddefdd�Z dS )� )�iscoroutinefunction)�datetime� timedelta)�wraps)�isgeneratorfunction�isasyncgenfunction�isclass)�ceil�floor)� monotonic)�AnyStr�Iterable�closed�openZ half_openc s � fdd�}|S )zOBuild a predicate function that checks if an exception is a subtype from a listc s t | � �S �N)� issubclass)Z thrown_type�_�� exc_types� �2/usr/lib/python3.9/site-packages/circuitbreaker.py� matches_types s z(in_exception_list.<locals>.matches_typesr )r r r r r �in_exception_list s r c C sv t | �r t| t�r t| �}nVz&t| � t| t�r8td��t| � }W n. typ t | �r`t | �rhtd��| }Y n0 |S )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 � Exceptionr �iter� isinstance� STRING_TYPES� ValueError� TypeError�callable)�expected_exceptionZfailure_predicater r r �build_failure_predicate s r! c @ s� e Zd ZdZdZeZdZd2dd�Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zed d � �Zed d!� �Zed"d#� �Zed$d%� �Zed&d'� �Zed(d)� �Zed*d+� �Z ed,d-� �Z ed.d/� �Z d0d1� Z dS )3�CircuitBreaker� � Nc C s� d| _ d| _|p| j| _|p | j| _|sTzt| �jd }W n tyR t j }Y n0 t |�| _ |pf| 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 Nr �EXPECTED_EXCEPTION)� _last_failure�_failure_count�FAILURE_THRESHOLD�_failure_threshold�RECOVERY_TIMEOUT�_recovery_timeout�type�__dict__�KeyErrorr" r% r! � is_failure�FALLBACK_FUNCTION�_fallback_function�_name� STATE_CLOSED�_stater �_opened)�self�failure_threshold�recovery_timeoutr �name�fallback_functionr r r �__init__@ s zCircuitBreaker.__init__c C s | � |�S r )�decorate)r6 �wrappedr r r �__call__k s zCircuitBreaker.__call__c C s d S r r �r6 r r r � __enter__n s zCircuitBreaker.__enter__c C s, |r | � ||�r || _| �� n| �� dS )NF)r/ r& � _CircuitBreaker__call_failed� _CircuitBreaker__call_succeeded)r6 �exc_type� exc_valueZ _tracebackr r r �__exit__q s zCircuitBreaker.__exit__c C s` | j du r2z |j| _ W n ty0 |j| _ Y n0 t�| � t|�sLt|�rV| �|�S | � |�S )z; Applies the circuit breaker to a function N) r2 � __qualname__�AttributeError�__name__�CircuitBreakerMonitor�registerr r �_decorate_async�_decorate_sync)r6 �functionr r r r< z s zCircuitBreaker.decoratec s< t � �� �fdd��}t � �� �fdd��}t� �r8|S |S )Nc s< �j r$�jr �j| i |��S t����j� g| �R i |��S r )�openedr: �CircuitBreakerError�call��args�kwargs�rM r6 r r �wrapper� s z.CircuitBreaker._decorate_sync.<locals>.wrapperc ? sP �j r.�jr&�j| i |��E d H d S t����j� g| �R i |��E d H d S r )rN r: rO �call_generatorrQ rT r r � gen_wrapper� s z2CircuitBreaker._decorate_sync.<locals>.gen_wrapper)r r )r6 rM rU rW r rT r rL � s z CircuitBreaker._decorate_syncc s< t � �� �fdd��}t � �� �fdd��}t� �r8|S |S )Nc � sH �j r*�jr"�j| i |��I d H S t����j� g| �R i |��I d H S r )rN r: rO � call_asyncrQ rT r r �awrapper� s z0CircuitBreaker._decorate_async.<locals>.awrapperc sp �j r>�jr6�j| i |��2 z3 d H W }|V q 6 d S t����j� g| �R i |��2 z3 d H W }|V qV6 d S r )rN r: rO �call_async_generator)rR rS �elrT r r � gen_awrapper� s $z4CircuitBreaker._decorate_async.<locals>.gen_awrapper)r r )r6 rM rY r\ r rT r rK � s z CircuitBreaker._decorate_asyncc O s6 | � ||i |��W d � S 1 s(0 Y dS )z� Calls the decorated function and applies the circuit breaker rules on success or failure :param func: Decorated function Nr �r6 �funcrR rS r r r rP � s zCircuitBreaker.callc o sB | �* ||i |��D ] }|V qW d � n1 s40 Y dS )z� Calls the decorated generator function and applies the circuit breaker rules on success or failure :param func: Decorated generator function Nr �r6 r^ rR rS r[ r r r rV � s z CircuitBreaker.call_generatorc � s< | �$ ||i |��I dH W d � S 1 s.0 Y dS )z� Calls the decorated async function and applies the circuit breaker rules on success or failure :param func: Decorated async function Nr r] r r r rX � s zCircuitBreaker.call_asyncc O sL | �4 ||i |��2 z3 dH W }|V q6 W d � n1 s>0 Y dS )z� Calls the decorated async generator function and applies the circuit breaker rules on success or failure :param func: Decorated async generator function Nr r_ r r r rZ � s z#CircuitBreaker.call_async_generatorc C s t | _d| _d| _dS )zR Close circuit after successful execution and reset failure count Nr )r3 r4 r& r' r? r r r Z__call_succeeded� s z CircuitBreaker.__call_succeededc C s, | j d7 _ | j | jkr(t| _t� | _dS )zO Count failure and open circuit, if threshold has been reached � N)r' r) � STATE_OPENr4 r r5 r? r r r Z __call_failed� s z CircuitBreaker.__call_failedc C s | j tkr| jdkrtS | j S �Nr )r4 ra �open_remaining�STATE_HALF_OPENr? r r r �state� s zCircuitBreaker.statec C s t �� t| jd� S )zq The approximate datetime when the circuit breaker will try to recover :return: datetime )Zseconds)r Zutcnowr rc r? r r r � open_until� s zCircuitBreaker.open_untilc C s* | j | j t� }|dkr"t|�S t|�S )zk Number of seconds remaining, the circuit breaker stays in OPEN state :return: int r )r5 r+ r r r )r6 Zremainr r r rc � s z CircuitBreaker.open_remainingc C s | j S r )r' r? r r r � failure_count s z CircuitBreaker.failure_countc C s | j tkS r )re r3 r? r r r r s zCircuitBreaker.closedc C s | j tkS r )re ra r? r r r rN s zCircuitBreaker.openedc C s | j S r �r2 r? r r r r9 s zCircuitBreaker.namec C s | j S r )r& r? r r r � last_failure s zCircuitBreaker.last_failurec C s | j S r )r1 r? r r r r: s z CircuitBreaker.fallback_functionc O s | j S r rh �r6 rR rS r r r �__str__ s zCircuitBreaker.__str__)NNNNN) rH � __module__rF r( r* r r% r0 r; r> r@ rE r< rL rK rP rV rX rZ rB rA �propertyre rf rc rg r rN r9 ri r: rk r r r r r" : sR � + r" c s$ e Zd Z� fdd�Zdd� Z� ZS )rO c s t t| �j|i |�� || _dS )zf :param circuit_breaker: :param args: :param kwargs: :return: N)�superrO r; �_circuit_breaker)r6 �circuit_breakerrR rS �� __class__r r r; s z CircuitBreakerError.__init__c O s* d| j j| j j| j jt| j j�| j jf S )NzMCircuit "%s" OPEN until %s (%d failures, %d sec remaining) (last_failure: %r))ro r9 rf rg �roundrc ri rj r r r rk * s �zCircuitBreakerError.__str__)rH rl rF r; rk � __classcell__r r rq r rO s rO c @ s� e Zd Zi Zedd� �Zeed�dd��Zeee d�dd��Z ee e d�d d ��Z eee d�d d ��Z eee d�d d��ZdS )rI c C s || j |j< d S r )�circuit_breakersr9 )�clsrp r r r rJ 7 s z CircuitBreakerMonitor.register)�returnc C s t t| �� ��dkS rb )�len�list�get_open�rv r r r � all_closed; s z CircuitBreakerMonitor.all_closedc C s | j �� S r )ru �valuesr{ r r r � get_circuits? s z"CircuitBreakerMonitor.get_circuits)r9 rw c C s | j �|�S r )ru �get)rv r9 r r r r C s zCircuitBreakerMonitor.getc c s | � � D ]}|jr|V qd S r )r~ rN �rv �circuitr r r rz G s z CircuitBreakerMonitor.get_openc c s | � � D ]}|jr|V qd S r )r~ r r� r r r � get_closedM s z CircuitBreakerMonitor.get_closedN)rH rl rF ru � classmethodrJ �boolr| r r" r~ r r rz r� r r r r rI 4 s rI Nc C s* t | �r|� �| �S || ||||d�S d S )N)r7 r8 r r9 r: )r r< )r7 r8 r r9 r: rv r r r r� T s �r� ) Zasyncior r r � functoolsr �inspectr r r Zmathr r �timer �typingr r �bytes�strr r3 ra rd r r! �objectr" r rO rI r� r r r r �<module> s. " f �