U (h)@sddlmZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z ddl m Z dd lmZdd lmZd ZGd d d ejZGdddZeGdddeZdddddZdddddZdS)) annotationsN) TracebackType)finalOptionalType)events) exceptions)tasks)install)Self)Timeouttimeout timeout_atc@s eZdZdZdZdZdZdZdS)_StatecreatedactiveZexpiringexpiredfinishedN)__name__ __module__ __qualname__CREATEDENTEREDEXPIRINGEXPIREDEXITEDrr6/tmp/pip-unpacked-wheel-rrqp3fat/taskgroup/timeouts.pyrs rc@seZdZdZdddddZdddd Zdddd d Zd dd dZddddZddddZ dddddddZ ddddZ dS)_TimeoutzAsynchronous context manager for cancelling overdue coroutines. Use `timeout()` or `timeout_at()` rather than instantiating this class directly. Optional[float]NonewhenreturncCstj|_d|_d|_||_dS)zSchedule a timeout that will trigger at a given loop time. - If `when` is `None`, the timeout will never trigger. - If `when < loop.time()`, the timeout will trigger on the next iteration of the event loop. N)rr_state_timeout_handler_task_when)selfr#rrr__init__)sz_Timeout.__init__r$cCs|jS)zReturn the current deadline.)r(r)rrrr#6sz _Timeout.whencCs|jtjk r4|jtjkr tdtd|jjd||_|jdk rN|j|dkr^d|_n4t }|| kr| |j |_n|||j |_dS)zReschedule the timeout.zTimeout has not been enteredzCannot change state of z TimeoutN)r%rrr RuntimeErrorvaluer(r&cancelrget_running_looptimeZ call_soon _on_timeoutZcall_at)r)r#looprrr reschedule:s     z_Timeout.rescheduleboolcCs|jtjtjfkS)z$Is timeout expired during execution?)r%rrrr,rrrrQsz_Timeout.expiredstrcCs\dg}|jtjkr<|jdk r(t|jdnd}|d|d|}d|jjd|dS)Nzwhen= z )r%rrr(roundappendjoinr.)r)infor#Zinfo_strrrr__repr__Us   z_Timeout.__repr__r csV|jtjk rtdt}|dkr,tdtj|_||_|j|_ | |j |S)Nz Timeout has already been enteredz$Timeout should be used inside a task) r%rrr-r Z current_taskrr'Z cancelling _cancellingr4r()r)Ztaskrrr __aenter__]s   z_Timeout.__aenter__Optional[Type[BaseException]]Optional[BaseException]Optional[TracebackType]Optional[bool]exc_typeexc_valexc_tbr$cs~|jtjtjfkst|jdk r0|jd|_|jtjkrftj|_|j |j krz|t j krzt |n|jtjkrztj|_dSN)r%rrrAssertionErrorr&r/rr'ZuncancelrArZCancelledError TimeoutErrorrr)rHrIrJrrr __aexit__is    z_Timeout.__aexit__cCs:|jdk st|jtjkst|jtj|_d|_dSrK)r'rLr%rrr/rr&r,rrrr2s  z_Timeout._on_timeoutN) rrr__doc__r*r#r4rr@rBrOr2rrrrr#s  rcs@eZdZUded<ddfdd Zddd d d d d ZZS)r zcontextlib.AsyncExitStack_Timeout__stackr r+c sbt4IdHB}|tIdHtIdH|tj| |_ W5QIdHRX|SrK) contextlibAsyncExitStackenter_async_context_installZinstall_uncancelsuperrBpush_async_exitrOpop_allrQ)r)stack __class__rrrBs zTimeout.__aenter__rCrDrErFrGcs|j|||IdHSrK)rQrOrNrrrrOszTimeout.__aexit__)rrr__annotations__rBrO __classcell__rrrZrr s r r )delayr$cCs$t}t|dk r||ndS)a Timeout async context manager. Useful in cases when you want to apply timeout logic around block of code or in cases when asyncio.wait_for is not suitable. For example: >>> async with asyncio.timeout(10): # 10 seconds timeout ... await long_running_task() delay - value in seconds or None to disable timeout logic long_running_task() is interrupted by raising asyncio.CancelledError, the top-most affected timeout() context manager converts CancelledError into TimeoutError. N)rr0r r1)r^r3rrrrsrr"cCst|S)abSchedule the timeout at absolute time. Like timeout() but argument gives absolute time in the same clock system as loop.time(). Please note: it is not POSIX time but a time with undefined starting base, e.g. the time of the system power on. >>> async with asyncio.timeout_at(loop.time() + 10): ... await long_running_task() when - a deadline when timeout occurs or None to disable timeout logic long_running_task() is interrupted by raising asyncio.CancelledError, the top-most affected timeout() context manager converts CancelledError into TimeoutError. )r )r#rrrrsr) __future__rrRenumtypesrtypingrrrZasynciorrr r7r rUZtyping_extensionsr __all__Enumrrr rrrrrrs       j