U
|
¸ý°dN$ ã @ s d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm
|
Z
|
d d lm Z d d
|
lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlZddlmZ ddlmZ ddlmZ ddlmZ ededZededdZ ed edZ G d d d e e Z G d d d ee ej Z G d!d" d"e e Z!d#d$d%d&d'Z"G d(d) d)e e Z#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 @ sÒ e Zd ZU i Zded< dZedddddZedddd dZd
|
d
|
dd dZed d dddddZ edddddZ
|
eed ddddddZ eed!ddd dd dZ ed"ddd dd dZ d S )#ÚReversibleProxyzCClassVar[Dict[weakref.ref[Any], weakref.ref[ReversibleProxy[Any]]]]Ú_proxy_objects)Ú __weakref__r )ÚtargetÚreturnc C s d S ©N© ©Úselfr r" r" úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/ext/asyncio/base.pyÚ_assign_proxied. s z ReversibleProxy._assign_proxiedÚNonec C s d S r! r" r# r" r" r% r& 2 s z Optional[_PT]c C s: |d k r6t |tj¡}t | t tj|¡¡}|tj|< |S r! )ÚweakrefÚrefr Ú _target_gcedÚ functoolsÚpartialr )r$ r Z
|
target_refÚ proxy_refr" r" r% r& 6 s ÿ ÿþ
|
Nzweakref.ref[_PT]zOptional[weakref.ref[Self]])r) r- r c C s | j |d ¡ d S r! )r Úpop)Úclsr) r- r" r" r% r* E s z ReversibleProxy._target_gcedr c C s
|
t d S r! ©ÚNotImplementedError)r/ r r" r" r% Ú _regenerate_proxy_for_targetM s z,ReversibleProxy._regenerate_proxy_for_target.z Literal[True])r Ú
|
regenerater c C s d S r! r" ©r/ r r3 r" r" r% Ú_retrieve_proxy_for_targetQ s z*ReversibleProxy._retrieve_proxy_for_targetTÚboolzOptional[Self]c C s d S r! r" r4 r" r" r% r5 Z s c C sR z| j t |¡ }W n tk
|
r( Y nX | }|d k r<|S |rJ| |¡S d S d S r! )r r( r) ÚKeyErrorr2 )r/ r r3 r- Úproxyr" r" r% r5 a s
|
)N).)T)T) Ú__name__Ú
|
__module__Ú __qualname__r Ú__annotations__Ú __slots__r r& Ú classmethodr* r2 r5 r" r" r" r% r ( s0
|
þ
|
ý ý ÿ ÿr c @ sl e Zd ZdZejddddddZdd d
|
d Zdd d d ZejdddddddZ dd ddZ
|
dS )ÚStartableContextr" Fr6 r ©Ú is_ctxmanagerr c à s
|
t d S r! r0 )r$ rA r" r" r% Ústartw s zStartableContext.startzGenerator[Any, Any, _T_co]©r c C s | ¡ ¡ S r! )rB Ú __await__©r$ r" r" r% rD { s zStartableContext.__await__c à s | j ddI d H S )NT)rA )rB rE r" r" r% Ú
|
__aenter__~ s zStartableContext.__aenter__r úOptional[bool])Útype_ÚvalueÚ tracebackr c à s d S r! r" )r$ rH rI rJ r" r" r% Ú __aexit__ s zStartableContext.__aexit__r c C s t d| jj ¡d S )Nz@%s context has not been started and object has not been awaited.)Ú async_excZAsyncContextNotStartedÚ __class__r9 rE r" r" r% Ú_raise_for_not_started s
|
ÿÿz'StartableContext._raise_for_not_startedN)F) r9 r: r; r= ÚabcÚabstractmethodrB rD rF rK rN r" r" r" r% r? t s r? c @ sR e Zd ZU dZded< dddddd Zdd d d ddZdddddddZdS )ÚGeneratorStartableContext©ÚgenzAsyncGenerator[_T_co, Any]rS ú#Callable[..., AsyncIterator[_T_co]]zTuple[Any, ...]zDict[str, Any])ÚfuncÚargsÚkwdsc C s |||| _ d S r! rR )r$ rU rV rW r" r" r% Ú__init__ s z"GeneratorStartableContext.__init__Fr6 r r@ c à sN zt | j¡I d H }W n tk
|
r4 tdd Y nX |sJ| j ¡ I d H |S )Nzgenerator didn't yield)r Úanext_rS ÚStopAsyncIterationÚ RuntimeErrorÚaclose)r$ rA Z start_valuer" r" r% rB s z GeneratorStartableContext.startr rG )ÚtyprI rJ r c
|
à s. |d kr@zt | j¡I d H W n tk
|
r4 Y dS X tdnê|d krN| }z| j |||¡I d H W n¸ tk
|
r } z||k W Y ¢S d }~X Y n tk
|
rì } z:||kr¸W Y ¢,dS t|ttfrÚ|j|krÚW Y ¢
|
dS W 5 d }~X Y n6 t k
|
r } z||k r
|
W Y ¢dS d }~X Y nX tdd S )NFzgenerator didn't stopz$generator didn't stop after athrow())
|
r rY rS rZ r[ ÚathrowÚ
|
isinstanceÚ StopIterationÚ __cause__Ú BaseException)r$ r] rI rJ r r" r" r% rK ª s4
|
|
ÿþ
|
|
z#GeneratorStartableContext.__aexit__N)F)r9 r: r; r= r< rX rB rK r" r" r" r% rQ s
|
|
rQ rT z/Callable[..., GeneratorStartableContext[_T_co]])rU r c s$ t ¡dddd 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::
|
|
@asyncstartablecontext
|
async def some_async_generator(<arguments>):
|
<setup>
|
try:
|
yield <value>
|
except GeneratorExit:
|
# return value was awaited, no context manager is present
|
# and caller will .close() the resource explicitly
|
pass
|
else:
|
<context manager cleanup>
|
|
|
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.
|
|
|
r z GeneratorStartableContext[_T_co])rV rW r c s t | |S r! )rQ )rV rW ©rU r" r% Úhelper s z%asyncstartablecontext.<locals>.helper)r+ Úwraps)rU rd r" rc r% Úasyncstartablecontextß s #rf c @ sR e Zd ZdZejddddZddddZd d
|
d d d Zd d
|
d ddZ dS )ÚProxyComparabler" r rC c C s
|
t d S r! r0 rE r" r" r% Ú_proxied s zProxyComparable._proxiedÚintc C s t | S r! )ÚidrE r" r" r% Ú__hash__ s zProxyComparable.__hash__r r6 )Úotherr c C s t || jo| j|jkS r! ©r_ rM rh ©r$ rl r" r" r% Ú__eq__ s
|
þzProxyComparable.__eq__c C s t || j p| j|jkS r! rm rn r" r" r% Ú__ne__ s
|
þzProxyComparable.__ne__N)
|
r9 r: r; r= r Zro_non_memoized_propertyrh rk ro rp r" r" r" r% rg s rg )$Ú
|
__future__r rO r+ Útypingr r r r r r r r
|
r r r r r r r( Ú r rL r Z util.typingr r r r r r ÚABCr? rQ rf rg r" r" r" r% Ú<module> s: LQ*
|
