o wi @sddlZddlZddlZddlZddlZddlmZddlZddlm Z e e Z e dddddde df ddZe dddddde df d d Zd d Zd dZddZddZde dfddZe dddddde df ddZdde dddddde df ddZdS)N)partial decoratorFc Cs||} } | rHz|WS|yC} z,| d8} |r"t|||| | | | s%| dur-| | t| t|||| } WYd} ~ nd} ~ ww| sdSdSNr) _log_attempttimesleep _new_delay f exceptionstriesdelay max_delaybackoffjittershow_tracebacklogger fail_callback_tries_delayerH/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/reretry/api.py__retry_internals"   rc s||} } | rRz|IdHWS|yM} z2| d8} |r&t|||| | | | s)| dur4| | IdHt| IdHt|||| } WYd} ~ nd} ~ ww| sdSdSr)rasyncior r r rrr__retry_internal_async2s$  rcCsf|r!|rdtd||j}|||d|||||dS|dkr1|d||||dSdS)Nz4%s, attempt %s/%s failed - retrying in %s seconds...rz%%s, attempt %s/%s failed - giving up!)join tracebackformat_exception __traceback__warning)rrrrrrtb_strrrrrUs   rcCs:||9}|t|trtj|n|7}|durt||}|SN) isinstancetuplerandomuniformmin)rrrrrrrr is  r cCst|o t| Sr&)riscoroutinefunctioninspectisgeneratorfunctionr rrr _is_asyncssr0cCst|rtStSr&)r0rrr/rrr_get_internal_functionwsr1cCsL|r |dus Jd|r t|rt|s"t|st|r$JddSdSdS)Nz`show_traceback` needs `logger`zWIf the retried function is async, fail_callback needs to be async as well or vice versa)r0)r rrrrrr _check_params{sr2c s$tf dd} | S)aReturns a retry decorator. :param exceptions: an exception or a tuple of exceptions to catch. default: Exception. :param tries: the maximum number of attempts. default: -1 (infinite). :param delay: initial delay between attempts (in seconds). default: 0. :param max_delay: the maximum value of delay (in seconds). default: None (no limit). :param backoff: multiplier applied to delay between attempts. default: 1 (no backoff). :param jitter: extra seconds added to delay between attempts. default: 0. fixed if a number, random if a range tuple (min, max) :param show_traceback: if True, the traceback of the exception will be logged. :param logger: logger.warning(fmt, error, delay) will be called on failed attempts. default: retry.logging_logger. if None, logging is disabled. :param fail_callback: fail_callback(e) will be called on failed attempts. :returns: a retry decorator. c st||| Sr&) retry_call)r fargsfkwargs rrrrrrrrrrrretry_decoratorszretry..retry_decoratorr) rrrrrrrrrr7rr6rretrysr8c  CsV|pt} |p t} t|| | | t|}|t|g| Ri| ||||||| | | S)a Calls a function and re-executes it if it failed. :param f: the function to execute. :param fargs: the positional arguments of the function to execute. :param fkwargs: the named arguments of the function to execute. :param exceptions: an exception or a tuple of exceptions to catch. default: Exception. :param tries: the maximum number of attempts. default: -1 (infinite). :param delay: initial delay between attempts (in seconds). default: 0. :param max_delay: the maximum value of delay (in seconds). default: None (no limit). :param backoff: multiplier applied to delay between attempts. default: 1 (no backoff). :param jitter: extra seconds added to delay between attempts. default: 0. fixed if a number, random if a range tuple (min, max) :param show_traceback: if True, the traceback of the exception will be logged. :param logger: logger.warning(fmt, error, delay) will be called on failed attempts. default: retry.logging_logger. if None, logging is disabled. :param fail_callback: fail_callback(e) will be called on failed attempts. :returns: the result of the f function. )listdictr2r1r)r r4r5rrrrrrrrrargskwargsfuncrrrr3s ! r3)rloggingr)r r! functoolsrr-compatr getLogger__name__logging_logger Exceptionrrrr r0r1r2r8r3rrrrsp    % #   1