3
Pf& @ sl d dl Z d dlmZ d dlZd dlZd dlZdZdd ZG dd deZ G dd d eZ
G d
d d e Z dS ) N)sixi?c sB t dkr,t d r,dd }| d S fdd}|S dS )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 j fdd}|S )Nc s t j f| |S )N)Retryingcall)argskw)f /usr/lib/python3.6/retrying.py wrapped_f$ s z-retry..wrap_simple..wrapped_f)r wraps)r r r )r r
wrap_simple" s zretry..wrap_simplec s t j fdd}|S )Nc s t jf| |S )N)r r )r r )dargsdkwr r r
r / s z&retry..wrap..wrapped_f)r r )r r )r r )r r
wrap- s zretry..wrapN)lencallable)r r r
r r )r r r
retry s
r c @ sn e Zd ZdddZdd Zdd Zd d
Zd d Zd
d Zdd Z dd Z
dd Z dd Z dd Z
dd ZdS ) r NFc s |d kr dn|| _ |d kr dn|| _|d kr0dn|| _|d krBdn|| _|d krTdn|| _|d krfdn|| _| d krxdn| | _|
d krdn|
| _| d krtn| | _ |d krdn|| _
g |d k r̈ j | j |d k r j | j
|d k r|| _n&|d kr
fdd| _n t| || _dd g|d k r6j | j |d k sJ|d k rVj | j |d k sj| d k rvj | j |
d k s| d k rj | j |d k r|| _n&|d krfd d| _n t| || _| d kr| j| _n| | _|
d kr| j| _n|
| _|| _d S )
N d i r r c s t fddD S )Nc 3 s | ]}| V qd S )Nr ).0r )attemptsdelayr r
a s z6Retrying.__init__....)any)r r )
stop_funcs)r r r
a s z#Retrying.__init__..c _ s dS )Nr r )r kwargsr r r
r h s c s t fddD S )Nc 3 s | ]}| V qd S )Nr )r r )r r r r
r y s z6Retrying.__init__....)max)r r )
wait_funcs)r r r
r y s )_stop_max_attempt_number_stop_max_delay _wait_fixed_wait_random_min_wait_random_max_wait_incrementing_start _wait_incrementing_increment _wait_exponential_multiplierMAX_WAIT_wait_exponential_max_wait_jitter_maxappendstop_after_attemptstop_after_delaystopgetattr fixed_sleep random_sleepincrementing_sleepexponential_sleepwait
always_reject_retry_on_exception never_reject_retry_on_result_wrap_exception)selfr. r4 Zstop_max_attempt_numberZstop_max_delayZ
wait_fixedZwait_random_minZwait_random_maxZwait_incrementing_startZwait_incrementing_incrementZwait_exponential_multiplierZwait_exponential_maxZretry_on_exceptionZretry_on_resultwrap_exceptionZ stop_funcZ wait_funcZwait_jitter_maxr )r r r
__init__: sR
zRetrying.__init__c C s
|| j kS )z;Stop after the previous attempt >= stop_max_attempt_number.)r )r: previous_attempt_number delay_since_first_attempt_msr r r
r, s zRetrying.stop_after_attemptc C s
|| j kS )z=Stop after the time from the first attempt >= stop_max_delay.)r! )r: r= r> r r r
r- s zRetrying.stop_after_delayc C s dS )z#Don't sleep at all before retrying.r r )r: r= r> r r r
no_sleep s zRetrying.no_sleepc C s | j S )z0Sleep a fixed amount of time between each retry.)r" )r: r= r> r r r
r0 s zRetrying.fixed_sleepc C s t j| j| jS )zISleep a random amount of time between wait_random_min and wait_random_max)randomZrandintr# r$ )r: r= r> r r r
r1 s zRetrying.random_sleepc C s$ | j | j|d }|dk 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 )r% r& )r: r= r> resultr r r
r2 s zRetrying.incrementing_sleepc C s2 d| }| j | }|| jkr"| j}|dk r.d}|S )N r )r' r) )r: r= r> ZexprA r r r
r3 s
zRetrying.exponential_sleepc C s dS )NFr )r: rA r r r
r7 s zRetrying.never_rejectc C s dS )NTr )r: rA r r r
r5 s zRetrying.always_rejectc C s4 d}|j r || j|jd O }n|| j|jO }|S )NFr )
has_exceptionr6 valuer8 )r: attemptZrejectr r r
should_reject s
zRetrying.should_rejectc
O s t ttj d }d}xyt||||d}W n tj }t||d}Y nX | j|sh|j| jS t ttj d | }| j ||r| j r|j
r|j qt |n<| j ||} | j
rtj | j
}
| td|
} tj| d |d7 }qW d S )Ni r FTr g @@)introundtimeAttemptsysexc_inforF getr9 r. rC
RetryErrorr4 r* r@ r sleep) r: fnr r Z
start_timeattempt_numberrE tbr> rO Zjitterr r r
r s*
z
Retrying.call)NNNNNNNNNNNNNFNNN)__name__
__module__ __qualname__r< r, r- r? r0 r1 r2 r3 r7 r5 rF r r r r r
r 8 s0
F
r c @ s* e Zd ZdZdd Zd
ddZdd Zd S ) rJ 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 )N)rD rQ rC )r: rD rQ rC r r r
r< s zAttempt.__init__Fc C s@ | j r6|rt| q s *!