KiW d Z ddlmZ ddlZddlZddlmZ ddlmZm Z m
Z
m Z erddl m
Z
ddlmZ d d
lmZmZ G d d ed Zy)
z Timeout dependency for tasks. ) annotationsN) timedelta)
TYPE_CHECKINGAny AwaitableCallable ) Execution) cancel_task )Runtimecurrent_docketc t e Zd ZU dZded< ded< ddZddZddZdd Zdd
Z ddd Z
dd
Z y )Timeoutac Configures a timeout for a task. You can specify the base timeout, and the
task will be cancelled if it exceeds this duration. The timeout may be extended
within the context of a single running task.
Example:
```python
@task
async def my_task(timeout: Timeout = Timeout(timedelta(seconds=10))) -> None:
...
```
r basefloat _deadlinec || _ y)zD
Args:
base: The base timeout duration.
Nr )selfr s h/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/docket/dependencies/_timeout.py__init__zTimeout.__init__" s
c 6 K t | j S w)Nr )r r r s r
__aenter__zTimeout.__aenter__) s DII&&s c l t j | j j z | _ y N)time monotonicr
total_secondsr r s r startz
Timeout.start, s" )DII,C,C,EEr c D t j | j k\ S r )r r r r s r expiredzTimeout.expired/ s ~~ 4>>11r c X t | j t j z
S )z1Get the remaining time until the timeout expires.)seconds)r r r r r s r remainingzTimeout.remaining2 s $..2B!BCCr Nc f | | j }| xj |j z
c_ y)zExtend the timeout by a given duration. If no duration is provided, the
base timeout will be used.
Args:
by: The duration to extend the timeout by.
N)r r r! )r bys r extendzTimeout.extend6 s) :B "**,,r c ` K | j t j }t j ||i ||j
d|j }d} |j s| j rd}n t j t j | | j j d{ |j sWd|j d| j j d }t || d{ |rt j | S S |j sWd|j d| j j d }t || d{ |rt j | yy7 7 # t j $ r Y [w xY w7 ;# |j sXd|j d| j j d }t || d{ 7 |rt j | w w xY ww)
z.Execute the function with timeout enforcement.z - task:)nameFT)timeoutNz Docket task z exceeded timeout of s)r" r getasyncio create_taskr, keydoner$ wait_forshieldr' r! r r TimeoutError) r executionfunctionargskwargsdockettask timed_outtimeout_reasons r runz Timeout.runA s
##%"" d %f %KK= 8
?iik<<> $I !(!1!1 t,dnn6F6T6T6V" 99;"9==/ 2""&))"9"9";!>
499;"9==/ 2""&))"9"9";!>
8 ++ 8 99;"9==/ 2""&))"9"9";!>
s A H. #G A F%
F!F% A
H.F# A'H.F? H.!F% #H.%F<8G ;F<<G ?H.A H+ H
H++H.)r r returnNone)r@ r )r@ rA )r@ bool)r@ r r )r) ztimedelta | Noner@ rA )
r7 r
r8 z Callable[..., Awaitable[Any]]r9 ztuple[Any, ...]r: zdict[str, Any]r@ r ) __name__
__module__ __qualname____doc____annotations__r r r" r$ r' r* r? r r r r si O'F2D -%? %? 0%? %?
%?
%?r r )rF
__future__r r0 r datetimer typingr r r r r7 r
_cancellationr _baser
r r rH r r rN s9 # " : :% ' *U?gi U?r