o j@s UddlmZdZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZddlmZdd lmZe r?dd lmZe d Ze d Zd$ddZd$ddZd$ddZeedddGdddZd%ddZeZded<GdddejZ Gd d!d!e eZ!Gd"d#d#e eZ"dS)&) annotations)EventLoopToken RunvarTokenRunVar checkpointcheckpoint_if_cancelledcancel_shielded_checkpoint current_tokenN) dataclass) TracebackType) TYPE_CHECKINGAnyGenericLiteralTypeVarfinaloverload)WeakKeyDictionary)get_async_backend) AsyncBackendTDreturnNonectIdHdS)z Check for cancellation and allow the scheduler to switch to another task. Equivalent to (but more efficient than):: await checkpoint_if_cancelled() await cancel_shielded_checkpoint() .. versionadded:: 3.0 N)rrrrL/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/anyio/lowlevel.pyr rcr)z Enter a checkpoint if the enclosing cancel scope has been cancelled. This does not allow the scheduler to switch to a different task. .. versionadded:: 3.0 N)rrrrrrr+s rcr)z Allow the scheduler to switch to another task but without checking for cancellation. Equivalent to (but potentially more efficient than):: with CancelScope(shield=True): await checkpoint() .. versionadded:: 3.0 N)rrrrrrr7rrTF)frozenreprc@s"eZdZUdZded<ded<dS)rza An opaque object that holds a reference to an event loop. .. versionadded:: 4.11.0 ztype[AsyncBackend] backend_classobject native_tokenN)__name__ __module__ __qualname____doc____annotations__rrrrrFs  rcCst}|}t||S)z Return a token object that can be used to call code in the current event loop from another thread. :raises NoEventLoopError: if no supported asynchronous event loop is running in the current thread .. versionadded:: 4.11.0 )rr r)r! raw_tokenrrrr Ss  r z1WeakKeyDictionary[object, dict[RunVar[Any], Any]] _run_varsc@seZdZeZdS) _NoValueSetN)r$r%r&enumauto NO_VALUE_SETrrrrr+fs r+c@s2eZdZdZdZdddZdd d ZdddZdS)rz A token that can be used to restore a :class:`RunVar` to its previous value. Returned by :meth:`RunVar.set`. Can be used as a context manager to automatically reset the variable on exit, or passed directly to :meth:`RunVar.reset`. _var_value _redeemedvar RunVar[T]value%T | Literal[_NoValueSet.NO_VALUE_SET]cCs||_||_d|_dS)NFr/)selfr3r5rrr__init__ts zRunvarToken.__init__rRunvarToken[T]cCs|SNrr7rrr __enter__yzRunvarToken.__enter__exc_typetype[BaseException] | Noneexc_valBaseException | Noneexc_tbTracebackType | NonercCs|j|dSr:)r0reset)r7r>r@rBrrr__exit__|szRunvarToken.__exit__N)r3r4r5r6)rr9)r>r?r@rArBrCrr)r$r%r&r' __slots__r8r<rErrrrrjs   rc@seZdZUdZdZejZded<efd"d d Ze d#d dZ e d$ddZ e d%ddZ efd&ddZ d'ddZ d(ddZd)dd Zd!S)*ra Like a :class:`~contextvars.ContextVar`, except scoped to the running event loop. Can be used as a context manager, Just like :class:`~contextvars.ContextVar`, that will reset the variable to its previous value when the context block is exited. _name_defaultz!Literal[_NoValueSet.NO_VALUE_SET]r.namestrdefaultr6cCs||_||_dSr:rG)r7rJrLrrrr8s zRunVar.__init__rdict[RunVar[T], T]cCs6tj}zt|WStyi}t|<|YSwr:)r r#r*KeyError)r7r#run_varsrrr _current_varss   zRunVar._current_varsrT | DcCdSr:rr7rLrrrgetr=z RunVar.getrcCrRr:rr;rrrrTr=%D | Literal[_NoValueSet.NO_VALUE_SET]cCsZz|j|WSty#|tjur|YS|jtjur!|jYSYnwtd|jd)a2 Return the current value of this run variable. :param default: a fallback value to return if no value has been set :return: the current value, the provided default, or the variable's own default :raises LookupError: if no value is set and no default is available zRun variable "z!" has no value and no default set)rPrNrr.rI LookupErrorrHrSrrrrTs      r5r9cCs&|j}t|||tj}|||<|S)z Set the value of this run variable for the current event loop. :param value: the new value :return: a token that can be used to restore the previous value )rPrrTrr.)r7r5 current_varstokenrrrsetsz RunVar.setrXrcCsb|j|ur td|jrtd|jtjur&z|j|=Wnty%Ynw|j|j|<d|_dS)a Restore this run variable to the value it held before the matching :meth:`set`. :param token: the token returned by :meth:`set` :raises ValueError: if the token belongs to a different :class:`RunVar` or the token has already been used z)This token does not belong to this RunVarz This token has already been usedTN)r0 ValueErrorr2r1r+r.rPrN)r7rXrrrrDs      z RunVar.resetcCsd|jdS)Nz )rHr;rrr__repr__szRunVar.__repr__N)rJrKrLr6)rrM)rLrrrQ)rr)rLrUrrQ)r5rrr9)rXr9rr)rrK)r$r%r&r'rFr+r.r(r8propertyrPrrTrYrDr\rrrrrs"       r)rr)rr)# __future__r__all__r, dataclassesr typesr typingr r rrrrrweakrefr_core._eventlooprabcrrrrrrrr r*r(Enumr+rrrrrrs.  $