o ;'hI@sUddlmZddlZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZe dZe d Zd#d d Zd#ddZd#ddZd$ddZe Zded<iZded<eddGdddZGdddejZGdd d eeZGd!d"d"eeZdS)%) annotationsN) dataclass)AnyGenericLiteralTypeVaroverload)WeakKeyDictionary)get_async_backendTDreturnNonectIdHdS)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)r checkpointrrDC:\Users\shubh\Videos\agent\nova\Lib\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)r checkpoint_if_cancelledrrrrrs 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)r cancel_shielded_checkpointrrrrr*rrobjectcCs tS)ze Return a backend specific token object that can be used to get back to the event loop. )r current_tokenrrrrr:s rz&WeakKeyDictionary[Any, dict[str, Any]] _run_varszdict[Any, _TokenWrapper]_token_wrappersT)frozenc@seZdZUdZded<dS) _TokenWrapper)_token __weakref__rrN)__name__ __module__ __qualname__ __slots____annotations__rrrrrGs  rc@seZdZeZdS) _NoValueSetN)rr r!enumauto NO_VALUE_SETrrrrr$Ms r$c@seZdZdZd ddZdS) RunvarToken_var_value _redeemedvar RunVar[T]value%T | Literal[_NoValueSet.NO_VALUE_SET]cCs||_||_d|_dS)NFr))selfr-r/rrr__init__Ts zRunvarToken.__init__N)r-r.r/r0)rr r!r"r2rrrrr(Qsr(c@seZdZUdZdZejZded<eZ ded<efd$d d Z e d%ddZ e d&ddZe d'ddZefd(ddZd)ddZd*dd Zd+d!d"Zd#S),RunVarz[ Like a :class:`~contextvars.ContextVar`, except scoped to the running event loop. _name_defaultz!Literal[_NoValueSet.NO_VALUE_SET]r'zset[_TokenWrapper]rnamestrdefaultr0cCs||_||_dSNr4)r1r7r9rrrr2es zRunVar.__init__r dict[str, T]cCs4t}zt|WStyi}t|<|YSwr:)rrKeyError)r1tokenZrun_varsrrr _current_varsks   zRunVar._current_varsr T | DcCdSr:rr1r9rrrgettz RunVar.getr cCr@r:rr1rrrrBwrC%D | Literal[_NoValueSet.NO_VALUE_SET]cCs\z|j|jWSty$|tjur|YS|jtjur"|jYSYnwtd|jd)NzRun variable "z!" has no value and no default set)r>r5r<r3r'r6 LookupErrorrArrrrBzs     r/RunvarToken[T]cCs*|j}t|||jtj}|||j<|Sr:)r>r(rBr5r3r')r1r/Z current_varsr=rrrsets z RunVar.setr=rcCsf|j|ur td|jrtd|jtjur'z|j|j=Wnty&Ynw|j|j|j<d|_dS)Nz)This token does not belong to this RunVarz This token has already been usedT) r* ValueErrorr,r+r$r'r>r5r<)r1r=rrrresets    z RunVar.resetcCsd|jdS)Nz )r5rDrrr__repr__szRunVar.__repr__N)r7r8r9r0)rr;)r9r rr?)rr )r9rErr?)r/r rrG)r=rGrr)rr8)rr r!__doc__r"r$r'r#rHrr2propertyr>rrBrJrLrrrrr3Zs$       r3)rr)rr) __future__rr% dataclassesrtypingrrrrrweakrefr Z_core._eventloopr r r rrrrrr#rrEnumr$r(r3rrrrs&