U
    b+d                     @   sn   d dl mZ d dlmZmZmZmZmZmZm	Z	 d dl
mZmZmZ erTd dlmZ e	dZG dd dZdS )	    )sleep)TYPE_CHECKINGAny	AwaitableCallableTupleTypeTypeVar)ConnectionError
RedisErrorTimeoutError)AbstractBackoffTc                   @   sp   e Zd ZdZdZeeffdeee	e
 df dddZedd	d
Zeg ee f ee
gef edddZdS )Retryz0Retry a specific number of times after a failure_backoff_retries_supported_errorsr   .)backoffretriessupported_errorsc                 C   s   || _ || _|| _dS )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.
        Nr   )selfr   r   r    r   7/tmp/pip-unpacked-wheel-cdsyf3nb/redis/asyncio/retry.py__init__   s    zRetry.__init__)specified_errorsc                 C   s   t t| jt | | _dS )zM
        Updates the supported errors with the specified error types
        N)tuplesetr   )r   r   r   r   r   update_supported_errors&   s    zRetry.update_supported_errors)dofailreturnc              
      s   | j   d}z| I dH W S  | jk
r } zT|d7 }||I dH  | jdkr^|| jkr^|| j |}|dkrt|I dH  W 5 d}~X Y qX qdS )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   resetr   r   Zcomputer   )r   r   r    Zfailureserrorr   r   r   r   call_with_retry.   s    	
zRetry.call_with_retryN)__name__
__module____qualname____doc__	__slots__r
   r   intr   r   r   r   listr   r   r   r   r   r%   r   r   r   r   r      s   	 r   N)Zasyncior   typingr   r   r   r   r   r   r	   Zredis.exceptionsr
   r   r   Zredis.backoffr   r   r   r   r   r   r   <module>   s   $