0P@i"ddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlZddlmZddlmZddlmZddlmZedeZededZedeZGdde eZGdd eeej>Z Gd!d"e eZ! d&d#Z"Gd$d%eeZ#y)') 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<dZed dZeddZddZe d ddZ edd Z ee d dd Z ee d dd Z e d dd Z y)ReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]_proxy_objects) __weakref__cyNselftargets e/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/sqlalchemy/ext/asyncio/base.py_assign_proxiedzReversibleProxy._assign_proxied. cyr"r#r$s r'r(zReversibleProxy._assign_proxied2r)r*c|otj|tj}tj|t j tj|}|tj |<|Sr")weakrefrefr _target_gced functoolspartialr)r%r& target_ref proxy_refs r'r(zReversibleProxy._assign_proxied6sb  +2;;44,J !!/">"> KI:CO * *: 6 r*Nc<|jj|dyr")rpop)clsr.r3s r'r/zReversibleProxy._target_gcedCs sD)r*ctr"NotImplementedError)r6r&s r'_regenerate_proxy_for_targetz,ReversibleProxy._regenerate_proxy_for_targetK !##r*cyr"r#r6r& regenerates r'_retrieve_proxy_for_targetz*ReversibleProxy._retrieve_proxy_for_targetOs r*cyr"r#r=s r'r?z*ReversibleProxy._retrieve_proxy_for_targetXs r*c |jtj|}|}||S|r|j |Sy#t$rYwxYwr")rr-r.KeyErrorr:)r6r&r>r3proxys r'r?z*ReversibleProxy._retrieve_proxy_for_target_s_ **7;;v+>?IKE  33F; ;   s"A AA)r&rreturnr)r&NonerDrE)r& Optional[_PT]rDrFr")r.zweakref.ref[_PT]r3zOptional[weakref.ref[Self]]rDrE)r&rrDr).)r&rr>z Literal[True]rDr)T)r&rr>boolrDzOptional[Self]) __name__ __module__ __qualname__r__annotations__ __slots__rr( classmethodr/r:r?r#r*r'rr(s4  !I      26* */*  **$$%(  "     -1  &*   -1&* r*rceZdZdZej dd dZd dZd dZej d dZ d dZ y)StartableContextr#cKtwr"r8)r% is_ctxmanagers r'startzStartableContext.startus!##s c>|jjSr")rR __await__r%s r'rTzStartableContext.__await__yszz|%%''r*cBK|jdd{S7w)NT)rQ)rRrUs r' __aenter__zStartableContext.__aenter__|sZZdZ3333s c Kywr"r#)r%type_value tracebacks r' __aexit__zStartableContext.__aexit__s  scZtjd|jjz)Nz@%s context has not been started and object has not been awaited.) async_excAsyncContextNotStarted __class__rHrUs r'_raise_for_not_startedz'StartableContext._raise_for_not_starteds*.. N~~&& (  r*NFrQrGrDr)rDzGenerator[Any, Any, _T_co])rDr)rYrrZrr[rrDOptional[bool])rDr ) rHrIrJrLabcabstractmethodrRrTrWr\rar#r*r'rOrOrshI$$(4   !$ 14    r*rOcReZdZUdZded< ddZd d dZ d dZy) GeneratorStartableContextgenzAsyncGenerator[_T_co, Any]rjc||i||_yr"ri)r%funcargskwdss r'__init__z"GeneratorStartableContext.__init__s &&r*cK tj|jd{}|s"|jj d{|S7*#t$r t ddwxYw7!w)Nzgenerator didn't yield)ranext_rjStopAsyncIteration RuntimeErroraclose)r%rQ start_values r'rRzGeneratorStartableContext.startsi C $ DHH 55K((//# # #6! C78d B C $s7A."AAAA. A, A.AA))A.cK|3 tj|jd{td||} |jj |d{td7G#t$rYywxYw7 #t$r}||ucYd}~Sd}~wt$r8}||urYd}~yt |ttfr|j|urYd}~yd}~wt$r}||urYd}~yd}~wwxYww)Nzgenerator didn't stopFz$generator didn't stop after athrow()) rrqrjrsrrathrow isinstance StopIteration __cause__ BaseException)r%typrZr[rs r'r\z#GeneratorStartableContext.__aexit__s ; <kk$((+++##:;;}! hhooe,,,BEF FY,%  -% (%'' %< u}6H&IJ .  e# sC2"A2A0A2C2B B!B% C20A22 A>;C2=A>>C2B C/ BC/C2 C/!C%C2*$CC2C C/ C*%C2*C//C2N)rl#Callable[..., AsyncIterator[_T_co]]rmzTuple[Any, ...]rnzDict[str, Any]rbrc)r|rrZrr[rrDrd)rHrIrJrLrKrorRr\r#r*r'rhrhs]I ##'1'' ' 2G2G"2G/22G 2Gr*rhcDtjdfd }|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. ct||Sr")rh)rmrnrls r'helperz%asyncstartablecontext..helpers(tT::r*)rmrrnrrDz GeneratorStartableContext[_T_co])r0wraps)rlrs` r'asyncstartablecontextrs(F__T;; Mr*cNeZdZdZej ddZddZd dZd dZ y) ProxyComparabler#ctr"r8rUs r'_proxiedzProxyComparable._proxied r;r*ct|Sr")idrUs r'__hash__zProxyComparable.__hash__s $xr*cdt||jxr|j|jk(Sr"rxr`rr%others r'__eq__zProxyComparable.__eq__s* udnn - 0 / r*cft||j xs|j|jk7Sr"rrs r'__ne__zProxyComparable.__ne__s-5$..1 1 /}}. r*N)rDr)rDint)rrrDrG) rHrIrJrLrro_non_memoized_propertyrrrrr#r*r'rrs.I ""$#$  r*r)rlr}rDz/Callable[..., GeneratorStartableContext[_T_co]])$ __future__rrer0typingrrrrrr r r r r rrrrr-rr^r util.typingrrrrrrABCrOrhrrr#r*r'rs# ! " Tsd3e3GgclGT y' 4NG 0 7NGb' -'4'T oc* r*