o g"@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*ddZ e d)d*d dZ d S)+ReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]_proxy_objects) __weakref__targetrreturncCdSNselfrr#r#W/var/www/html/flaskapp/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%r target_ref proxy_refr#r#r&r'4s  Nr,weakref.ref[_PT]r1Optional[weakref.ref[Self]]cCs|j|ddSr")rpop)clsr,r1r#r#r&r-AszReversibleProxy._target_gcedrcCtr"NotImplementedError)r5rr#r#r&_regenerate_proxy_for_targetIz,ReversibleProxy._regenerate_proxy_for_target. regenerate Literal[True]cCr!r"r#r5rr;r#r#r&_retrieve_proxy_for_targetMsz*ReversibleProxy._retrieve_proxy_for_targetTboolOptional[Self]cCr!r"r#r=r#r#r&r>UscCsLz |jt|}Wn tyYn w|}|dur|S|r$||SdSr")rr+r,KeyErrorr9)r5rr;r1proxyr#r#r&r>[s  )rrr r)rr)r r))rr*r r*r")r,r2r1r3r r))rrr r).)rrr;r<r r)T)rrr;r?r 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_excAsyncContextNotStarted __class__rCrNr#r#r&_raise_for_not_starteds z'StartableContext._raise_for_not_startedNFrJr?r r)r rL)r r)rPrrQrrRrr rS)r r ) rCrDrErGabcabstractmethodrKrMrOrTrXr#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`rbrdr#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%rJ 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()) rrgr_rhriathrow isinstance StopIteration __cause__ BaseException)r%rlrQrRrr#r#r&rTs> !  z#GeneratorStartableContext.__aexit__N)r`rarbrcrdrerYrZ)rlrrQrrRrr rS)rCrDrErGrFrfrKrTr#r#r#r&r]s   r]r`rar /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. rbrrdr GeneratorStartableContext[_T_co]cs t||Sr")r])rbrdr`r#r&helpers z%asyncstartablecontext..helperN)rbrrdrr rs)r.wraps)r`rur#rtr&asyncstartablecontexts%rwc@s>eZdZdZejdddZdddZdd d ZdddZ dS)ProxyComparabler#r rcCr6r"r7rNr#r#r&_proxiedr:zProxyComparable._proxiedintcCst|Sr")idrNr#r#r&__hash__ szProxyComparable.__hash__otherrr?cCst||jo |j|jkSr"rnrWryr%r}r#r#r&__eq__s  zProxyComparable.__eq__cCst||j p |j|jkSr"r~rr#r#r&__ne__s zProxyComparable.__ne__N)r r)r rz)r}rr r?) rCrDrErGrro_non_memoized_propertyryr|rrr#r#r#r&rxs   rx)r`rar rr)$ __future__rr[r.typingrrrrrrr r r r r rrrr+rrUr util.typingrrrrrrABCrIr]rwrxr#r#r#r&s<                     F Q,