B f#@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 dZedd dddddZ edddddZ eed ddddddZ eed!ddddddZ ed"ddddddZ d S)#ReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]_proxy_objects) __weakref__r)targetreturncCsdS)N)selfrr!r!]C:\Users\Lenovo\AppData\Local\Temp\pip-unpacked-wheel-4nia5yor\sqlalchemy\ext\asyncio\base.py_assign_proxied.szReversibleProxy._assign_proxiedNonecCsdS)Nr!)r"rr!r!r#r$1sz Optional[_PT]cCs:|dk r6t|tj}t|ttj|}|tj|<|S)N)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|ddS)N)rpop)clsr'r+r!r!r#r(AszReversibleProxy._target_gcedrcCs tdS)N)NotImplementedError)r-rr!r!r#_regenerate_proxy_for_targetIsz,ReversibleProxy._regenerate_proxy_for_target.z Literal[True])r regenerater cCsdS)Nr!)r-rr0r!r!r#_retrieve_proxy_for_targetMsz*ReversibleProxy._retrieve_proxy_for_targetTboolzOptional[Self]cCsdS)Nr!)r-rr0r!r!r#r1UscCsRy|jt|}Wntk r(YnX|}|dk r<|S|rJ||SdSdS)N)rr&r'KeyErrorr/)r-rr0r+proxyr!r!r#r1[s )N).)T)T) __name__ __module__ __qualname__r__annotations__ __slots__rr$ classmethodr(r/r1r!r!r!r#r(s   rc@sleZdZdZejddddddZdd d d Zdd d d ZejdddddddZ dd ddZ dS)StartableContextr!Fr2r) is_ctxmanagerr cs tdS)N)r.)r"r<r!r!r#startqszStartableContext.startzGenerator[Any, Any, _T_co])r cCs |S)N)r= __await__)r"r!r!r#r>uszStartableContext.__await__cs|jddIdHS)NT)r<)r=)r"r!r!r# __aenter__xszStartableContext.__aenter__rzOptional[bool])type_value tracebackr csdS)Nr!)r"r@rArBr!r!r# __aexit__{szStartableContext.__aexit__r cCstd|jjdS)Nz@%s context has not been started and object has not been awaited.) async_excZAsyncContextNotStarted __class__r5)r"r!r!r#_raise_for_not_startedsz'StartableContext._raise_for_not_startedN)F) r5r6r7r9abcabstractmethodr=r>r?rCrFr!r!r!r#r;nsr;c@sReZdZUdZded<dddddd Zdd d d ddZdddddddZdS)GeneratorStartableContext)genzAsyncGenerator[_T_co, Any]rJz#Callable[..., AsyncIterator[_T_co]]zTuple[Any, ...]zDict[str, Any])funcargskwdscCs||||_dS)N)rJ)r"rKrLrMr!r!r#__init__sz"GeneratorStartableContext.__init__Fr2r)r<r csNyt|jIdH}Wntk r4tddYnX|sJ|jIdH|S)Nzgenerator didn't yield)ranext_rJStopAsyncIteration RuntimeErroraclose)r"r<Z start_valuer!r!r#r=szGeneratorStartableContext.startrzOptional[bool])typrArBr c s |dkr>yt|jIdHWntk r2dSXtdn|dkrL|}y|j|IdHWntk r}z||k Sd}~XYnxtk r}z.||krdSt|ttfr|j|krdSWdd}~XYn.t k r}z||k rdSd}~XYnXtddS)NFzgenerator didn't stopz$generator didn't stop after athrow()) rrOrJrPrQathrow isinstance StopIteration __cause__ BaseException)r"rSrArBrr!r!r#rCs0   z#GeneratorStartableContext.__aexit__N)F)r5r6r7r9r8rNr=rCr!r!r!r#rIs rIz#Callable[..., AsyncIterator[_T_co]]z/Callable[..., GeneratorStartableContext[_T_co]])rKr 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:: @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])rLrMr cs t||S)N)rI)rLrM)rKr!r#helpersz%asyncstartablecontext..helper)r)wraps)rKrYr!)rKr#asyncstartablecontexts# r[c@sReZdZdZejddddZddddZd d d d d Zd d d ddZ dS)ProxyComparabler!r)r cCs tdS)N)r.)r"r!r!r#_proxiedszProxyComparable._proxiedintcCst|S)N)id)r"r!r!r#__hash__ szProxyComparable.__hash__rr2)otherr cCst||jo|j|jkS)N)rUrEr])r"rar!r!r#__eq__ s zProxyComparable.__eq__cCst||j p|j|jkS)N)rUrEr])r"rar!r!r#__ne__szProxyComparable.__ne__N) r5r6r7r9rZro_non_memoized_propertyr]r`rbrcr!r!r!r#r\s r\)$ __future__rrGr)typingrrrrrrr r r r r rrrr&rrDrZ util.typingrrrrrrABCr;rIr[r\r!r!r!r#s:                     FQ*