U [hI#@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%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)rZ covariant_PTc@seZdZUiZded<dZedddddZedddd dZd d dd dZed!d dddddZ eddddddZ eed"dddddddZ eed#dddddddZ ed$dddddd dZ d S)%ReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]_proxy_objects) __weakref__r)targetreturncCsdSNselfrr!r!A./venv/lib/python3.8/site-packages/sqlalchemy/ext/asyncio/base.py_assign_proxied.szReversibleProxy._assign_proxiedNonecCsdSr r!r"r!r!r$r%1sz Optional[_PT]cCs:|dk r6t|tj}t|ttj|}|tj|<|Sr )weakrefrefr _target_gced functoolspartialr)r#rZ target_ref proxy_refr!r!r$r%4s  Nzweakref.ref[_PT]zOptional[weakref.ref[Self]])r(r,rcCs|j|ddSr )rpop)clsr(r,r!r!r$r)AszReversibleProxy._target_gcedrr)r additional_kwrcKs tdSr NotImplementedError)r.rr/r!r!r$_regenerate_proxy_for_targetIsz,ReversibleProxy._regenerate_proxy_for_target.z Literal[True])r regenerater/rcKsdSr r!r.rr3r/r!r!r$_retrieve_proxy_for_targetOsz*ReversibleProxy._retrieve_proxy_for_targetTboolzOptional[Self]cKsdSr r!r4r!r!r$r5UscKsVz|jt|}Wntk r(YnX|}|dk r<|S|rN|j|f|SdSdSr )rr'r(KeyErrorr2)r.rr3r/r,proxyr!r!r$r5[s)N).)T)T) __name__ __module__ __qualname__r__annotations__ __slots__rr% classmethodr)r2r5r!r!r!r$r(s0   rc@sleZdZdZejddddddZdd d d Zdd d d ZejdddddddZ dd ddZ dS)StartableContextr!Fr6r is_ctxmanagerrcs tdSr r0)r#rAr!r!r$startqszStartableContext.startzGenerator[Any, Any, _T_co]rcCs |Sr )rB __await__r#r!r!r$rDuszStartableContext.__await__cs|jddIdHS)NT)rA)rBrEr!r!r$ __aenter__xszStartableContext.__aenter__rOptional[bool])type_value tracebackrcsdSr r!)r#rHrIrJr!r!r$ __aexit__{szStartableContext.__aexit__r cCstd|jjdS)Nz@%s context has not been started and object has not been awaited.) async_excZAsyncContextNotStarted __class__r9rEr!r!r$_raise_for_not_starteds z'StartableContext._raise_for_not_startedN)F) r9r:r;r=abcabstractmethodrBrDrFrKrNr!r!r!r$r?nsr?c@sReZdZUdZded<dddddd Zdd d d ddZdddddddZdS)GeneratorStartableContextgenzAsyncGenerator[_T_co, Any]rS#Callable[..., AsyncIterator[_T_co]]zTuple[Any, ...]zDict[str, Any])funcargskwdscCs||||_dSr rR)r#rUrVrWr!r!r$__init__sz"GeneratorStartableContext.__init__Fr6rr@csNzt|jIdH}Wntk r4tddYnX|sJ|jIdH|S)Nzgenerator didn't yield)ranext_rSStopAsyncIteration RuntimeErroraclose)r#rAZ start_valuer!r!r$rBszGeneratorStartableContext.startrrG)typrIrJrc s*|dkr@zt|jIdHWntk r4YdSXtdn|dkrN|}z|j|IdHWntk r}z||k WYSd}~XYntk r}z:||krWY,dSt|ttfr|j|krWY dSW5d}~XYn6t k r}z||k rWYdSd}~XYnXtddS)NFzgenerator didn't stopz$generator didn't stop after athrow()) rrYrSrZr[athrow isinstance StopIteration __cause__ BaseException)r#r]rIrJrr!r!r$rKs4    z#GeneratorStartableContext.__aexit__N)F)r9r:r;r=r<rXrBrKr!r!r!r$rQs rQrTz/Callable[..., GeneratorStartableContext[_T_co]])rUrcs$tddddfdd }|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. rz GeneratorStartableContext[_T_co])rVrWrcs t||Sr )rQ)rVrWrUr!r$helpersz%asyncstartablecontext..helper)r*wraps)rUrdr!rcr$asyncstartablecontexts%rfc@sReZdZdZejddddZddddZd d d d d Zd d d ddZ dS)ProxyComparabler!rrCcCs tdSr r0rEr!r!r$_proxiedszProxyComparable._proxiedintcCst|Sr )idrEr!r!r$__hash__ szProxyComparable.__hash__rr6)otherrcCst||jo|j|jkSr r_rMrhr#rlr!r!r$__eq__s  zProxyComparable.__eq__cCst||j p|j|jkSr rmrnr!r!r$__ne__s zProxyComparable.__ne__N) r9r:r;r=rZro_non_memoized_propertyrhrkrorpr!r!r!r$rgs rg)$Z __future__rrOr*typingrrrrrrr r r r r rrrr'rrLrZ util.typingrrrrrrABCr?rQrfrgr!r!r!r$s:                     FQ,