File: //usr/local/lib/python3.10/dist-packages/tenacity/asyncio/__pycache__/__init__.cpython-310.pyc
o
���g] � @ sP d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d dlm Z d d lm Z d d
lm
Z
d d lmZ d d
lmZ d dlmZ d dlmZ d dlmZ dd lmZ ejryd dlmZ d dlmZ e�d�Zejdej dej ej f d�Z de dej d fdd�Z!G d d � d e�Z"g d �Z#dS ) � N)�AttemptManager)� BaseRetrying)� DoAttempt)�DoSleep)�RetryCallState)�
RetryError)�
after_nothing)�before_nothing)�_utils� )�
RetryBaseT)� retry_all)� retry_any)�retry_if_exception)�retry_if_result� )� StopBaseT)� WaitBaseT�WrappedFnReturnT� WrappedFn.)�bound�seconds�returnc C sB dt jv rdd l}dd l}|�� dkr|�| �S dd l}|�| �S )N�trior )�sys�modulesr �sniffio�current_async_library�sleep�asyncio)r r r r � r �D/usr/local/lib/python3.10/dist-packages/tenacity/asyncio/__init__.py�_portable_async_sleep1 s
r" c s� e Zd Zeejjej�� e� � e
e dde df
de
je
jeef ge
jde
jd f f dddddd d
e
jd ge
jde
jd f f d e
jd ge
jde
jd f f d
e
je
jd ge
jde
jd f f dede
jd de
je
jd ge
je
je
je
j f f ddf� fdd�
Zdede
jde
jdefdd�Zde
jde
jf ddfdd �Zd.d d �Z d.d d!�Z d.d"d#�Z d d de
je e e
jf fd$d%�Z!de
j"e#ddf fd&d'�Z$d/d(d)�Z%de#fd*d+�Z&dedef� fd,d-� Z'� Z(S )0�
AsyncRetryingNFr �stopr �waitr �retryz#t.Union[SyncRetryBaseT, RetryBaseT]�beforer �after� before_sleep�reraise�retry_error_clsr �retry_error_callbackr c s$ t � j||||||||| |
d�
d S )N)
r r$ r% r& r' r( r) r* r+ r, )�super�__init__) �selfr r$ r% r& r' r( r) r* r+ r, �� __class__r r! r. D s
�zAsyncRetrying.__init__�fn�args�kwargsc � s� �| � � t| |||d�} | j|d�I d H }t|t�r?z ||i |��I d H }W n ty8 |�t�� � Y nw |� |� nt|t
�rQ|� � | � |�I d H n|S q)N)� retry_objectr2 r3 r4 T�� retry_state)
�beginr �iter�
isinstancer �
BaseException�
set_exceptionr �exc_info�
set_resultr �prepare_for_next_attemptr )r/ r2 r3 r4 r7 �do�resultr r r! �__call__h s" �
�
�zAsyncRetrying.__call__.c C s | j j�t�|�� d S �N)�
iter_state�actions�appendr
�wrap_to_async_func)r/ r2 r r r! �_add_action_func} s z AsyncRetrying._add_action_funcr7 c � s �t �| j�|�I d H | j_d S rC )r
rG r& rD �retry_run_result�r/ r7 r r r! �
_run_retry� s �
�zAsyncRetrying._run_retryc � s. �| j rt�| j �|�I d H }nd}||_d S )Ng )r% r
rG �upcoming_sleep)r/ r7 r r r r! � _run_wait� s
�
zAsyncRetrying._run_waitc � s, �|j | jd< t�| j�|�I d H | j_d S )N�delay_since_first_attempt)�seconds_since_start�
statisticsr
rG r$ rD �stop_run_resultrJ r r r! � _run_stop� s
�
�zAsyncRetrying._run_stopc � s0 �| � |� d }| jjD ] }||�I d H }q |S rC )� _begin_iterrD rE )r/ r7 rA �actionr r r! r9 � s �
zAsyncRetrying.iterc C s t d��)Nz$AsyncRetrying object is not iterable)� TypeError�r/ r r r! �__iter__� s zAsyncRetrying.__iter__c C s | � � t| d di d�| _| S )Nr )r2 r3 r4 )r8 r � _retry_staterV r r r! � __aiter__� s zAsyncRetrying.__aiter__c � sf � | j | jd�I d H }|d u rt�t|t�r t| jd�S t|t�r0| j�� | �|�I d H nt�q)NTr6 ) r9 rX �StopAsyncIterationr: r r r r? r )r/ r@ r r r! � __anext__� s �
�zAsyncRetrying.__anext__c sZ t � ���}t��tjd �dtjdtjdtjf� ��fdd� �� �� _|j� _i � _� S )N)� __defaults__�__kwdefaults__r3 r4 r c � s. ��� � }|j� _|�g| �R i |��I d H S rC )�copyrP )r3 r4 r^ ��
async_wrappedr2 r/ r r! r` � s � z*AsyncRetrying.wraps.<locals>.async_wrapped) r- �wraps� functools�WRAPPER_ASSIGNMENTS�t�Anyr& �
retry_withrP )r/ r2 �wrappedr0 r_ r! ra � s
�&zAsyncRetrying.wraps)r7 r r N)r r# ))�__name__�
__module__� __qualname__r" �tenacityr$ �
stop_neverr% � wait_none�retry_if_exception_typer r r rd �Callable�Union�int�float� Awaitable�Optional�bool�Typere r. r r rB rH rK rM rR r r r9 � Generatorr rW rY r[ ra �
__classcell__r r r0 r! r# C s� �"������� �� ����"���$���
�
�
�
r# )r
r r r r r# )$rb r �typingrd rk r r r r r r r r r
r& r r
r r r �SyncRetryBaseT�
TYPE_CHECKING�
tenacity.stopr �
tenacity.waitr �TypeVarr ro rs re r rr r" r# �__all__r r r r! �<module> s8