3
�Pf�& � @ sl d dl Z d dlmZ d dlZd dlZd dlZdZdd� ZG dd� de�Z G dd � d e�Z
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)�Retrying�call)�args�kw)�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 � ��j�f| �|�S )N)r r )r r )�dargs�dkwr r r
r / s z&retry..wrap..wrapped_f)r r )r r )r r )r r
�wrap- s zretry..wrapN)�len�callable)r r r
r r )r r r
�retry s
r c @ sn e Zd Zddd�Zdd� 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 kr�dn|
| _| d kr�tn| | _ |d kr�dn|| _
g � |d k r̈ j | j � |d k r�� j | j
� |d k r�|| _n&|d k�r
� fdd�| _n t| |�| _dd� g�|d k �r6�j | j� |d k �sJ|d k �rV�j | j� |d k �sj| d k �rv�j | j� |
d k �s�| d k �r��j | j� |d k �r�|| _n&|d k�r‡fd d�| _n t| |�| _| d k�r�| j| _n| | _|
d k�r�| j| _n|
| _|| _d S )
N� �d i� r r c s t � �fdd��D ��S )Nc 3 s | ]}|� ��V qd S )Nr )�.0r )�attempts�delayr 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 � �fdd��D ��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_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)�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_result�wrap_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| j�S )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|j�O }|S )NFr )�
has_exceptionr6 �valuer8 )r: �attemptZrejectr r r
�
should_reject� s
zRetrying.should_rejectc
O s� t ttj� d ��}d}x�yt|||�|d�}W n tj� }t||d�}Y nX | j|�sh|j| j�S t ttj� d ��| }| j ||�r�| j r�|j
r�|j� �q�t |��n<| j ||�} | j
r�tj� | j
}
| td|
� } tj| d � |d7 }qW d S )Ni� r FTr g @�@)�int�round�time�Attempt�sys�exc_inforF �getr9 r. rC �
RetryErrorr4 r* r@ r �sleep) r: �fnr r Z
start_time�attempt_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
dd�Zdd� 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 *!