o XJ@hH#@sddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlZddlmZddlmZddlmZddlmZededZededdZededZGddde eZGdd d eeejZ Gd!d"d"e eZ!d+d'd(Z"Gd)d*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_PTc@seZdZUiZded<dZed#ddZed$d dZd%d dZe d&d'ddZ ed(ddZ ee d)d*ddZ ee d+d,d!dZ e d+d,d"dZ d S)-ReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]_proxy_objects) __weakref__targetrreturncCdSNselfrr#r#N/var/www/html/venv/lib/python3.10/site-packages/sqlalchemy/ext/asyncio/base.py_assign_proxied.zReversibleProxy._assign_proxiedNonecCr!r"r#r$r#r#r&r'1r( Optional[_PT]cCs:|durt|tj}t|ttj|}|tj|<|Sr")weakrefrefr _target_gced functoolspartialr)r%rZ target_ref proxy_refr#r#r&r'4s  Nr,weakref.ref[_PT]r0Optional[weakref.ref[Self]]cCs|j|ddSr")rpop)clsr,r0r#r#r&r-AszReversibleProxy._target_gced additional_kwrrcKtr"NotImplementedError)r4rr5r#r#r&_regenerate_proxy_for_targetIsz,ReversibleProxy._regenerate_proxy_for_target. regenerate Literal[True]cKr!r"r#r4rr:r5r#r#r&_retrieve_proxy_for_targetOz*ReversibleProxy._retrieve_proxy_for_targetTboolOptional[Self]cKr!r"r#r<r#r#r&r=Ur>cKsTz |jt|}Wn tyYn w|}|dur|S|r(|j|fi|SdSr")rr+r,KeyErrorr9)r4rr:r5r0proxyr#r#r&r=[s )rrr r)rr)r r))rr*r r*r")r,r1r0r2r r))rrr5rr r).)rrr:r;r5rr r)T)rrr:r?r5rr r@) __name__ __module__ __qualname__r__annotations__ __slots__rr' classmethodr-r9r=r#r#r#r&r(s2       rc@sPeZdZdZejddddZdd d Zdd d ZejdddZ dddZ dS)StartableContextr#F is_ctxmanagerr?r rcstr"r7)r%rJr#r#r&startqszStartableContext.startGenerator[Any, Any, _T_co]cCs |Sr")rK __await__r%r#r#r&rMus zStartableContext.__await__cs|jddIdHS)NT)rJ)rKrNr#r#r& __aenter__xszStartableContext.__aenter__type_rvalue tracebackOptional[bool]csdSr"r#)r%rPrQrRr#r#r& __aexit__{szStartableContext.__aexit__r cCstd|jj)Nz@%s context has not been started and object has not been awaited.) async_excZAsyncContextNotStarted __class__rCrNr#r#r&_raise_for_not_starteds z'StartableContext._raise_for_not_startedNFrJr?r r)r rL)r r)rPrrQrrRrr rS)r r ) rCrDrErGabcabstractmethodrKrMrOrTrWr#r#r#r&rIns   rIc@s:eZdZUdZded<dd d ZddddZdddZdS)GeneratorStartableContextgenzAsyncGenerator[_T_co, Any]r^func#Callable[..., AsyncIterator[_T_co]]argsTuple[Any, ...]kwdsDict[str, Any]cCs||i||_dSr"r])r%r_rarcr#r#r&__init__sz"GeneratorStartableContext.__init__FrJr?r rcsJz t|jIdH}Wn tytddw|s#|jIdH|S)Nzgenerator didn't yield)ranext_r^StopAsyncIteration RuntimeErroraclose)r%rJZ start_valuer#r#r&rKs  zGeneratorStartableContext.starttyprrQrRrSc s|durzt|jIdHWtdtyYdSw|dur%|}z|j|IdHWtdtyI}z ||uWYd}~Sd}~wtys}z||ur[WYd}~dSt|ttfrn|j|urnWYd}~dSd}~wt y}z ||urWYd}~dSd}~ww)NFzgenerator didn't stopz$generator didn't stop after athrow()) rrfr^rgrhathrow isinstance StopIteration __cause__ BaseException)r%rjrQrRrr#r#r&rTs> !  z#GeneratorStartableContext.__aexit__N)r_r`rarbrcrdrXrY)rjrrQrrRrr rS)rCrDrErGrFrerKrTr#r#r#r&r\s   r\r_r`r /Callable[..., GeneratorStartableContext[_T_co]]cstd fdd }|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. rarrcr GeneratorStartableContext[_T_co]cs t||Sr")r\)rarcr_r#r&helpers z%asyncstartablecontext..helperN)rarrcrr rq)r.wraps)r_rsr#rrr&asyncstartablecontexts%ruc@s>eZdZdZejdddZdddZdd d ZdddZ dS)ProxyComparabler#r rcCr6r"r7rNr#r#r&_proxiedszProxyComparable._proxiedintcCst|Sr")idrNr#r#r&__hash__ szProxyComparable.__hash__otherrr?cCst||jo |j|jkSr"rlrVrwr%r{r#r#r&__eq__s  zProxyComparable.__eq__cCst||j p |j|jkSr"r|r}r#r#r&__ne__s zProxyComparable.__ne__N)r r)r rx)r{rr r?) rCrDrErGrZro_non_memoized_propertyrwrzr~rr#r#r#r&rvs   rv)r_r`r rp)$ __future__rrZr.typingrrrrrrr r r r r rrrr+rrUrZ util.typingrrrrrrABCrIr\rurvr#r#r#r&s<                     F Q,