hH#ddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlZddlmZddlmZddlmZddlmZedeZededZedeZGdde eZGdd eeejZ Gd!d"e eZ!d*d'Z"Gd(d)eeZ#dS)+) annotationsN)Any)AsyncGenerator) AsyncIterator) Awaitable)Callable)ClassVar)Dict) Generator)Generic)NoReturn)Optional)overload)Tuple)TypeVar)exc)util)Literal)Self_T)bound_T_coT)r covariant_PTceZdZUiZded<dZeddZed d Zd!d Ze d"d#dZ ed$dZ ee d%d&dZ ee d'd(dZ e d'd(dZ d S))ReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]_proxy_objects) __weakref__targetrreturncdSNselfr!s ]/var/www/html/prod/cognitive/venv/lib/python3.11/site-packages/sqlalchemy/ext/asyncio/base.py_assign_proxiedzReversibleProxy._assign_proxied.s363NonecdSr$r%r&s r(r)zReversibleProxy._assign_proxied1s58Sr* Optional[_PT]c|`tj|tj}tj|t jtj|}|tj|<|Sr$)weakrefrefr _target_gced functoolspartialr)r'r! target_ref proxy_refs r(r)zReversibleProxy._assign_proxied4s_  +2;4,,J !/"> KKI:CO *: 6 r*Nr0weakref.ref[_PT]r5Optional[weakref.ref[Self]]c<|j|ddSr$)rpop)clsr0r5s r(r1zReversibleProxy._target_gcedAs# sD)))))r* additional_kwrrc tr$NotImplementedError)r:r!r;s r(_regenerate_proxy_for_targetz,ReversibleProxy._regenerate_proxy_for_targetIs"###r*. regenerate Literal[True]c dSr$r%r:r!r@r;s r(_retrieve_proxy_for_targetz*ReversibleProxy._retrieve_proxy_for_targetOs sr*TboolOptional[Self]c dSr$r%rCs r(rDz*ReversibleProxy._retrieve_proxy_for_targetUs r*c  |jtj|}|}||Sn#t$rYnwxYw|r|j|fi|SdSr$)rr/r0KeyErrorr?)r:r!r@r;r5proxys r(rDz*ReversibleProxy._retrieve_proxy_for_target[s *7;v+>+>?IIKKE  !     D   333FLLmLL L4s 0 ==)r!rr"r)r!r+r"r+)r!r-r"r-r$)r0r6r5r7r"r+)r!rr;rr"r).)r!rr@rAr;rr"r)T)r!rr@rEr;rr"rF) __name__ __module__ __qualname__r__annotations__ __slots__rr) classmethodr1r?rDr%r*r(rr(sQ     !I 666X6 888X8    26****[*$$$[$ 69[X-1[X-1[r*rcpeZdZdZejdddZdd Zdd ZejddZ ddZ dS)StartableContextr%F is_ctxmanagerrEr"rc"Ktr$r=)r'rSs r(startzStartableContext.startqs!###r*Generator[Any, Any, _T_co]cN|Sr$)rU __await__r's r(rXzStartableContext.__await__uszz||%%'''r*c>K|dd{VS)NT)rS)rUrYs r( __aenter__zStartableContext.__aenter__xs,ZZdZ333333333r*type_rvalue tracebackOptional[bool]c KdSr$r%)r'r\r]r^s r( __aexit__zStartableContext.__aexit__{s  r*r cDtjd|jjz)Nz@%s context has not been started and object has not been awaited.) async_excAsyncContextNotStarted __class__rKrYs r(_raise_for_not_startedz'StartableContext._raise_for_not_starteds'. N~& (   r*NFrSrEr"r)r"rV)r"r)r\rr]rr^rr"r_)r"r ) rKrLrMrOabcabstractmethodrUrXr[rarfr%r*r(rRrRnsI$$$$$((((4444           r*rRc8eZdZUdZded<dd ZdddZddZdS)GeneratorStartableContextgenzAsyncGenerator[_T_co, Any]rnfunc#Callable[..., AsyncIterator[_T_co]]argsTuple[Any, ...]kwdsDict[str, Any]c ||i||_dSr$rm)r'rorqrss r(__init__z"GeneratorStartableContext.__init__s 4&&&r*FrSrEr"rcK tj|jd{V}n#t$rt ddwxYw|s|jd{V|S)Nzgenerator didn't yield)ranext_rnStopAsyncIteration RuntimeErroraclose)r'rS start_values r(rUzGeneratorStartableContext.starts C $ DH 5 5555555KK! C C C788d B C $(//## # # # # # # #s$?typrr]r^r_cK|@ tj|jd{Vtd#t$rYdSwxYw| |} |j|d{Vnv#t$r}||ucYd}~Sd}~wt$r;}||urYd}~dSt |ttfr|j|urYd}~dSd}~wt$r}||urYd}~dSd}~wwxYwtd)Nzgenerator didn't stopFz$generator didn't stop after athrow()) rrxrnrzryathrow isinstance StopIteration __cause__ BaseException)r'r}r]r^rs r(raz#GeneratorStartableContext.__aexit__s ; <k$(+++++++++##:;;;&   uu  }! hooe,,,,,,,,,,% ( ( (%'''''''   %<< 55555u}6H&IJJ! .. 55555   e##uuuuu EFF FsQ4 AA A33 C&=BC& C&C %C C  C&C!!C&N)rorprqrrrsrtrgrh)r}rr]rr^rr"r_)rKrLrMrOrNrvrUrar%r*r(rlrlsnI####''''     2G2G2G2G2G2Gr*rlrorpr"/Callable[..., GeneratorStartableContext[_T_co]]cHtjdfd }|S)a=@asyncstartablecontext decorator. the decorated function can be called either as ``async with fn()``, **or** ``await fn()``. This is decidedly different from what ``@contextlib.asynccontextmanager`` supports, and the usage pattern is different as well. Typical usage: .. sourcecode:: text @asyncstartablecontext async def some_async_generator(): try: yield except GeneratorExit: # return value was awaited, no context manager is present # and caller will .close() the resource explicitly pass else: Above, ``GeneratorExit`` is caught if the function were used as an ``await``. In this case, it's essential that the cleanup does **not** occur, so there should not be a ``finally`` block. If ``GeneratorExit`` is not invoked, this means we're in ``__aexit__`` and we were invoked as a context manager, and cleanup should proceed. rqrrsr" GeneratorStartableContext[_T_co]c&t||Sr$)rl)rqrsros r(helperz%asyncstartablecontext..helpers(tT:::r*)rqrrsrr"r)r2wraps)rors` r(asyncstartablecontextrs@J_T;;;;;; Mr*cLeZdZdZejd dZddZdd Zdd Z d S)ProxyComparabler%r"rctr$r=rYs r(_proxiedzProxyComparable._proxieds!###r*intc t|Sr$)idrYs r(__hash__zProxyComparable.__hash__ s $xxr*otherrrEcLt||jo|j|jkSr$rrerr'rs r(__eq__zProxyComparable.__eq__s' udn - - 0 / r*cNt||j p|j|jkSr$rrs r(__ne__zProxyComparable.__ne__s*5$.11 1 /}. r*N)r"r)r"r)rrr"rE) rKrLrMrOrro_non_memoized_propertyrrrrr%r*r(rrswI "$$$#"$          r*r)rorpr"r)$ __future__rrir2typingrrrrrr r r r r rrrrr/rrcr util.typingrrrrrrABCrRrlrrr%r*r(rs#""""" !!!!!! """""" WTsd333ge3CCCCCgclCCCL     y'   4NGNGNGNGNG 0 7NGNGNGb))))X     oc*     r*