관리-도구
편집 파일: registry.cpython-38.pyc
U -?�fS* � @ 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 eddd�Z G dd� d�Z 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/d0�Z+d&d-d!d1�d2d3�Z,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 �I/opt/hc_python/lib64/python3.8/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�t�n|| |< | ||< 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 d�dd�Zedd�dd��Zddd�dd�Z d dd�d d �Z d d d d �d d!�Z d d�d"d#�Z d$d�d%d&�Z d8d$d$d$d(d$d d)�d*d+�Zedd�d,d-��Zd.d/d$d0�d1d2�Zd.d/d d0�d3d4�Zd.d/d$d0�d5d6�ZdS )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|tj�r6t|j�t|j�f| _ 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|d�S 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|| jd�S 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 | d�r�| � � |s�|r�| �t j | j|d��j ||� n| jjj| f|�|� d S )N�onceFZ_once_unless_exception�named�_sa_warn)Zretry_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 r` 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 rc 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_asynciorf �append) rR re rf r\ rg rh rB rC ra ri r r r � base_listenM s z_EventKey.base_listenc C s | j p | jS r7 )rF rD rU r r r r3 b s z_EventKey._listen_fnr- zDeque[_ListenerFnType])r0 �list_r% c C s" t | |�r|�| j� dS dS d S �NTF)r6 rj r3 �rR r0 rl r r r �append_to_listf s z_EventKey.append_to_listc C s t | |� |�| j� d S r7 )r8 rc r3 rn r r r �remove_from_listq s z_EventKey.remove_from_listc C s" t | |�r|�| j� dS dS d S rm )r6 � appendleftr3 rn r r r �prepend_to_listy s z_EventKey.prepend_to_list)N)FFFNF)r r r r r r rS �propertyr1 rW rX r` rc rd rk r3 ro rp rr 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 �� ��� $