o 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#o/var/www/html/construction_image-detection-poc/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_gced additional_kwrrcKtr"NotImplementedError)r5rr6r#r#r&_regenerate_proxy_for_targetIsz,ReversibleProxy._regenerate_proxy_for_target. regenerate Literal[True]cKr!r"r#r5rr;r6r#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,KeyErrorr:)r5rr;r6r1proxyr#r#r&r>[s )rrr r)rr)r r))rr*r r*r")r,r2r1r3r r))rrr6rr r).)rrr;r<r6rr r)T)rrr;r@r6rr rA) __name__ __module__ __qualname__r__annotations__ __slots__rr' classmethodr-r:r>r#r#r#r&r(s2       rc@sPeZdZdZejddddZdd d Zdd d ZejdddZ dddZ dS)StartableContextr#F is_ctxmanagerr@r rcstr"r8)r%rKr#r#r&startqszStartableContext.startGenerator[Any, Any, _T_co]cCs |Sr")rL __await__r%r#r#r&rNus zStartableContext.__await__cs|jddIdHS)NT)rK)rLrOr#r#r& __aenter__xszStartableContext.__aenter__type_rvalue tracebackOptional[bool]csdSr"r#)r%rQrRrSr#r#r& __aexit__{szStartableContext.__aexit__r cCstd|jj)Nz@%s context has not been started and object has not been awaited.) async_excAsyncContextNotStarted __class__rDrOr#r#r&_raise_for_not_starteds z'StartableContext._raise_for_not_startedNFrKr@r r)r rM)r r)rQrrRrrSrr rT)r r ) rDrErFrHabcabstractmethodrLrNrPrUrYr#r#r#r&rJns   rJc@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%rarcrer#r#r&__init__sz"GeneratorStartableContext.__init__FrKr@r rcsJz t|jIdH}Wn tytddw|s#|jIdH|S)Nzgenerator didn't yield)ranext_r`StopAsyncIteration RuntimeErroraclose)r%rK start_valuer#r#r&rLs  zGeneratorStartableContext.starttyprrRrSrTc 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()) rrhr`rirjathrow isinstance StopIteration __cause__ BaseException)r%rmrRrSrr#r#r&rUs> !  z#GeneratorStartableContext.__aexit__N)rarbrcrdrerfrZr[)rmrrRrrSrr rT)rDrErFrHrGrgrLrUr#r#r#r&r^s   r^rarbr /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. rcrrer GeneratorStartableContext[_T_co]cs t||Sr")r^)rcrerar#r&helpers z%asyncstartablecontext..helperN)rcrrerr rt)r.wraps)rarvr#rur&asyncstartablecontexts%rxc@s>eZdZdZejdddZdddZdd d ZdddZ dS)ProxyComparabler#r rcCr7r"r8rOr#r#r&_proxiedszProxyComparable._proxiedintcCst|Sr")idrOr#r#r&__hash__ szProxyComparable.__hash__otherrr@cCst||jo |j|jkSr"rorXrzr%r~r#r#r&__eq__s  zProxyComparable.__eq__cCst||j p |j|jkSr"rrr#r#r&__ne__s zProxyComparable.__ne__N)r r)r r{)r~rr r@) rDrErFrHrro_non_memoized_propertyrzr}rrr#r#r#r&rys   ry)rarbr rs)$ __future__rr\r.typingrrrrrrr r r r r rrrr+rrVr util.typingrrrrrrABCrJr^rxryr#r#r#r&s<                     F Q,