U
|
¸ý°dñ+ ã @ sÞ U d Z ddlmZ ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
|
Z
|
dd lm Z dd
|
lm Z dd lm Z dd lmZ dd lmZ ddlmZ ddlZddlmZ ddlmZ ejrìddlmZ ddlmZ edef Zeeeeef f Z eee e f Z edddZ G dd dZ e
|
d Z!e "e#¡Z$de%d < e
|
d e f Z&e "e#¡Z'd e%d < d d!d"d#d$Z(d%d&d'd(d)d*Z)d%d&d!d(d+d,Z*d&d&d-d!d.d/d0Z+d&d-d!d1d2d3Z,G d4d5 d5e e Z-dS )6aJ Provides managed registration services on behalf of :func:`.listen`
|
arguments.
|
|
By "managed registration", we mean that event listening functions and
|
other objects can be added to various collections in such a way that their
|
membership in all those collections can be revoked at once, based on
|
an equivalent :class:`._EventKey`.
|
|
é )Ú annotationsN)ÚAny)ÚCallable)Úcast)ÚDeque)ÚDict)ÚGeneric)ÚIterable)ÚOptional)ÚTuple)ÚTypeVar)ÚUnioné )Úexc)Úutilé )Ú RefCollection)Ú
|
dispatcher.Ú_ETÚ EventTarget)Úboundc @ s e Zd ZU dZdZded< dS )r zírepresents an event target, that is, something we can listen on
|
either with that target as a class or as an instance.
|
|
Examples include: Connection, Mapper, Table, Session,
|
InstrumentedAttribute, Engine, Pool, Dialect.
|
|
© zdispatcher[Any]ÚdispatchN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú __slots__Ú__annotations__r r r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/event/registry.pyr 2 s
|
)z weakref.ref[RefCollection[Any]]ú weakref.ref[_ListenerFnType]z6Dict[_EventKeyTupleType, _RefCollectionToListenerType]Ú_key_to_collectionr z>Dict[weakref.ref[RefCollection[Any]], _ListenerToEventKeyType]Ú_collection_to_keyzweakref.ref[Any]ÚNone)ÚrefÚreturnc C s^ t r | t krd S td| } t | ¡}| ¡ D ],}|tkr,t| }| | ¡ |s,t |¡ q,d S )Nz'weakref.ref[RefCollection[EventTarget]])r" r ÚpopÚvaluesr! )r$ Úlistener_to_keyÚkeyÚ dispatch_regr r r Ú_collection_gcedg s
|
|
|
r+ ú_EventKey[_ET]úRefCollection[_ET]Úbool)Ú event_keyÚownerr% c C sH | j }t| }|j}t | j¡}||kr,dS |||< t| }|||< dS )NFT)Ú_keyr! r$ ÚweakrefÚ
|
_listen_fnr" )r/ r0 r) r* Ú owner_refÚ
|
listen_refr( r r r Ú_stored_in_collectionx s r6 c C sT | j }t| }t | j¡}|j}| |d ¡ |s6t|= |tkrPt| }| |¡ d S ©N)r1 r! r2 r$ r3 r& r" )r/ r0 r) r* r5 r4 r( r r r Ú_removed_from_collection s r8 zIterable[_ListenerFnType])ÚnewownerÚoldownerÚelementsr% c C s² |sd S |j }| j }t| }t| }|D ]}t |¡}z || } W n tk
|
rZ Y q(Y nX z t| }
|
W n tk
|
r Y q(Y nX ||
|
kr|
|
| |ks¤tn||
|
|< | ||< q(d S r7 )r$ r" r2 ÚKeyErrorr! ÚAssertionError) r9 r: r; Z oldowner_refZ newowner_refZold_listener_to_keyZnew_listener_to_keyÚ listen_fnr5 r) r* r r r Ú_stored_in_collection_multi s(
|
|
|
r? )r0 r; r% c C sT |sd S | j }t| }|D ]4}t |¡}|| }t| }| |d ¡ |st|= qd S r7 )r$ r" r2 r! r& )r0 r; r4 r( r> r5 r) r* r r r Ú_clearÄ s
|
r@ c @ s e Zd ZU dZdZded< ded< ded< d ed
|
< d ed < d ed< d7dddd d dddZeddddZdddddZ d ddd d Z
|
d d d d d d!Z d dd"d#Z d$dd%d&Z d8d$d$d$d(d$d d)d*d+Zeddd,d-Zd.d/d$d0d1d2Zd.d/d d0d3d4Zd.d/d$d0d5d6ZdS )9Ú _EventKeyz$Represent :func:`.listen` arguments.)ÚtargetÚ
|
identifierÚfnÚfn_keyÚfn_wrapÚdispatch_targetr rB ÚstrrC Ú_ListenerFnTyperD Ú_ListenerFnKeyTyperE r rG zOptional[_ListenerFnType]Ú_fn_wrapN)rB rC rD rG rK c C sP || _ || _|| _t|tjr6t|jt|jf| _ n
|
t|| _ || _
|
|| _ d S r7 ) rB rC rD Ú
|
isinstanceÚtypesÚ
|
MethodTypeÚidÚ__func__Ú__self__rE rF rG )ÚselfrB rC rD rG rK r r r Ú__init__ê s
|
z_EventKey.__init__Ú_EventKeyTupleType)r% c C s t | j| j| jfS r7 )rO rB rC rE ©rR r r r r1 ü s z_EventKey._keyr, )rF r% c C s, || j kr| S t| j| j| j| j|dS d S ©N)rK )r3 rA rB rC rD rG )rR rF r r r Ú with_wrapper s
|
ûz_EventKey.with_wrapper)rG r% c C s, || j kr| S t| j| j| j|| jdS d S rV )rG rA rB rC rD rF )rR rG r r r Úwith_dispatch_target s
|
ûz _EventKey.with_dispatch_targetr# )ÚargsÚkwr% c O s | dd¡}| dd¡}| dd¡}| j| j| j }}}t|j|} | ||¡}
|
| |
|
¡} t| jjj| j} t | dr|
|
¡ |s|r¨| t j | j|d¡j || n| jjj| f|| d S )NÚonceFZ_once_unless_exceptionÚnamedÚ_sa_warn)Úretry_on_exception)r& rG rC r3 Úgetattrr Z_adjust_fn_specrW Z_eventsÚhasattrr] r Z only_onceÚlistenZ_listen) rR rY rZ r[ Zonce_unless_exceptionr\ rB rC rD Údispatch_collectionZ adjusted_fnZ stub_functionr r r ra s6 ý
|
|
ÿ
|
ÿÿ üz_EventKey.listenc C st | j }|tkr(t d| j| j| jf ¡t |¡}| ¡ D ]4\}}| }| }|d k r:|d k r:| |
|
|¡¡ q:d S )Nz*No listeners found for event %s / %r / %s ) r1 r! r ZInvalidRequestErrorrB rC rD r& ÚitemsÚremoverW )rR r) r* Zcollection_refZ listener_refZ
|
collectionZ listener_fnr r r rd 8 s ÿÿ
|
z_EventKey.remover. c C s
|
| j tkS )z6Return True if this event key is registered to listen.)r1 r! rU r r r ÚcontainsI s z_EventKey.containsFzOptional[bool])Ú propagateÚinsertr\ ÚretvalÚasyncior% c
|
C sT | j | j }}t|j|}| |j¡} |r2| ¡ |rD| | |¡ n | | |¡ d S r7 )rG rC r_ r Ú
|
for_modifyZ _set_asynciorg Úappend)
|
rR rf rg r\ rh ri rB rC rb rj r r r Ú base_listenM s z_EventKey.base_listenc C s | j p
|
| jS r7 )rF rD rU r r r r3 c s z_EventKey._listen_fnr- zDeque[_ListenerFnType])r0 Úlist_r% c C s" t | |r| | j¡ dS dS d S ©NTF)r6 rk r3 ©rR r0 rm r r r Úappend_to_listg s
|
z_EventKey.append_to_listc C s t | | | | j¡ d S r7 )r8 rd r3 ro r r r Úremove_from_listr s
|
z_EventKey.remove_from_listc C s" t | |r| | j¡ dS dS d S rn )r6 Ú
|
appendleftr3 ro r r r Úprepend_to_listz s
|
z_EventKey.prepend_to_list)N)FFFNF)r r r r r r rS Úpropertyr1 rW rX ra rd re rl r3 rp rq rs r r r r rA × s8
|
ú ú rA ).r Ú
|
__future__r Ú collectionsrM Útypingr r r r r r r r
|
r r r r2 Ú r r Ú TYPE_CHECKINGÚattrr Úbaser rI ÚintrJ rH rT r r Z _RefCollectionToListenerTypeÚ defaultdictÚdictr! r Z_ListenerToEventKeyTyper" r+ r6 r8 r? r@ rA r r r r Ú<module> sX ÿþ
|
ÿÿý
|
$
|
