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)r 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"?/tmp/pip-unpacked-wheel-y9ln43lq/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-r cCs|j|ddSr!)rpop)clsr)r-r"r"r%r*AszReversibleProxy._target_gcedrr)r additional_kwr cKs tdSr!NotImplementedError)r/rr0r"r"r%_regenerate_proxy_for_targetIsz,ReversibleProxy._regenerate_proxy_for_target.z Literal[True])r regenerater0r cKsdSr!r"r/rr4r0r"r"r%_retrieve_proxy_for_targetOsz*ReversibleProxy._retrieve_proxy_for_targetTboolzOptional[Self]cKsdSr!r"r5r"r"r%r6UscKsVz|jt|}Wntk r(YnX|}|dk r<|S|rN|j|f|SdSdSr!)rr(r)KeyErrorr3)r/rr4r0r-proxyr"r"r%r6[s)N).)T)T) __name__ __module__ __qualname__r__annotations__ __slots__rr& classmethodr*r3r6r"r"r"r%r(s0   rc@sleZdZdZejddddddZdd d d Zdd d d ZejdddddddZ dd ddZ dS)StartableContextr"Fr7r is_ctxmanagerr cs tdSr!r1)r$rBr"r"r%startqszStartableContext.startzGenerator[Any, Any, _T_co]r cCs |Sr!)rC __await__r$r"r"r%rEuszStartableContext.__await__cs|jddIdHS)NT)rB)rCrFr"r"r% __aenter__xszStartableContext.__aenter__rOptional[bool])type_value tracebackr csdSr!r")r$rIrJrKr"r"r% __aexit__{szStartableContext.__aexit__r cCstd|jjdS)Nz@%s context has not been started and object has not been awaited.) async_excZAsyncContextNotStarted __class__r:rFr"r"r%_raise_for_not_starteds z'StartableContext._raise_for_not_startedN)F) r:r;r<r>abcabstractmethodrCrErGrLrOr"r"r"r%r@nsr@c@sReZdZUdZded<dddddd Zdd d d ddZdddddddZdS)GeneratorStartableContextgenzAsyncGenerator[_T_co, Any]rT#Callable[..., AsyncIterator[_T_co]]zTuple[Any, ...]zDict[str, Any])funcargskwdscCs||||_dSr!rS)r$rVrWrXr"r"r%__init__sz"GeneratorStartableContext.__init__Fr7rrAcsNzt|jIdH}Wntk r4tddYnX|sJ|jIdH|S)Nzgenerator didn't yield)ranext_rTStopAsyncIteration RuntimeErroraclose)r$rBZ start_valuer"r"r%rCszGeneratorStartableContext.startrrH)typrJrKr c 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()) rrZrTr[r\athrow isinstance StopIteration __cause__ BaseException)r$r^rJrKrr"r"r%rLs4    z#GeneratorStartableContext.__aexit__N)F)r:r;r<r>r=rYrCrLr"r"r"r%rRs rRrUz/Callable[..., GeneratorStartableContext[_T_co]])rVr cs$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])rWrXr cs t||Sr!)rR)rWrXrVr"r%helpersz%asyncstartablecontext..helper)r+wraps)rVrer"rdr%asyncstartablecontexts%rgc@sReZdZdZejddddZddddZd d d d d Zd d d ddZ dS)ProxyComparabler"rrDcCs tdSr!r1rFr"r"r%_proxiedszProxyComparable._proxiedintcCst|Sr!)idrFr"r"r%__hash__ szProxyComparable.__hash__rr7)otherr cCst||jo|j|jkSr!r`rNrir$rmr"r"r%__eq__s  zProxyComparable.__eq__cCst||j p|j|jkSr!rnror"r"r%__ne__s zProxyComparable.__ne__N) r:r;r<r>rZro_non_memoized_propertyrirlrprqr"r"r"r%rhs rh)$ __future__rrPr+typingrrrrrrr r r r r rrrr(rrMrZ util.typingrrrrrrABCr@rRrgrhr"r"r"r%s:                     FQ,