o i@sddlZddlZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlmZmZe dZe deddZerAdd lmZGd d d eeejZGd d d eeZdS)N)sleep) TYPE_CHECKINGAnyCallableGenericIterableOptionalTupleTypeTypeVarUnion)ConnectionError TimeoutErrorTET)bound covariant)AbstractBackoffc@seZdZUdZeeedfed<dddedeeedffdd Z e j d e d e fd d Zd efddZdeeed dfddZd efddZded dfddZdS) AbstractRetryz0Retry a specific number of times after a failure._supported_errorsbackoffrretriessupported_errorscCs||_||_||_dS)a' Initialize a `Retry` object with a `Backoff` object that retries a maximum of `retries` times. `retries` can be negative to retry forever. You can specify the types of supported errors which trigger a retry with the `supported_errors` parameter. N)_backoff_retriesrselfrrrr?/home/ubuntu/.local/lib/python3.10/site-packages/redis/retry.py__init__s  zAbstractRetry.__init__otherreturncCstSN)NotImplementedrr rrr__eq__0szAbstractRetry.__eq__cCst|j|jt|jfSr")hashrr frozensetrrrrr__hash__4szAbstractRetry.__hash__specified_errorsNcCstt|jt||_dS)zM Updates the supported errors with the specified error types N)tuplesetr)rr*rrrupdate_supported_errors7s z%AbstractRetry.update_supported_errorscCs|jS)z, Get the number of retries. rr(rrr get_retries?szAbstractRetry.get_retriesvaluecCs ||_dS)z, Set the number of retries. Nr.)rr0rrrupdate_retriesEs zAbstractRetry.update_retries)__name__ __module__ __qualname____doc__r r r__annotations__intrabcabstractmethodrboolr%r)rr-r/r1rrrrrs  rc seZdZejZeeejffddde de e e dfffdd Z ded efd d Z ddegefdeee gefee e geffdeee gefded ef ddZZS)Retryrrrr.cst|||dSr")superrr __class__rrrOs zRetry.__init__r r!cCs:t|tstS|j|jko|j|jkot|jt|jkSr") isinstancer;r#rrr,rr$rrrr%[s   z Retry.__eq__NFdofail is_retryablewith_failure_countc Cs|jd} z|WS|jyP}z7|r||s|d7}|r(|||n|||jdkr8||jkr8||j|}|dkrFt|WYd}~nd}~wwq)a Execute an operation that might fail and returns its result, or raise the exception that was thrown depending on the `Backoff` object. `do`: the operation to call. Expects no argument. `fail`: the failure handler, expects the last error that was thrown ``is_retryable``: optional function to determine if an error is retryable ``with_failure_count``: if True, the failure count is passed to the failure handler rTN)rresetrrcomputer)rr@rArBrCfailureserrorrrrrcall_with_retryes(    zRetry.call_with_retry)NF)r2r3r4rr)r rsockettimeoutr7r r Exceptionrrr:r%rrr rrI __classcell__rrr=rr;Ls6   r;)r8rJtimertypingrrrrrrr r r r redis.exceptionsr rrrLr redis.backoffrABCrr;rrrrs 0  2