@g"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<dZeddZedd Zdd 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 O/var/www/piapp/venv/lib/python3.11/site-packages/sqlalchemy/ext/asyncio/base.py_assign_proxiedzReversibleProxy._assign_proxied. NonecdSr$r%r&s r(r)zReversibleProxy._assign_proxied2r*r+ Optional[_PT]c|`tj|tj}tj|t jtj|}|tj|<|Sr$)weakrefrefr _target_gced functoolspartialr)r'r! target_ref proxy_refs r(r)zReversibleProxy._assign_proxied6s_  +2;4,,J !/"> KKI:CO *: 6 r+Nr1weakref.ref[_PT]r6Optional[weakref.ref[Self]]c<|j|ddSr$)rpop)clsr1r6s r(r2zReversibleProxy._target_gcedCs# sD)))))r+rctr$NotImplementedError)r;r!s r(_regenerate_proxy_for_targetz,ReversibleProxy._regenerate_proxy_for_targetK!###r+. regenerate Literal[True]cdSr$r%r;r!rAs r(_retrieve_proxy_for_targetz*ReversibleProxy._retrieve_proxy_for_targetOs  r+TboolOptional[Self]cdSr$r%rDs r(rEz*ReversibleProxy._retrieve_proxy_for_targetXs r+c |jtj|}|}||Sn#t$rYnwxYw|r||SdSr$)rr0r1KeyErrorr?)r;r!rAr6proxys r(rEz*ReversibleProxy._retrieve_proxy_for_target_s *7;v+>+>?IIKKE  !     D   33F;; ;4s 0 ==)r!rr"r)r!r,r"r,)r!r.r"r.r$)r1r7r6r8r"r,)r!rr"r).)r!rrArBr"r)T)r!rrArFr"rG) __name__ __module__ __qualname__r__annotations__ __slots__rr) classmethodr2r?rEr%r+r(rr(se     !I    X    X     26****[*$$$[$%(    [X -1    [X -1[r+rcpeZdZdZejdddZdd Zdd ZejddZ ddZ dS)StartableContextr%F is_ctxmanagerrFr"rc"Ktr$r=)r'rTs r(startzStartableContext.startus!###r+Generator[Any, Any, _T_co]cN|Sr$)rV __await__r's r(rYzStartableContext.__await__yszz||%%'''r+c>K|dd{VS)NT)rT)rVrZs r( __aenter__zStartableContext.__aenter__|s,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__rLrZs r(_raise_for_not_startedz'StartableContext._raise_for_not_starteds'. N~& (   r+NFrTrFr"r)r"rW)r"r)r]rr^rr_rr"r`)r"r ) rLrMrNrPabcabstractmethodrVrYr\rbrgr%r+r(rSrSrsI$$$$$((((4444           r+rSc8eZdZUdZded<dd ZdddZddZdS)GeneratorStartableContextgenzAsyncGenerator[_T_co, Any]rofunc#Callable[..., AsyncIterator[_T_co]]argsTuple[Any, ...]kwdsDict[str, Any]c ||i||_dSr$rn)r'rprrrts r(__init__z"GeneratorStartableContext.__init__s 4&&&r+FrTrFr"rcK tj|jd{V}n#t$rt ddwxYw|s|jd{V|S)Nzgenerator didn't yield)ranext_roStopAsyncIteration RuntimeErroraclose)r'rT start_values r(rVzGeneratorStartableContext.starts C $ DH 5 5555555KK! C C C788d B C $(//## # # # # # # #s$?typrr^r_r`cK|@ tj|jd{Vtd#t$rYdSwxYw| |} tj|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()) rryror{rzathrow isinstance StopIteration __cause__ BaseException)r'r~r^r_rs r(rbz#GeneratorStartableContext.__aexit__s ; <k$(+++++++++##:;;;&   uu  }! k$(C BBBBBBBBBB% ( ( (%'''''''   %<< 55555u}6H&IJJ! .. 55555   e##uuuuu EFF FsQ4 AA"A55 C(?BC( C(C %C  C  C(C##C(N)rprqrrrsrtrurhri)r~rr^rr_rr"r`)rLrMrNrPrOrwrVrbr%r+r(rmrmsnI####''''     2G2G2G2G2G2Gr+rmrprqr"/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:: @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. rrrrtr" GeneratorStartableContext[_T_co]c&t||Sr$)rm)rrrtrps r(helperz%asyncstartablecontext..helpers(tT:::r+)rrrrtrr"r)r3wraps)rprs` r(asyncstartablecontextrs@F_T;;;;;; Mr+cLeZdZdZejd dZddZdd Zdd Z d S)ProxyComparabler%r"rctr$r=rZs r(_proxiedzProxyComparable._proxied r@r+intc t|Sr$)idrZs r(__hash__zProxyComparable.__hash__s $xxr+otherrrFcLt||jo|j|jkSr$rrfrr'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"rF) rLrMrNrPrro_non_memoized_propertyrrrrr%r+r(rrswI "$$$#"$          r+r)rprqr"r)$ __future__rrjr3typingrrrrrr r r r r rrrrr0rrdr util.typingrrrrrrABCrSrmrrr%r+r(rs#""""" !!!!!! """""" WTsd333ge3GGGGGgclGGGT     y'   4NGNGNGNGNG 0 7NGNGNGb''''T     oc*     r+