File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/__pycache__/instrumentation.cpython-310.pyc
o
���g_ � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
Z
dd
lm Z dd lm Z dd lm
Z
dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z dd lm Z dd l m Z dd l m!Z! dd l"m#Z# dd l"m$Z$ er�dd!lm%Z% dd"l m&Z& dd#l m'Z' dd$lm(Z( dd%lm)Z) dd&l*m+Z+ dd'l,m-Z- dd(l.m/Z/ dd)lm0Z0 dd*l m1Z1 ed+ed,�Z2e �3d-�Z4G d.d/� d/e$�Z5G d0d1� d1e$�Z6G d2d3� d3e!ee7d4f e e e �Z8G d5d6� d6�Z9G d7d8� d8e �Z:e:� Z;ej< Z<Z=ej> Z>Z?ej@ Z@ZAejB ZBZC 9 dRdSdJdK�ZDdLdM� ZEdNdO� ZFdPdQ� ZGdS )Ta� Defines SQLAlchemy's system of class instrumentation.
This module is usually not directly visible to user applications, but
defines a large part of the ORM's interactivity.
instrumentation.py deals with registration of end-user classes
for state tracking. It interacts closely with state.py
and attributes.py which establish per-instance and per-class-attribute
instrumentation, respectively.
The class instrumentation system can be customized on a per-class
or global basis using the :mod:`sqlalchemy.ext.instrumentation`
module, which provides the means to build and specify
alternate instrumentation forms.
.. versionchanged: 0.8
The instrumentation extension system was moved out of the
ORM and into the external :mod:`sqlalchemy.ext.instrumentation`
package. When that package is imported, it installs
itself within sqlalchemy.orm so that its more comprehensive
resolution mechanics take effect.
� )� annotations)�Any)�Callable)�cast)�
Collection)�Dict)�Generic)�Iterable)�List)�Optional)�Set)�Tuple)�Type)�
TYPE_CHECKING)�TypeVar)�UnionN� )�base)� collections)�exc)�
interfaces)�state)�_O)� _is_collection_attribute_impl� )�util)� EventTarget)� HasMemoized)�Literal)�Protocol)�
_RegistryType)�
AttributeImpl)�QueryableAttribute)�_AdaptedCollectionProtocol)�_CollectionFactoryType)�
_MapperConfig)�InstanceEvents)�Mapper)�
InstanceState)�
dispatcher�_T)�bound�DEL_ATTRc @ s e Zd Zd d d
�Zd S )
� _ExpiredAttributeLoaderProtor �state.InstanceState[Any]�toload�Set[str]�passive�base.PassiveFlag�return�Nonec C � d S �N� )�selfr r/ r1 r7 r7 �I/usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/instrumentation.py�__call__S s z%_ExpiredAttributeLoaderProto.__call__N)r r. r/ r0 r1 r2 r3 r4 ��__name__�
__module__� __qualname__r: r7 r7 r7 r9 r- R � r- c @ s e Zd Zddd�ZdS ) �_ManagerFactory�class_�Type[_O]r3 �ClassManager[_O]c C r5 r6 r7 �r8 rA r7 r7 r9 r: \ � z_ManagerFactory.__call__N�rA rB r3 rC r; r7 r7 r7 r9 r@ [ r? r@ c @ s� e Zd ZU dZded< ejZejZ e
e � e ��Z
ded< ded< dZded < d
ed < dZd ed
< ded< es9dZded< ded< ee jddd�dd� ��Zeje jddd�dd� ��Zdd� Z d�d�d&d'�Zd�d(d)�Zd�d+d,�Zd�d/d0�Zed�d1d2��Ze j d3d4� �Z e j d5d6� �Z e j d7d8� �Z e j d9d:� �Z!e j"d�d<d=��Z#d�d>d?�Z$d�d@dA�Z%d�dDdE�Z&d�dIdJ�Z'dKdL� Z(e j"d�dNdO��Z)dPdQ� Z*e j+dRdS� �Z,e j+dTdU� �Z-e j+dVdW� �Z. d�d�d[d\�Z/d]d^� Z0d_d`� Z1d�dadb�Z2d�dcdd�Z3d�dedf�Z4d�dgdh�Z5d�djdk�Z6d�dldm�Z7d�dqdr�Z8d�dvdw�Z9d�d�dydz�Z:d�d|d}�Z;ed�dd���Z<d�d�d�d��Z= d�d�d�d��Z>d�d�d��Z?d�d�d��Z@d�d�d��ZAd�d�d��ZB d�d�d�d��ZCd�d�d��ZDd�d�d��ZEdS )�� ClassManagerz,Tracks state information at the class level.z dispatcher[ClassManager[_O]]�dispatchr- �expired_attribute_loader� Optional[Callable[..., None]]� init_methodN�
original_initzOptional[_ManagerFactory]�factoryz$Optional[weakref.ref[_MapperConfig]]�declarative_scanr �registryrB rA zList[ClassManager[Any]]�_basesz1.4zWThe ClassManager.deferred_scalar_loader attribute is now named expired_attribute_loader)�messagec C s | j S r6 �rI �r8 r7 r7 r9 �deferred_scalar_loader� s z#ClassManager.deferred_scalar_loaderc C s
|| _ d S r6 rR )r8 �objr7 r7 r9 rT � s
c C s� || _ i | _d | _i | _i | _d| _d | _d | _dd� tddd� | j j D ��D �| _
| j
D ]}| � |� q-td| j j
��|| � |jD ]}t|�}|d urT| j �|j � qC| �� d|jv rgt�d| � d S d S ) NFc S s g | ]}|d ur|�qS r6 r7 )�.0�mgrr7 r7 r9 �
<listcomp>� s
� z)ClassManager.__init__.<locals>.<listcomp>z!List[Optional[ClassManager[Any]]]c S s g | ] }t |t�rt|��qS r7 )�
isinstance�type�opt_manager_of_class)rV r r7 r7 r9 rX � s ��r& �__del__z�__del__() method on class %s will cause unreachable cycles and memory leaks, as SQLAlchemy instrumentation often creates reference cycles. Please remove this method.)rA �info�new_init� local_attrs� originals�
_finalizedrM rK r � __bases__rP �updaterH �_events�_new_classmanager_instance�__mro__r[ �_update�manage�__dict__r �warn)r8 rA �base_�baseclsrW r7 r7 r9 �__init__� sF ���
�
�
���zClassManager.__init__F�finalize�bool�mapper�Optional[Mapper[_O]]�Optional[_RegistryType]�Optional[_MapperConfig]�&Optional[_ExpiredAttributeLoaderProto]r3 r4 c C s� |r|| _ |r |�| � |rt�|�| _|r|| _|r'| jr$J d| ��|| _| js>| jd ur9| jj t
j u r9| jn| jj | _ |rI| jsK| � � d S d S d S )Nz>class is already instrumented, init_method %s can't be applied)
rp � _add_manager�weakref�refrN rI ra rK rA rm �objectrL � _finalize)r8 rn rp rO rN rI rK r7 r7 r9 �
_update_state� s0
��
��
�zClassManager._update_statec C s* | j rd S d| _ | �� tj�| j� d S �NT)ra �_instrument_init�_instrumentation_factoryrH �class_instrumentrA rS r7 r7 r9 ry � s
zClassManager._finalize�intc C � t | �S r6 )�idrS r7 r7 r9 �__hash__� � zClassManager.__hash__�otherr c C s || u S r6 r7 )r8 r� r7 r7 r9 �__eq__� r� zClassManager.__eq__c C s
d| j v S )Nrp )ri rS r7 r7 r9 � is_mapped� s
zClassManager.is_mappedc C r� r6 )� frozensetrS r7 r7 r9 � _all_key_set� s zClassManager._all_key_setc C � t dd� | �� D ��S )Nc S s g | ] }|j jr|j�qS r7 )�impl�
collection�key�rV �attrr7 r7 r9 rX s z6ClassManager._collection_impl_keys.<locals>.<listcomp>�r� �valuesrS r7 r7 r9 �_collection_impl_keys� s �z"ClassManager._collection_impl_keysc C r� )Nc S s g | ] }|j jr|j �qS r7 )r� �accepts_scalar_loaderr� r7 r7 r9 rX s ��z5ClassManager._scalar_loader_impls.<locals>.<listcomp>r� rS r7 r7 r9 �_scalar_loader_impls s
��z!ClassManager._scalar_loader_implsc C r� )Nc S s g | ]}|j �qS r7 �r� r� r7 r7 r9 rX s z.ClassManager._loader_impls.<locals>.<listcomp>r� rS r7 r7 r9 �
_loader_impls s zClassManager._loader_impls�
Mapper[_O]c C s t �| j��r6 )r �UnmappedClassErrorrA rS r7 r7 r9 rp s zClassManager.mapperc c sx � i }| j jdd� D ].}|jd }|jD ]#}|�||� ||jv r#q|| j| }t|tj�r8|jr8||fV qq dS )a return an iterator of all classbound attributes that are
implement :class:`.InspectionAttr`.
This includes :class:`.QueryableAttribute` as well as extension
types such as :class:`.hybrid_property` and
:class:`.AssociationProxy`.
r ���r N)rA rf ri �
setdefaultrY r �InspectionAttr� is_attribute)r8 �exclude�found�supercls�inheritsr� �valr7 r7 r9 �_all_sqla_attributes s" �
��
���z!ClassManager._all_sqla_attributesc C s* | j jD ]}||jv r|j| S q|S )z5return an attribute on the class without tripping it.)rA rf ri )r8 r� �defaultr� r7 r7 r9 �_get_class_attr_mro8 s
�z ClassManager._get_class_attr_mror� �strc C s || v o
| | j duS )z\Return True if the given attribute is fully initialized.
i.e. has an impl.
Nr� �r8 r� r7 r7 r9 �_attr_has_implA s zClassManager._attr_has_impl�cls�Type[_T]�ClassManager[_T]c C s t |dd�S )a� Create a new ClassManager for a subclass of this ClassManager's
class.
This is called automatically when attributes are instrumented so that
the attributes can be propagated to subclasses against their own
class-local manager, without the need for mappers etc. to have already
pre-configured managers for the full class hierarchy. Mappers
can post-configure the auto-generated ClassManager when needed.
F)rn )�register_class)r8 r� r7 r7 r9 �_subclass_managerI s z ClassManager._subclass_managerc C s$ t | j| | j�| _| �d| j� d S )Nrm )�_generate_initrA rL r^ �install_memberrS r7 r7 r9 r| V s z ClassManager._instrument_init� Type[state.InstanceState[_O]]c C s | j �| | j� tjS r6 )rH �
first_initrA r r( rS r7 r7 r9 �_state_constructorZ s z ClassManager._state_constructorc C s t | j| j| � dS )z0Mark this instance as the manager for its class.N)�setattrrA � MANAGER_ATTRrS r7 r7 r9 rh _ s zClassManager.managec C � t S r6 )�_default_manager_getterrS r7 r7 r9 �manager_getterd � zClassManager.manager_getterc C r� )z�Return a (instance) -> InstanceState callable.
"state getter" callables should raise either KeyError or
AttributeError if no InstanceState could be found for the
instance.
)�_default_state_getterrS r7 r7 r9 � state_getterh s zClassManager.state_getterc C r� r6 )�_default_dict_getterrS r7 r7 r9 � dict_getters r� zClassManager.dict_getter�inst�QueryableAttribute[Any]�
propagatedc C sf |r
|| j v r d S n || j |< | �||� | �� || |< | j�� D ]}| �|�}|�||d� q"d S r{ )r_ �install_descriptor�_reset_memoizationsrA �__subclasses__r� �instrument_attribute)r8 r� r� r� r� �managerr7 r7 r9 r� w s
�
�z!ClassManager.instrument_attributec c sH � | j �� D ]}t|�}|d ur!|| ur!|V |r!|�d�E d H qd S r{ )rA r� r[ �subclass_managers)r8 � recursiver� rW r7 r7 r9 r� � s ���z ClassManager.subclass_managersc C s t j�| j|| | � d S r6 )r} rH �attribute_instrumentrA r� r7 r7 r9 �post_configure_attribute� s �z%ClassManager.post_configure_attributec C sl || vrd S |r|| j v rd S n | j |= | �|� | �� | |= | j�� D ]}t|�}|r3|�|d� q%d S r{ )r_ �uninstall_descriptorr� rA r� r[ �uninstrument_attribute)r8 r� r� r� r� r7 r7 r9 r� � s
�
��z#ClassManager.uninstrument_attributec C s| t | j�D ]}| �|� qd| _d| _d| _| j�� t | �D ] }|| jv r+| � |� q | j
| j j v r<t
| j | j
� dS dS )z<remove all instrumentation established by this ClassManager.N)�listr` �uninstall_memberrp rH r^ r] �clearr_ r� r� rA ri �delattrr� r7 r7 r9 �
unregister� s
��zClassManager.unregisterc C s. || j | jfv rtd| ��t| j||� d S �NzW%r: requested attribute name conflicts with instrumentation attribute of the same name.)�
STATE_ATTRr� �KeyErrorr� rA )r8 r� r� r7 r7 r9 r� � s ��z ClassManager.install_descriptorc C s t | j|� d S r6 )r� rA r� r7 r7 r9 r� � � z!ClassManager.uninstall_descriptor�implementationc C sH || j | jfv rtd| ��| j�|| jj�|t�� t | j||� d S r� )
r� r� r� r` r� rA ri �getr, r� )r8 r� r� r7 r7 r9 r� � s ��zClassManager.install_memberc C s8 | j �|d �}|turt| j||� d S t| j|� d S r6 )r` �popr, r� rA r� )r8 r� �originalr7 r7 r9 r� � s z ClassManager.uninstall_member�collection_class�Type[Collection[Any]]r$ c C s
t �|�S r6 )r �prepare_instrumentation)r8 r� r� r7 r7 r9 �instrument_collection_class� �
z(ClassManager.instrument_collection_classr �InstanceState[_O]�@Tuple[collections.CollectionAdapter, _AdaptedCollectionProtocol]c C s2 |� }| � |�}t|�sJ �t�|||�}||fS r6 )�get_implr r �CollectionAdapter)r8 r� r rM � user_datar� �adapterr7 r7 r9 �initialize_collection� s
z"ClassManager.initialize_collection�searchc C s |r|| v S || j v S r6 )r_ )r8 r� r� r7 r7 r9 �is_instrumented� s
z ClassManager.is_instrumentedr! c C s
| | j S r6 r� r� r7 r7 r9 r� � s
zClassManager.get_impl�
Iterable[Any]c C s t | �� �S r6 )�iterr� rS r7 r7 r9 �
attributes� s zClassManager.attributes�Optional[InstanceState[_O]]r c C s2 | j �| j �}|d u r| �|| �}| �||� |S r6 )rA �__new__r� �
_state_setter)r8 r �instancer7 r7 r9 � new_instance� s
zClassManager.new_instancer� c C s$ |d u r
| � || �}| �||� d S r6 )r� r� �r8 r� r r7 r7 r9 �setup_instance s zClassManager.setup_instancec C s t || j� d S r6 )r� r� �r8 r� r7 r7 r9 �teardown_instance r� z ClassManager.teardown_instance�
state_dict�Dict[str, Any]�_SerializeManagerc C s
t ||�S r6 )r� )r8 r r� r7 r7 r9 �
_serialize r� zClassManager._serialize�(Union[Literal[False], InstanceState[_O]]c C sP t || j�rdS | j|jur| jr| �|j��|�S | �|| �}| �||� |S )z�Install a default InstanceState if none is present.
A private convenience method used by the __init__ decorator.
F) �hasattrr� rA � __class__r� r� �_new_state_if_noner� r� r� r7 r7 r9 r� s �� z ClassManager._new_state_if_nonec C s t || j�S r6 )r� r� r� r7 r7 r9 � has_state( s zClassManager.has_state�
optimisticc C s | � |�j||d�S )�TODO)r� )r� � hasparent)r8 r r� r� r7 r7 r9 �
has_parent+ s zClassManager.has_parentc C � dS )z=All ClassManagers are non-zero regardless of attribute state.Tr7 rS r7 r7 r9 �__bool__1 r� zClassManager.__bool__c C s d| j j| jt| �f S )Nz<%s of %r at %x>)r� r< rA r� rS r7 r7 r9 �__repr__5 s
�zClassManager.__repr__)FNNNNN)rn ro rp rq rO rr rN rs rI rt rK rJ r3 r4 )r3 r4 )r3 r )r� r r3 ro )r3 ro )r3 r� r6 )r� r� r3 ro )r� r� r3 r� )r3 r� )F)r� r� r� r� r� ro r3 r4 )r� r� r� r� r3 r4 )r� r� r3 r4 )r� r� r� r r3 r4 )r� r� r� r� r3 r$ )r� r� r r� rM r$ r3 r� )r� r� r� ro r3 ro )r� r� r3 r! )r3 r� )r r� r3 r )r� r r r� r3 r4 )r� r r3 r4 )r r� r� r� r3 r� )r� r r3 r� )r� r r3 ro )r r� r� r� r� ro r3 ro )r3 r� )Fr< r= r> �__doc__�__annotations__r �DEFAULT_MANAGER_ATTRr� �DEFAULT_STATE_ATTRr� � staticmethodr �
attrsetterr� rL rN r rO �property�
deprecatedrT �setterrm rz ry r� r� r� r �memoized_attributer� r� r� r� �memoized_propertyrp r� r� r� r� r| r� rh � hybridmethodr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r7 r7 r7 r9 rG _ s�
� � / �
&
!
�
�
�
rG r� c @ s" e Zd ZdZd dd�Zdd � Zd
S ) r� z�Provide serialization of a :class:`.ClassManager`.
The :class:`.InstanceState` uses ``__init__()`` on serialize
and ``__call__()`` on deserialize.
r r. �dr� c C s |j | _ |j}|j�||� d S r6 )rA r� rH �pickle)r8 r r r� r7 r7 r9 rm E s z_SerializeManager.__init__c C sh t | j� |_}|d u rt�|d| j ��|jr!|jjs!|j�� |d ur+|� ||� |j
� ||� d S )NzxCannot deserialize object of type %r - no mapper() has been configured for this class within the current Python process!) r[ rA r� r �UnmappedInstanceErrorr� rp �
configured�_check_configurer� rH �unpickle)r8 r r� r� r� r7 r7 r9 r: J s ��
z_SerializeManager.__call__N)r r. r r� )r< r= r> r rm r: r7 r7 r7 r9 r� = s
r� c @ sB e Zd ZU dZded< ddd �Zdd d �Zddd�Zddd�ZdS )�InstrumentationFactoryz'Factory for new ClassManager instances.z"dispatcher[InstrumentationFactory]rH rA rB r3 rC c C sb |d usJ �t |�d u sJ �| �|�\}}|d u r t}t|�}n|d us&J �| �||� ||_|S r6 )r[ �_locate_extended_factoryrG �_check_conflictsrM )r8 rA r� rM r7 r7 r9 �create_manager_for_clsd s
z-InstrumentationFactory.create_manager_for_cls�<Tuple[Optional[ClassManager[_O]], Optional[_ManagerFactory]]c C r� )z2Overridden by a subclass to do an extended lookup.)NNr7 rD r7 r7 r9 r x s z/InstrumentationFactory._locate_extended_factoryrM �&Callable[[Type[_O]], ClassManager[_O]]r4 c C r� )z;Overridden by a subclass to test for conflicting factories.Nr7 )r8 rA rM r7 r7 r9 r ~ rE z'InstrumentationFactory._check_conflictsc C s t |�}|�� | j�|� d S r6 )�manager_of_classr� rH �class_uninstrument)r8 rA r� r7 r7 r9 r� � s z!InstrumentationFactory.unregisterNrF )rA rB r3 r )rA rB rM r r3 r4 )rA rB r3 r4 ) r< r= r> r r r r r r� r7 r7 r7 r9 r _ s
r TrA rB rn ro rp rq rO rr rN rs rI rt rK rJ r3 rC c C s4 t | �}|du r
t�| �}|j||||||d� |S )z_Register class instrumentation.
Returns the existing or newly created class manager.
N)rp rO rN rI rK rn )r[ r} r rz )rA rn rp rO rN rI rK r� r7 r7 r9 r� � s
� r� c C s t �| � dS )z!Unregister class instrumentation.N)r} r� )rA r7 r7 r9 �unregister_class� s r c C s t | j�j|dd�S )z�Return True if the given attribute on the given instance is
instrumented by the attributes package.
This function may be used regardless of instrumentation
applied directly to the class, i.e. no descriptors are required.
T)r� )r r� r� )r� r� r7 r7 r9 r� � s
�r� c
C s� |du r| j }d}tj|dd�}|| }t|dd�}t|dd�}t� �� }t|d< t||� |d } |j| _|| _ |r>|| _
|rC|| _ | S ) z>Build an __init__ decorator that triggers ClassManager events.Nz�def __init__(%(apply_pos)s):
new_state = class_manager._new_state_if_none(%(self_arg)s)
if new_state:
return new_state._initialize_instance(%(apply_kw)s)
else:
return original_init(%(apply_kw)s)
F)�grouped� __defaults__�__kwdefaults__r< rm ) rm r �format_argspec_init�getattr�locals�copyr< �execr �_sa_original_initr r )
rA �
class_managerrL � func_body� func_vars� func_text�
func_defaults�func_kw_defaults�envrm r7 r7 r9 r� � s$
r� )TNNNNN)rA rB rn ro rp rq rO rr rN rs rI rt rK rJ r3 rC )Hr �
__future__r �typingr r r r r r r r
r r r
r r r r rv � r r r r r �_typingr r� r r �eventr r � util.typingr r r r! r" r# r$ � decl_baser% �eventsr& rp r' r( r) r* �symbolr, r- r@ r� rG r� r r} �instance_stater� �
instance_dictr� r r� r[ �_default_opt_manager_getterr� r r� r� r7 r7 r7 r9 �<module> s�
� a",
�