File: //usr/local/lib/python3.10/dist-packages/sqlalchemy/ext/__pycache__/instrumentation.cpython-310.pyc
o
���g[= � @ s d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z
dd lm Z ddlmZ
dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ dZ dd� ZegZ G dd� de�Ze� e _Zee _G dd� d�ZG dd� de�Zdd� Z dd � Z d d � Z dS ) a� Extensible class instrumentation.
The :mod:`sqlalchemy.ext.instrumentation` package provides for alternate
systems of class instrumentation within the ORM. Class instrumentation
refers to how the ORM places attributes on the class which maintain
data and track changes to that data, as well as event hooks installed
on the class.
.. note::
The extension package is provided for the benefit of integration
with other object management packages, which already perform
their own instrumentation. It is not intended for general use.
For examples of how the instrumentation extension is used,
see the example :ref:`examples_instrumentation`.
� N� )�util)�
attributes)�base)� collections)�exc)�instrumentation)�_default_dict_getter)�_default_manager_getter)�_default_opt_manager_getter)�_default_state_getter)� ClassManager)�InstrumentationFactory� __sa_instrumentation_manager__c C s t | td�S )z;Find user-specified instrumentation management for a class.N)�getattr�INSTRUMENTATION_MANAGER��cls� r �I/usr/local/lib/python3.10/dist-packages/sqlalchemy/ext/instrumentation.py�%find_native_user_instrumentation_hookE s r c s| e Zd ZdZe�� Ze�� Ze�� ZdZ dd� Z
dd� Z dd� Z d d
� Z
� fd d �Zd
d� Zdd� Zdd� Zdd� Z� ZS )� ExtendedInstrumentationRegistryz�Extends :class:`.InstrumentationFactory` with additional
bookkeeping, to accommodate multiple types of
class managers.
Fc C s6 t D ]}||�}|d ur| �||�}||f S qdS )N)NN)�instrumentation_finders�_extended_class_manager)�self�class_�finder�factory�managerr r r �_locate_extended_factoryg s �z8ExtendedInstrumentationRegistry._locate_extended_factoryc C s0 | � |��|g�}|rtd|jt|�f ��d S )NzRmultiple instrumentation implementations specified in %s inheritance hierarchy: %r)�!_collect_management_factories_for�
difference� TypeError�__name__�list)r r r �existing_factoriesr r r �_check_conflictsp s �� ���z0ExtendedInstrumentationRegistry._check_conflictsc C sd ||�}t |t�st||�}|tkr| jsd| _t� |�� | j|< |�� | j|< |� � | j
|< |S )NT) �
isinstancer
� _ClassInstrumentationAdapter� _extended� _install_instrumented_lookups�manager_getter�_manager_finders� state_getter�_state_finders� dict_getter�
_dict_finders)r r r r r r r r { s
z7ExtendedInstrumentationRegistry._extended_class_managerc C st t �|�}t� }|D ](}| �|�}|dur |�|j� q
tD ] }||�}|dur* nq d}|�|� q
|�d� |S )ac Return a collection of factories in play or specified for a
hierarchy.
Traverses the entire inheritance graph of a cls and returns a
collection of instrumentation factories for those classes. Factories
are extracted from active ClassManagers, if available, otherwise
instrumentation_finders is consulted.
N)r �class_hierarchy�set�opt_manager_of_class�addr r �discard)r r � hierarchy� factories�memberr r r r r r r � s
�
zAExtendedInstrumentationRegistry._collect_management_factories_forc s6 t � �|� || jv r| j|= | j|= | j|= d S d S �N)�super�
unregisterr, r. r0 �r r �� __class__r r r; � s
�z*ExtendedInstrumentationRegistry.unregisterc C s. z | j �|t�}W ||�S ty Y d S w r9 )r, �getr r" )r r r r r r r3 � s � ��z4ExtendedInstrumentationRegistry.opt_manager_of_classc C sZ z | j �|t�}W n ty t�|d|� ���w ||�}|d u r+t�|d|� ���|S )Nz2Can't locate an instrumentation manager for class )r, r? r
r" �orm_exc�UnmappedClassError)r r r r r r r �manager_of_class� s
���z0ExtendedInstrumentationRegistry.manager_of_classc C �$ |d u rt d��| j�|jt�|�S �Nz None has no persistent state.)�AttributeErrorr. r? r> r �r �instancer r r �state_of� � ��z(ExtendedInstrumentationRegistry.state_ofc C rC rD )rE r0 r? r> r rF r r r �dict_of� rI z'ExtendedInstrumentationRegistry.dict_of)r# �
__module__� __qualname__�__doc__�weakref�WeakKeyDictionaryr, r. r0 r) r r& r r r; r3 rB rH rJ �
__classcell__r r r= r r [ s r c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$S )%�InstrumentationManageraW User-defined class instrumentation extension.
:class:`.InstrumentationManager` can be subclassed in order
to change
how class instrumentation proceeds. This class exists for
the purposes of integration with other object management
frameworks which would like to entirely modify the
instrumentation methodology of the ORM, and is not intended
for regular usage. For interception of class instrumentation
events, see :class:`.InstrumentationEvents`.
The API for this class should be considered as semi-stable,
and may change slightly with new releases.
c C � d S r9 r r< r r r �__init__� � z InstrumentationManager.__init__c C s t |d|� d S �N�_default_class_manager��setattr�r r r r r r �manage� � z InstrumentationManager.managec C s t |d� d S rU ��delattrrY r r r r; � � z!InstrumentationManager.unregisterc C s dd� }|S )Nc S � | j S r9 )rV r r r r r? � � z2InstrumentationManager.manager_getter.<locals>.getr )r r r? r r r r+ � s z%InstrumentationManager.manager_getterc C rR r9 r �r r �key�instr r r �instrument_attribute rT z+InstrumentationManager.instrument_attributec C rR r9 r ra r r r �post_configure_attribute rT z/InstrumentationManager.post_configure_attributec C � t |||� d S r9 rW ra r r r �install_descriptor r[ z)InstrumentationManager.install_descriptorc C � t ||� d S r9 r\ �r r rb r r r �uninstall_descriptor r^ z+InstrumentationManager.uninstall_descriptorc C rf r9 rW )r r rb �implementationr r r �install_member r[ z%InstrumentationManager.install_memberc C rh r9 r\ ri r r r �uninstall_member r^ z'InstrumentationManager.uninstall_memberc C s
t �|�S r9 )r �prepare_instrumentation)r r rb �collection_classr r r �instrument_collection_class s
z2InstrumentationManager.instrument_collection_classc C s |j S r9 )�__dict__�r r rG r r r �get_instance_dict r` z(InstrumentationManager.get_instance_dictc C rR r9 r rr r r r �initialize_instance_dict rT z/InstrumentationManager.initialize_instance_dictc C s t |d|� d S �N�_default_staterW )r r rG �stater r r �
install_state r[ z$InstrumentationManager.install_statec C s t |d� d S ru r\ rr r r r � remove_state! r^ z#InstrumentationManager.remove_statec C s dd� S )Nc S s
t | d�S ru )r )rG r r r �<lambda>% s
z5InstrumentationManager.state_getter.<locals>.<lambda>r r< r r r r- $ s z#InstrumentationManager.state_getterc s � �fdd�S )Nc s �� � | �S r9 )rs )rc �r r r r rz ( s z4InstrumentationManager.dict_getter.<locals>.<lambda>r r< r r{ r r/ ' r^ z"InstrumentationManager.dict_getterN)r# rK rL rM rS rZ r; r+ rd re rg rj rl rm rp rs rt rx ry r- r/ r r r r rQ � s& rQ c s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd*d d �Z� fd
d�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd+d d �Zd d � Zd+d d!�Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z� ZS ),r( z?Adapts a user-defined InstrumentationManager to a ClassManager.c C s2 || _ | j �|�| _| j �|�| _t�| |� d S r9 )�_adaptedr- �
_get_stater/ � _get_dictr
rS )r r �overrider r r rS . s z%_ClassInstrumentationAdapter.__init__c C � | j �| j| � d S r9 )r| rZ r �r r r r rZ 5 � z#_ClassInstrumentationAdapter.managec C r� r9 )r| r; r r� r r r r; 8 r� z'_ClassInstrumentationAdapter.unregisterc C s | j �| j�S r9 )r| r+ r r� r r r r+ ; r^ z+_ClassInstrumentationAdapter.manager_getterFc C s. t �| |||� |s| j�| j||� d S d S r9 )r
rd r| r )r rb rc �
propagatedr r r rd > s �z1_ClassInstrumentationAdapter.instrument_attributec s& t � �|� | j�| j|| | � d S r9 )r: re r| r �r rb r= r r re C s z5_ClassInstrumentationAdapter.post_configure_attributec C � | j �| j||� d S r9 )r| rg r )r rb rc r r r rg G � z/_ClassInstrumentationAdapter.install_descriptorc C � | j �| j|� d S r9 )r| rj r r� r r r rj J r� z1_ClassInstrumentationAdapter.uninstall_descriptorc C r� r9 )r| rl r )r rb rk r r r rl M r� z+_ClassInstrumentationAdapter.install_memberc C r� r9 )r| rm r r� r r r rm P r� z-_ClassInstrumentationAdapter.uninstall_memberc C s | j �| j||�S r9 )r| rp r )r rb ro r r r rp S s �z8_ClassInstrumentationAdapter.instrument_collection_classc C s. t | jdd �}|r||||�S t�| |||�S )N�initialize_collection)r r| r
r� )r rb rw r �delegater r r r� X s �z2_ClassInstrumentationAdapter.initialize_collectionNc C s | j �| j �}| �||� |S r9 )r �__new__�setup_instance)r rw rG r r r � new_instancea s z)_ClassInstrumentationAdapter.new_instancec C s | � |�rdS | �|�S )z�Install a default InstanceState if none is present.
A private convenience method used by the __init__ decorator.
F)� has_stater� rF r r r �_new_state_if_nonef s
z/_ClassInstrumentationAdapter._new_state_if_nonec C s: | j �| j|� |d u r| �|| �}| j �| j||� |S r9 )r| rt r �_state_constructorrx )r rG rw r r r r� p s
z+_ClassInstrumentationAdapter.setup_instancec C r� r9 )r| ry r rF r r r �teardown_instancez r� z._ClassInstrumentationAdapter.teardown_instancec C s( z| � |� W dS tjy Y dS w )NFT)r} r@ �NO_STATErF r r r r� } s ��z&_ClassInstrumentationAdapter.has_statec C r_ r9 )r} r� r r r r- � r` z)_ClassInstrumentationAdapter.state_getterc C r_ r9 )r~ r� r r r r/ � r` z(_ClassInstrumentationAdapter.dict_getter)Fr9 )r# rK rL rM rS rZ r; r+ rd re rg rj rl rm rp r� r� r� r� r� r� r- r/ rP r r r= r r( + s*
r( c C s t ttjtjtjtjd�� dS )a� Replace global class/object management functions
with ExtendedInstrumentationRegistry implementations, which
allow multiple types of class managers to be present,
at the cost of performance.
This function is called only by ExtendedInstrumentationRegistry
and unit tests specific to this behavior.
The _reinstall_default_lookups() function can be called
after this one to re-establish the default functions.
��instance_state�
instance_dictrB r3 N)�_install_lookups�dict�_instrumentation_factoryrH rJ rB r3 r r r r r* � s
��r* c C s t tttttd�� dt_dS )zRestore simplified lookups.r� FN)r� r� r r r
r r� r) r r r r �_reinstall_default_lookups� s ��
r� c C sr | d a | d a| d a| d at t_ t_ t_ t t_ t_t_t t_ t_t_t t_ t_ t_t_d S )Nr� r� rB r3 )r� r� rB r3 �orm_baser �orm_instrumentation�orm_util)�lookupsr r r r� � s" � � ��r� ) rM rN � r �ormr r r� r r r@ r r� r� �orm.instrumentationr r
r r r
r r r r r r� rQ r( r* r� r� r r r r �<module> s: �La