U Œàôgžã@srddlZddlmZddlmZmZmZmZmZm Z m Z ddl m Z m Z e dƒZer`ddlmZGdd„dƒZdS) éN)Úsleep)Ú TYPE_CHECKINGÚAnyÚCallableÚIterableÚTupleÚTypeÚTypeVar)ÚConnectionErrorÚ TimeoutErrorÚT)ÚAbstractBackoffc@sveZdZdZeeejffdee e e dfdœdd„Z e e e ddœd d „Zegefee gefed œd d „ZdS)ÚRetryz0Retry a specific number of times after a failurer .)ÚbackoffÚretriesÚsupported_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Ú_retriesÚ_supported_errors)Úselfrrr©rú//tmp/pip-unpacked-wheel-f3sx1i9r/redis/retry.pyÚ__init__szRetry.__init__N)Úspecified_errorsÚreturncCstt|jt|ƒƒƒ|_dS)zM Updates the supported errors with the specified error types N)ÚtupleÚsetr)rrrrrÚupdate_supported_errors%sÿzRetry.update_supported_errors)ÚdoÚfailrc Cs†|j ¡d}z|ƒWS|jk r~}zH|d7}||ƒ|jdkrR||jkrR|‚|j |¡}|dkrnt|ƒW5d}~XYqXqdS)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 réN)rÚresetrrZcomputer)rrrZfailuresÚerrorrrrrÚcall_with_retry/s  zRetry.call_with_retry)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r ÚsocketÚtimeoutÚintrrÚ Exceptionrrrrr rr#rrrrr s"ýüü  þ  ür)r(ÚtimerÚtypingrrrrrrr Zredis.exceptionsr r r Z redis.backoffr rrrrrÚs $