U
    _&                     @   sl   d dl Z d dlmZ d dlZd dlZd dlZdZdd ZG dd deZ	G dd	 d	eZ
G d
d deZdS )    N)sixi?c                     sB   t  dkr,t d r,dd }| d S  fdd}|S dS )z
    Decorator function that instantiates the Retrying object
    @param *dargs: positional arguments passed to Retrying object
    @param **dkw: keyword arguments passed to the Retrying object
       r   c                    s   t   fdd}|S )Nc                     s   t  j f| |S NRetryingcallargskwf G/home/toby/toby_env/lib/python3.8/site-packages/pip/_vendor/retrying.py	wrapped_f$   s    z-retry.<locals>.wrap_simple.<locals>.wrapped_fr   wrapsr   r   r   r   r   wrap_simple"   s    zretry.<locals>.wrap_simplec                    s   t   fdd}|S )Nc                     s   t  jf| |S r   r   r   )dargsdkwr   r   r   r   /   s    z&retry.<locals>.wrap.<locals>.wrapped_fr   r   r   r   r   r   wrap-   s    zretry.<locals>.wrapN)lencallable)r   r   r   r   r   r   r   retry   s
    r   c                   @   sn   e Zd ZdddZdd Zdd Zd	d
 Zdd Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd ZdS )r   NFc                    s  |d krdn|| _ |d krdn|| _|d kr0dn|| _|d krBdn|| _|d krTdn|| _|d krfdn|| _|	d krxdn|	| _|
d krdn|
| _|d krtn|| _	|d krdn|| _
g  |d k	r̈ | j |d k	r | j |d k	r|| _n&|d kr
 fdd| _nt| || _dd g|d k	r6| j |d k	sJ|d k	rV| j |d k	sj|	d k	rv| j |
d k	s|d k	r| j |d k	r|| _n&|d krfd	d| _nt| || _|d kr| j| _n|| _|d kr| j| _n|| _|| _d S )
N   d     r   r   c                    s   t  fddD S )Nc                 3   s   | ]}| V  qd S r   r   .0r   Zattemptsdelayr   r   	<genexpr>a   s     6Retrying.__init__.<locals>.<lambda>.<locals>.<genexpr>)anyr    )
stop_funcsr    r   <lambda>a       z#Retrying.__init__.<locals>.<lambda>c                  _   s   dS )Nr   r   )r	   kwargsr   r   r   r&   h   r'   c                    s   t  fddD S )Nc                 3   s   | ]}| V  qd S r   r   r   r    r   r   r"   y   s     r#   )maxr    )
wait_funcsr    r   r&   y   r'   )_stop_max_attempt_number_stop_max_delay_wait_fixed_wait_random_min_wait_random_max_wait_incrementing_start_wait_incrementing_increment_wait_exponential_multiplierMAX_WAIT_wait_exponential_max_wait_jitter_maxappendstop_after_attemptstop_after_delaystopgetattrfixed_sleeprandom_sleepincrementing_sleepexponential_sleepwaitalways_reject_retry_on_exceptionnever_reject_retry_on_result_wrap_exception)selfr9   r?   Zstop_max_attempt_numberstop_max_delay
wait_fixedZwait_random_minZwait_random_maxZwait_incrementing_startZwait_incrementing_incrementZwait_exponential_multiplierZwait_exponential_maxZretry_on_exceptionZretry_on_resultwrap_exceptionZ	stop_funcZ	wait_funcZwait_jitter_maxr   )r%   r*   r   __init__:   sR    








zRetrying.__init__c                 C   s
   || j kS )z;Stop after the previous attempt >= stop_max_attempt_number.)r+   rE   previous_attempt_numberdelay_since_first_attempt_msr   r   r   r7      s    zRetrying.stop_after_attemptc                 C   s
   || j kS )z=Stop after the time from the first attempt >= stop_max_delay.)r,   rJ   r   r   r   r8      s    zRetrying.stop_after_delayc                 C   s   dS )z#Don't sleep at all before retrying.r   r   rJ   r   r   r   no_sleep   s    zRetrying.no_sleepc                 C   s   | j S )z0Sleep a fixed amount of time between each retry.)r-   rJ   r   r   r   r;      s    zRetrying.fixed_sleepc                 C   s   t | j| jS )zISleep a random amount of time between wait_random_min and wait_random_max)randomrandintr.   r/   rJ   r   r   r   r<      s    zRetrying.random_sleepc                 C   s$   | j | j|d   }|dk r d}|S )z
        Sleep an incremental amount of time after each attempt, starting at
        wait_incrementing_start and incrementing by wait_incrementing_increment
        r   r   )r0   r1   )rE   rK   rL   resultr   r   r   r=      s    zRetrying.incrementing_sleepc                 C   s2   d| }| j | }|| jkr"| j}|dk r.d}|S )N   r   )r2   r4   )rE   rK   rL   exprP   r   r   r   r>      s    

zRetrying.exponential_sleepc                 C   s   dS )NFr   rE   rP   r   r   r   rB      s    zRetrying.never_rejectc                 C   s   dS )NTr   rS   r   r   r   r@      s    zRetrying.always_rejectc                 C   s4   d}|j r || |jd O }n|| |jO }|S )NFr   )has_exceptionrA   valuerC   )rE   attemptZrejectr   r   r   should_reject   s
    zRetrying.should_rejectc                 O   s   t tt d }d}zt||||d}W n    t }t||d}Y nX | |sf|| jS t tt d | }| 	||r| js|j
r| qt|n<| ||}	| jrt | j }
|	td|
 }	t|	d  |d7 }qd S )Nr   r   FTr   g     @@)introundtimeAttemptsysexc_inforW   getrD   r9   rT   
RetryErrorr?   r5   rN   r)   sleep)rE   fnr	   r(   Z
start_timeattempt_numberrV   tbrL   r`   Zjitterr   r   r   r      s(    


zRetrying.call)NNNNNNNNNNNNNFNNN)__name__
__module____qualname__rI   r7   r8   rM   r;   r<   r=   r>   rB   r@   rW   r   r   r   r   r   r   8   s:                        
S
		r   c                   @   s*   e Zd ZdZdd Zd
ddZdd Zd	S )r[   z
    An Attempt encapsulates a call to a target function that may end as a
    normal return value from the function or an Exception depending on what
    occurred during the execution.
    c                 C   s   || _ || _|| _d S r   )rU   rb   rT   )rE   rU   rb   rT   r   r   r   rI      s    zAttempt.__init__Fc                 C   s@   | j r6|rt| q<t| jd | jd | jd  n| jS dS )z
        Return the return value of this Attempt instance or raise an Exception.
        If wrap_exception is true, this Attempt is wrapped inside of a
        RetryError before being raised.
        r   r   rQ   N)rT   r_   r   reraiserU   )rE   rH   r   r   r   r^      s
    
"zAttempt.getc              	   C   s:   | j r&d| jdt| jd S d| j| jS d S )NzAttempts: {0}, Error:
{1} rQ   zAttempts: {0}, Value: {1})rT   formatrb   join	traceback	format_tbrU   rE   r   r   r   __repr__   s     zAttempt.__repr__N)F)rd   re   rf   __doc__rI   r^   rn   r   r   r   r   r[      s   
r[   c                   @   s    e Zd ZdZdd Zdd ZdS )r_   zU
    A RetryError encapsulates the last Attempt instance right before giving up.
    c                 C   s
   || _ d S r   )last_attempt)rE   rp   r   r   r   rI     s    zRetryError.__init__c                 C   s   d | jS )NzRetryError[{0}])ri   rp   rm   r   r   r   __str__
  s    zRetryError.__str__N)rd   re   rf   ro   rI   rq   r   r   r   r   r_     s   r_   )rN   pip._vendorr   r\   rZ   rk   r3   r   objectr   r[   	Exceptionr_   r   r   r   r   <module>   s    *!