a
lj @ s d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
ej dej d Z
e d
ZG d d d eZdS )
N)sleep)AttemptManager) BaseRetrying) DoAttempt)DoSleep)RetryCallState WrappedFn)bound_RetValTc s e Zd Zefejegejf ejdd fdd
Z ejdeje
f ejeje
dddZ d d d
d Z ej
eejf d d d
Zeed fdd Z ZS )
AsyncRetryingN)r kwargsreturnc s t jf i | || _d S N)super__init__r )selfr r __class__ A/usr/lib/python3.9/site-packages/pip/_vendor/tenacity/_asyncio.pyr " s zAsyncRetrying.__init__.)fnargsr r
c s | t| |||d}| j|d}t|trtz||i |I d H }W n tyf |t Y q0 | | qt|t
r| | |I d H q|S qd S )N)Z retry_objectr r r retry_state)
beginr iter
isinstancer
BaseExceptionZ
set_exceptionsysexc_infoZ
set_resultr prepare_for_next_attemptr )r r r r r doresultr r r __call__& s
zAsyncRetrying.__call__)r
c C s | t| d di d| _| S )Nr )r r r )r r _retry_state)r r r r __aiter__> s zAsyncRetrying.__aiter__c sb | j | jd}|d u r tq t|tr2t| jdS t|trX| j | |I d H q |S q d S )Nr ) r r$ StopAsyncIterationr r r r r r )r r! r r r __anext__C s
zAsyncRetrying.__anext__)r r
c sF t t tjtjtjd fdd } j|_ j|_|S )N)r r r
c s | i |I d H S r r )r r r r r
async_wrappedT s z*AsyncRetrying.wraps..async_wrapped)r wraps functoolstypingAnyZretryZ
retry_with)r r r) r r( r r* P s zAsyncRetrying.wraps)__name__
__module__ __qualname__r r, Callablefloat Awaitabler- r r
r# r% Unionr r' r r*
__classcell__r r r r r ! s *
r )r+ r r, Zasyncior Zpip._vendor.tenacityr r r r r TypeVarr1 r r
r r r r r s