a
    i,Me                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm	Z	 d dlm
Z
 d dlmZ edZejd	ejd
ejej f dZG dd deZdS )    N)sleep)AttemptManager)BaseRetrying)	DoAttempt)DoSleep)RetryCallStateWrappedFnReturnT	WrappedFn.)boundc                       s   e Zd ZU ejegejej f ed< e	fejegejej f ejdd fddZ
eejejedddZejeddf d	d
dZd d	ddZed	ddZeed fddZ  ZS )AsyncRetryingr   N)r   kwargsreturnc                    s   t  jf i | || _d S N)super__init__r   )selfr   r   	__class__ L/var/www/webapp/lib/python3.9/site-packages/pip/_vendor/tenacity/_asyncio.pyr   $   s    zAsyncRetrying.__init__)fnargsr   r   c                    s   |    t| |||d}| j|d}t|trtz||i |I d H }W n  tyf   |t  Y q0 |	| qt|t
r|  | |I d H  q|S qd S )N)retry_objectr   r   r   retry_state)beginr   iter
isinstancer   BaseExceptionset_exceptionsysexc_info
set_resultr   prepare_for_next_attemptr   )r   r   r   r   r   doresultr   r   r   __call__(   s    

zAsyncRetrying.__call__)r   c                 C   s   t dd S )Nz$AsyncRetrying object is not iterable)	TypeErrorr   r   r   r   __iter__=   s    zAsyncRetrying.__iter__c                 C   s   |    t| d di d| _| S )Nr   )r   r   r   )r   r   _retry_stater(   r   r   r   	__aiter__@   s    zAsyncRetrying.__aiter__c                    sb   | j | jd}|d u rtq t|tr2t| jdS t|trX| j  | |I d H  q tq d S )Nr   )	r   r*   StopAsyncIterationr   r   r   r   r#   r   )r   r$   r   r   r   	__anext__E   s    


zAsyncRetrying.__anext__)r   r   c                    sF   t    t tjtjtjd fdd} j|_ j|_|S )N)r   r   r   c                     s    | i |I d H S r   r   )r   r   r   r   r   async_wrappedV   s    z*AsyncRetrying.wraps.<locals>.async_wrapped)r   wraps	functoolstAnyretry
retry_with)r   r   r/   r   r.   r   r0   R   s    zAsyncRetrying.wraps)__name__
__module____qualname__r2   Callablefloat	Awaitabler3   __annotations__r   r   r	   r   r&   	Generatorr   r)   r+   r-   r0   __classcell__r   r   r   r   r   !   s   
0
r   )r1   r    typingr2   Zasyncior   pip._vendor.tenacityr   r   r   r   r   TypeVarr   r9   r;   r3   r	   r   r   r   r   r   <module>   s   
 