KiadZddlmZddlZddlmZmZmZddlmZm Z ddl m Z m Z m Z mZmZerdd lmZdd lmZej(d ZGd d eZGdde dZGddeZy)zRetry strategies for tasks.) annotationsN)datetime timedeltatimezone) TYPE_CHECKINGNoReturn)FailureHandler TaskOutcomecurrent_executioncurrent_workerformat_duration) Execution) TASKS_RETRIEDzdocket.dependenciesceZdZdZy) ForcedRetryz7Raised when a task requests a retry via `after` or `at`N)__name__ __module__ __qualname____doc__f/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/docket/dependencies/_retry.pyrrsArrc|eZdZUdZded<ded<ded<ded f dd Zdd Zdd Zdd Z ddZ ddZ y)RetryaConfigures linear retries for a task. You can specify the total number of attempts (or `None` to retry indefinitely), and the delay between attempts. Example: ```python @task async def my_task(retry: Retry = Retry(attempts=3)) -> None: ... ``` int | Noneattemptsrdelayintattemptr rc.||_||_d|_y)z Args: attempts: The total number of attempts to make. If `None`, the task will be retried indefinitely. delay: The delay between attempts. r N)rrr!)selfrrs r__init__zRetry.__init__.s!   rcKtj}t|j|j}|j |_|Sw)Nrr)r getrrrr!)r# executionretrys r __aenter__zRetry.__aenter__;s:%))+ t}}DJJ?!))  sA A c$||_t)z&Request a retry after the given delay.)rrr#rs rafterz Retry.afterAs mrctjtj}||z }|j dk\r|n t d}|j |y)z"Request a retry at the given time.rN)rnowrutc total_secondsrr-)r#whenr/diffs ratzRetry.atFsDll8<<(cz))+q0til 4rc&|j|y)z Deprecated: use after() instead.N)r-r,s rin_z Retry.in_Ms 5rcK|j|j|jk\rytjtj |j z|_|xjdz c_|jdd{tj}tjdi|j|j|jrRt j#dt%|j&j)|j+|jt j-dt%|j&j)|j+y7w) z8Handle failure by scheduling a retry if attempts remain.NFr T)replaceu ↩ [%s] %s)exc_infou ↫ [%s] %s)rr!rr/rr0rr2scheduler r'raddlabelsgeneral_labels exceptionloggererrorrdurationr1 call_reprinfo)r#r(outcomeworkers rhandle_failurezRetry.handle_failureQs( == $):):dmm)K!hll3djj@ Q   ...##%!N N93K3K3MNO    LL 0 0 > > @A##% **     G,,::< =    ! ' /sBE>E< C4E>N)rrrrreturnNone)rGr)rrrGr)r2rrGr)r(rrDr rGbool) rrrr__annotations__rr$r*r-r4r6rFrrrrrs[   L&'9Q< " /8     rrc^eZdZdZdededf dfd ZddZxZS) ExponentialRetryaOConfigures exponential retries for a task. You can specify the total number of attempts (or `None` to retry indefinitely), and the minimum and maximum delays between attempts. Example: ```python @task async def my_task(retry: ExponentialRetry = ExponentialRetry(attempts=3)) -> None: ... ``` r )seconds@c6t|||||_y)a Args: attempts: The total number of attempts to make. If `None`, the task will be retried indefinitely. minimum_delay: The minimum delay between attempts. maximum_delay: The maximum delay between attempts. r&N)superr$ maximum_delay)r#r minimum_delayrQ __class__s rr$zExponentialRetry.__init__|s (-@*rcbKtj}t|j|j|j }|j |_|j dkDrJd|j dz z}|j|z}||j kDr|j |_|S||_|Sw)N)rrRrQr r)r r'rLrrrQr!)r#r(r)backoff_factorcalculated_delays rr*zExponentialRetry.__aenter__s%))+  ]]**,,  "))   q 9#4#4q#89N#zzN: $"4"44"00  /  sB-B/)rrrRrrQrrGrH)rGrL)rrrrrr$r* __classcell__)rSs@rrLrLnsM  !#,Q#7#,R#8 ++!+! +  + rrL)r __future__rloggingrrrtypingrr_baser r r r rr(rinstrumentationr getLoggerr? ExceptionrrrLrrrr_sm!"22*%+   0 1B)BNN7 #Nb1u1r