File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/event/__pycache__/registry.cpython-310.pyc
o
���g�+ � @ 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 ejrvddlmZ 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 < d9d$d%�Z(d:d+d,�Z)d;d-d.�Z*d<d3d4�Z+d=d5d6�Z,G d7d8� d8e e �Z-dS )>aJ 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 �D/usr/local/lib/python3.10/dist-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_key�ref�weakref.ref[Any]�return�Nonec C s^ t r| t vrd S td| �} t �| �}|�� D ]}|tv r,t| }|�| � |s,t�|� qd S )Nz'weakref.ref[RefCollection[EventTarget]])r" r �pop�valuesr! )r# �listener_to_key�key� dispatch_regr r r �_collection_gcedg s
��r, � event_key�_EventKey[_ET]�owner�RefCollection[_ET]�boolc C sH | j }t| }|j}t�| j�}||v rdS |||< t| }|||< dS )NFT)�_keyr! r# �weakref�
_listen_fnr" )r- r/ r* r+ � owner_ref�
listen_refr) r r r �_stored_in_collectionx s r7 c C sZ | j }t| }t�| j�}|j}|�|d � |st|= |tv r+t| }|�|d � d S d S �N)r2 r! r3 r# r4 r' r" )r- r/ r* r+ r6 r5 r) r r r �_removed_from_collection� s �r9 �newowner�oldowner�elements�Iterable[_ListenerFnType]c C s� |sd S |j }| j }t| }t| }|D ]<}t� |�}z|| } W n ty* Y qw zt| }
W n ty: Y qw ||
v rH|
| |ksGJ �n||
|< | ||< qd S r8 )r# r" r3 �KeyErrorr! ) r: r; r< � oldowner_ref� newowner_ref�old_listener_to_key�new_listener_to_key� listen_fnr6 r* r+ r r r �_stored_in_collection_multi� s.
� �
�rD c C sT |sd S | j }t| }|D ]}t� |�}|| }t| }|�|d � |s't|= q
d S r8 )r# r" r3 r! r' )r/ r< r5 r) rC r6 r* r+ r r r �_clear� s
��rE c @ s� e Zd ZU dZdZded< ded< ded< d ed
< d ed < d
ed< d;d<dd�Zed=dd��Zd>dd�Z d?dd�Z
d@d d �Z dAd!d"�Z dBd$d%�Z
& & & &dCdDd-d.�ZedEd/d0��ZdFd5d6�ZdGd7d8�ZdFd9d:�ZdS )H� _EventKeyz$Represent :func:`.listen` arguments.)�target�
identifier�fn�fn_key�fn_wrap�dispatch_targetr rG �strrH �_ListenerFnTyperI �_ListenerFnKeyTyperJ r rL �Optional[_ListenerFnType]�_fn_wrapNc C sP || _ || _|| _t|tj�rt|j�t|j�f| _ nt|�| _ || _
|| _ d S r8 ) rG rH rI �
isinstance�types�
MethodType�id�__func__�__self__rJ rK rL )�selfrG rH rI rL rQ r r r �__init__� s
z_EventKey.__init__r% �_EventKeyTupleTypec C s t | j�| j| jfS r8 )rU rG rH rJ �rX r r r r2 s z_EventKey._keyrK r. c C s( || j u r| S t| j| j| j| j|d�S �N)rQ )r4 rF rG rH rI rL )rX rK r r r � with_wrapper s
�z_EventKey.with_wrapperc C s( || j u r| S t| j| j| j|| jd�S r\ )rL rF rG rH rI rK )rX rL r r r �with_dispatch_target s
�z _EventKey.with_dispatch_target�args�kwr& c O s� |� dd�}|� dd�}|� dd�}| j| j| j}}}t|j|�} | �||�}
| �|
�} t| jjj| j�} t | d�r@| �
� |sD|rW| �t j | j|d��j
|i |�� d S | jjj| g|�R i |�� d S )N�onceF�_once_unless_exception�named�_sa_warn)�retry_on_exception)r' rL rH r4 �getattrr �_adjust_fn_specr] �_events�hasattrrd r � only_once�listen�_listen) rX r_ r` ra �once_unless_exceptionrc rG rH rI �dispatch_collection� adjusted_fn�
stub_functionr r r rk s4
�
�
���
� z_EventKey.listenc C st | j }|tvrt�d| j| j| jf ��t�|�}|�� D ]\}}|� }|� }|d ur7|d ur7|� | �
|�� q d S )Nz*No listeners found for event %s / %r / %s ) r2 r! r �InvalidRequestErrorrG rH rI r' �items�remover] )rX r* r+ �collection_ref� listener_ref�
collection� listener_fnr r r rs < s ��
��z_EventKey.remover1 c C s
| j tv S )z6Return True if this event key is registered to listen.)r2 r! r[ r r r �containsM s
z_EventKey.containsF� propagate�insertrc �retval�Optional[bool]�asyncioc
C sV | j | j}}t|j|�}|�|j�} |r| �� |r#| �| |� d S | �| |� d S r8 )rL rH rf r �
for_modify� _set_asynciorz �append)
rX ry rz rc r{ r} rG rH rn r~ r r r � base_listenQ s z_EventKey.base_listenc C s | j p| jS r8 )rK rI r[ r r r r4 f s z_EventKey._listen_fnr/ r0 �list_�Deque[_ListenerFnType]c C � t | |�r
|�| j� dS dS �NTF)r7 r� r4 �rX r/ r� r r r �append_to_listj �
z_EventKey.append_to_listc C s t | |� |�| j� d S r8 )r9 rs r4 r� r r r �remove_from_listu s
z_EventKey.remove_from_listc C r� r� )r7 �
appendleftr4 r� r r r �prepend_to_list} r� z_EventKey.prepend_to_listr8 )
rG r rH rM rI rN rL r rQ rP )r% rZ )rK rN r% r. )rL r r% r. )r_ r r` r r% r& )r% r& )r% r1 )FFFNF) ry r1 rz r1 rc r1 r{ r| r} r1 r% r& )r% rN )r/ r0 r� r� r% r1 )r/ r0 r� r� r% r& )r r r r r r rY �propertyr2 r] r^ rk rs rx r� r4 r� r� r� r r r r rF � s:
�
�
rF )r# r$ r% r& )r- r. r/ r0 r% r1 )r- r. r/ r0 r% r& )r: r0 r; r0 r< r= r% r& )r/ r0 r< r= r% r& ).r �
__future__r � collectionsrS �typingr r r r r r r r
r r r
r3 � r r �
TYPE_CHECKING�attrr �baser rN �intrO rM rZ r r � _RefCollectionToListenerType� defaultdict�dictr! r �_ListenerToEventKeyTyper" r, r7 r9 rD rE rF r r r r �<module> s^ �
� ��
�
$