U ``&@shddlZddlZddlZddlZddlZdZddZGdddeZGdddeZ Gd d d e Z dS) Ni?csBtdkr,tdr,dd}|dSfdd}|SdS)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 rcstfdd}|S)Ncstjf||SNRetryingcallargskwf`/Volumes/workspace/Demo/src/IotBifrostDemoAssets/src/iot_sitewise_asset_model_lambda/retrying.py wrapped_f$sz-retry..wrap_simple..wrapped_fsixwrapsr rr r r wrap_simple"szretry..wrap_simplecstfdd}|S)Ncstjf||Srrr)dargsdkwr r r r/sz&retry..wrap..wrapped_frrrrr r wrap-szretry..wrapN)lencallable)rrrrr rr retrys  rc@sneZdZdddZddZddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZdS)rNFcs |dkr dn||_|dkrdn||_|dkr0dn||_|dkrBdn||_|dkrTdn||_|dkrfdn||_| dkrxdn| |_| dkrdn| |_| dkrtn| |_ |dkrdn||_ g|dk r̈ |j |dk r |j |dk r||_n&|dkr fdd|_n t|||_ddg|dk r6 |j|dk sJ|dk rV |j|dk sj| dk rv |j| dk s| dk r |j|dk r||_n&|dkr‡fd d|_n t|||_| dkr|j|_n| |_| dkr|j|_n| |_||_dS) NdrrcstfddDS)Nc3s|]}|VqdSrr .0r attemptsdelayr r as6Retrying.__init__....)anyr ) stop_funcsr r az#Retrying.__init__..c_sdS)Nrr )rkwargsr r r r'hr(cstfddDS)Nc3s|]}|VqdSrr rr r r r#ysr$)maxr ) wait_funcsr r r'yr()_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:r@stop_max_attempt_numberZstop_max_delay wait_fixedZwait_random_minZwait_random_maxZwait_incrementing_startZwait_incrementing_incrementZwait_exponential_multiplierZwait_exponential_maxretry_on_exceptionZretry_on_resultwrap_exceptionZ stop_funcZ wait_funcZwait_jitter_maxr )r&r+r __init__:sR                 zRetrying.__init__cCs ||jkS)z;Stop after the previous attempt >= stop_max_attempt_number.)r,rFprevious_attempt_numberdelay_since_first_attempt_msr r r r8szRetrying.stop_after_attemptcCs ||jkS)z=Stop after the time from the first attempt >= stop_max_delay.)r-rLr r r r9szRetrying.stop_after_delaycCsdS)z#Don't sleep at all before retrying.rr rLr r r no_sleepszRetrying.no_sleepcCs|jS)z0Sleep a fixed amount of time between each retry.)r.rLr r r r<szRetrying.fixed_sleepcCst|j|jS)zISleep a random amount of time between wait_random_min and wait_random_max)randomrandintr/r0rLr r r r=szRetrying.random_sleepcCs$|j|j|d}|dkr d}|S)z Sleep an incremental amount of time after each attempt, starting at wait_incrementing_start and incrementing by wait_incrementing_increment rr)r1r2)rFrMrNresultr r r r>szRetrying.incrementing_sleepcCs2d|}|j|}||jkr"|j}|dkr.d}|S)Nr)r3r5)rFrMrNexprRr r r r?s  zRetrying.exponential_sleepcCsdS)NFr rFrRr r r rCszRetrying.never_rejectcCsdS)NTr rUr r r rAszRetrying.always_rejectcCs4d}|jr |||jdO}n|||jO}|S)NFr) has_exceptionrBvaluerD)rFattemptZrejectr r r should_rejects zRetrying.should_rejectc Ostttd}d}zt||||d}Wn t}t||d}YnX||sf||jStttd|}| ||r|js|j r|qt |n<| ||} |j rt|j } | td| } t| d|d7}qdS)NrrFTrg@@)introundtimeAttemptsysexc_inforYgetrEr:rV RetryErrorr@r6rPr*sleep) rFfnrr)Z start_timeattempt_numberrXtbrNrbZjitterr r r rs(       z Retrying.call)NNNNNNNNNNNNNFNNN)__name__ __module__ __qualname__rKr8r9rOr<r=r>r?rCrArYrr r r r r8s: S   rc@s*eZdZdZddZd ddZddZd 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. cCs||_||_||_dSr)rWrdrV)rFrWrdrVr r r rKszAttempt.__init__FcCs@|jr6|rt|qs*!