File: //lib/python3.9/site-packages/__pycache__/circuitbreaker.cpython-39.opt-1.pyc
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 �