관리-도구
편집 파일: instrumentation.cpython-38.pyc
U -?�f_ � @ 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$ e�r�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 ZBZCdKd:d;d<d=d>d?d@dAdB�dCdD�ZDdEdF� ZEdGdH� ZFdIdJ� ZGdS )La� 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 Zddddd�dd�ZdS ) � _ExpiredAttributeLoaderProto�state.InstanceState[Any]zSet[str]zbase.PassiveFlag�None)r �toload�passive�returnc C s d S �N� )�selfr r0 r1 r4 r4 �N/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/instrumentation.py�__call__S s z%_ExpiredAttributeLoaderProto.__call__N��__name__� __module__� __qualname__r7 r4 r4 r4 r6 r- R s r- c @ s e Zd Zddd�dd�ZdS )�_ManagerFactory�Type[_O]�ClassManager[_O]��class_r2 c C s d S r3 r4 �r5 r@ r4 r4 r6 r7 \ � z_ManagerFactory.__call__Nr8 r4 r4 r4 r6 r< [ s r< c @ sb e Zd ZU dZded< ejZejZ e e � e ��Z ded< ded< dZded < d ed < dZd ed < ded< espdZded< ded< ee jddd�dd� ��Zeje jddd�dd� ��Zdd� Zd�d d d d d!dd"d#�d$d%�Zd"d&�d'd(�Zd)d&�d*d+�Zd,d d-�d.d/�Zed d&�d0d1��Ze j d2d3� �Z e j d4d5� �Z e j d6d7� �Z e j d8d9� �Z!e j"d:d&�d;d<��Z#d�d=d>�Z$d�d?d@�Z%dAd dB�dCdD�Z&dEdFdG�dHdI�Z'dJdK� Z(e j"dLd&�dMdN��Z)dOdP� Z*e j+dQdR� �Z,e j+dSdT� �Z-e j+dUdV� �Z.d�dAdWd d"dX�dYdZ�Z/d[d\� Z0d]d^� Z1d�d_d`�Z2d"d&�dadb�Z3dAdWd"dc�ddde�Z4dAd"dB�dfdg�Z5dAd,d"dh�didj�Z6dAd"dB�dkdl�Z7dAdmdndo�dpdq�Z8dAdrdndsdt�dudv�Z9d�dAd d dw�dxdy�Z:dAdzdB�d{d|�Z;ed}d&�d~d��Z<d�d�d�d��d�d��Z=d�d�d�d"d��d�d��Z>d�d"d��d�d��Z?drd�d�d��d�d��Z@d�d�d��d�d��ZAd�d d��d�d��ZBd�drdAd d d��d�d��ZCd d&�d�d��ZDdAd&�d�d��ZEdS )�� ClassManagerz,Tracks state information at the class level.z dispatcher[ClassManager[_O]]�dispatchr- �expired_attribute_loader�$Optional[Callable[(Ellipsis, None)]]� init_methodN� original_initzOptional[_ManagerFactory]�factoryz$Optional[weakref.ref[_MapperConfig]]�declarative_scanr �registryr= r@ zList[ClassManager[Any]]�_basesz1.4zWThe ClassManager.deferred_scalar_loader attribute is now named expired_attribute_loader)�messagec C s | j S r3 �rE �r5 r4 r4 r6 �deferred_scalar_loader� s z#ClassManager.deferred_scalar_loaderc C s || _ d S r3 rN )r5 �objr4 r4 r6 rP � s c C s� || _ i | _d | _i | _i | _d| _d | _d | _dd� tddd� | j j D ��D �| _ | j D ]}| � |� qZtd| j j ��|| � |jD ]"}t|�}|d k r�| j �|j � q�| �� d|jkr�t�d| � d S ) NFc S s g | ]}|d k r|�qS r3 r4 )�.0�mgrr4 r4 r6 � <listcomp>� s �z)ClassManager.__init__.<locals>.<listcomp>z!List[Optional[ClassManager[Any]]]c S s g | ]}t |t�rt|��qS r4 )� isinstance�type�opt_manager_of_class)rR r r4 r4 r6 rT � 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.)r@ �info�new_init� local_attrs� originals� _finalizedrI rG r � __bases__rL �updaterD Z_eventsZ_new_classmanager_instance�__mro__rW �_update�manage�__dict__r �warn)r5 r@ Zbase_ZbaseclsrS r4 r4 r6 �__init__� sH ��� � � ��zClassManager.__init__F�bool�Optional[Mapper[_O]]�Optional[_RegistryType]�Optional[_MapperConfig]�&Optional[_ExpiredAttributeLoaderProto]r/ )�finalize�mapperrK rJ rE rG r2 c C s� |r || _ |r|�| � |r(t�|�| _|r2|| _|rN| jrHtd| ��|| _| js|| jd k rr| j j t j krr| jn| j j | _ |r�| js�| � � d S )Nz>class is already instrumented, init_method %s can't be applied)rl Z _add_manager�weakref�refrJ rE r] �AssertionErrorrG r@ re �objectrH � _finalize)r5 rk rl rK rJ rE rG r4 r4 r6 � _update_state� s0 ��� �� zClassManager._update_state)r2 c C s* | j r d S d| _ | �� tj�| j� d S �NT)r] �_instrument_init�_instrumentation_factoryrD Zclass_instrumentr@ rO r4 r4 r6 rq � s zClassManager._finalize�intc C s t | �S r3 )�idrO r4 r4 r6 �__hash__� s zClassManager.__hash__r )�otherr2 c C s || kS r3 r4 )r5 ry r4 r4 r6 �__eq__� s zClassManager.__eq__c C s d| j kS )Nrl )rc rO r4 r4 r6 � is_mapped� s zClassManager.is_mappedc C s t | �S r3 )� frozensetrO r4 r4 r6 � _all_key_set� s zClassManager._all_key_setc C s t dd� | �� D ��S )Nc S s g | ]}|j jr|j�qS r4 )�implZ collection�key�rR �attrr4 r4 r6 rT s z6ClassManager._collection_impl_keys.<locals>.<listcomp>�r| �valuesrO r4 r4 r6 �_collection_impl_keys� s �z"ClassManager._collection_impl_keysc C s t dd� | �� D ��S )Nc S s g | ]}|j jr|j �qS r4 )r~ Zaccepts_scalar_loaderr� r4 r4 r6 rT s �z5ClassManager._scalar_loader_impls.<locals>.<listcomp>r� rO r4 r4 r6 �_scalar_loader_impls s ��z!ClassManager._scalar_loader_implsc C s t dd� | �� D ��S )Nc S s g | ] }|j �qS r4 �r~ r� r4 r4 r6 rT s z.ClassManager._loader_impls.<locals>.<listcomp>r� rO r4 r4 r6 � _loader_impls s zClassManager._loader_implsz Mapper[_O]c C s t �| j��d S r3 )r ZUnmappedClassErrorr@ rO r4 r4 r6 rl s zClassManager.mapperc c sv i }| j jdd� D ]\}|jd }|jD ]F}|�||� ||jkrDq(|| j| }t|tj�r(|jr(||fV q(qdS )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)r@ r` rc � setdefaultrU r ZInspectionAttrZ is_attribute)r5 �exclude�found�superclsZinheritsr �valr4 r4 r6 �_all_sqla_attributes s ��z!ClassManager._all_sqla_attributesc C s* | j jD ] }||jkr|j| S q|S )z5return an attribute on the class without tripping it.N)r@ r` rc )r5 r �defaultr� r4 r4 r6 �_get_class_attr_mro8 s z ClassManager._get_class_attr_mro�str)r r2 c C s || ko| | j dk S )z\Return True if the given attribute is fully initialized. i.e. has an impl. Nr� �r5 r r4 r4 r6 �_attr_has_implA s zClassManager._attr_has_implzType[_T]zClassManager[_T])�clsr2 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)rk )�register_class)r5 r� r4 r4 r6 �_subclass_managerI s z ClassManager._subclass_managerc C s$ t | j| | j�| _| �d| j� d S )Nre )�_generate_initr@ rH rZ �install_memberrO r4 r4 r6 rt V s z ClassManager._instrument_initz Type[state.InstanceState[_O]]c C s | j �| | j� tjS r3 )rD Z first_initr@ r r( rO r4 r4 r6 �_state_constructorZ s z ClassManager._state_constructorc C s t | j| j| � dS )z0Mark this instance as the manager for its class.N)�setattrr@ � MANAGER_ATTRrO r4 r4 r6 rb _ s zClassManager.managec C s t S r3 )�_default_manager_getterrO r4 r4 r6 �manager_getterd s zClassManager.manager_getterc C s t S )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_getterrO r4 r4 r6 � state_getterh s zClassManager.state_getterc C s t S r3 )�_default_dict_getterrO r4 r4 r6 � dict_getters s zClassManager.dict_getter�QueryableAttribute[Any])r �inst� propagatedr2 c C sf |r|| j kr*d S n|| j |< | �||� | �� || |< | j�� D ] }| �|�}|�||d� qDd S rs )r[ �install_descriptor�_reset_memoizationsr@ �__subclasses__r� �instrument_attribute)r5 r r� r� r� �managerr4 r4 r6 r� w s z!ClassManager.instrument_attributec c sF | j �� D ]6}t|�}|d k r || k r |V |r |�d�E d H q d S rs )r@ r� rW �subclass_managers)r5 � recursiver� rS r4 r4 r6 r� � s z ClassManager.subclass_managersc C s t j�| j|| | � d S r3 )ru rD Zattribute_instrumentr@ r� r4 r4 r6 �post_configure_attribute� s �z%ClassManager.post_configure_attributec C sl || kr d S |r || j kr2d S n| j |= | �|� | �� | |= | j�� D ] }t|�}|rJ|�|d� qJd S rs )r[ �uninstall_descriptorr� r@ r� rW �uninstrument_attribute)r5 r r� r� r� r4 r4 r6 r� � s z#ClassManager.uninstrument_attributec C sx t | j�D ]}| �|� q d| _d| _d| _| j�� t | �D ]}|| jkr>| � |� q>| j | j j krtt | j | j � dS )z<remove all instrumentation established by this ClassManager.N)�listr\ �uninstall_memberrl rD rZ rY �clearr[ r� r� r@ rc �delattrr� r4 r4 r6 � unregister� s zClassManager.unregister)r r� r2 c C s. || j | jfkr td| ��t| j||� d S �NzW%r: requested attribute name conflicts with instrumentation attribute of the same name.)� STATE_ATTRr� �KeyErrorr� r@ )r5 r r� r4 r4 r6 r� � s ��z ClassManager.install_descriptorc C s t | j|� d S r3 )r� r@ r� r4 r4 r6 r� � s z!ClassManager.uninstall_descriptor)r �implementationr2 c C sH || j | jfkr td| ��| j�|| jj�|t�� t | j||� d S r� ) r� r� r� r\ r� r@ rc �getr, r� )r5 r r� r4 r4 r6 r� � s ��zClassManager.install_memberc C s6 | j �|d �}|tk r&t| j||� n t| j|� d S r3 )r\ �popr, r� r@ r� )r5 r �originalr4 r4 r6 r� � s z ClassManager.uninstall_memberzType[Collection[Any]]r$ )r �collection_classr2 c C s t �|�S r3 )r Zprepare_instrumentation)r5 r r� r4 r4 r6 �instrument_collection_class� s z(ClassManager.instrument_collection_classzInstanceState[_O]z@Tuple[collections.CollectionAdapter, _AdaptedCollectionProtocol])r r rI r2 c C s2 |� }| � |�}t|�s t�t�|||�}||fS r3 )�get_implr ro r ZCollectionAdapter)r5 r r rI � user_datar~ �adapterr4 r4 r6 �initialize_collection� s z"ClassManager.initialize_collection)r �searchr2 c C s |r || kS || j kS d S r3 )r[ )r5 r r� r4 r4 r6 �is_instrumented� s z ClassManager.is_instrumentedr! c C s | | j S r3 r� r� r4 r4 r6 r� � s zClassManager.get_implz Iterable[Any]c C s t | �� �S r3 )�iterr� rO r4 r4 r6 � attributes� s zClassManager.attributeszOptional[InstanceState[_O]]r )r r2 c C s2 | j �| j �}|d kr"| �|| �}| �||� |S r3 )r@ �__new__r� � _state_setter)r5 r �instancer4 r4 r6 � new_instance� s zClassManager.new_instance)r� r r2 c C s$ |d kr| � || �}| �||� d S r3 )r� r� �r5 r� r r4 r4 r6 �setup_instance s zClassManager.setup_instance)r� r2 c C s t || j� d S r3 )r� r� �r5 r� r4 r4 r6 �teardown_instance s z ClassManager.teardown_instance�Dict[str, Any]�_SerializeManager)r � state_dictr2 c C s t ||�S r3 )r� )r5 r r� r4 r4 r6 � _serialize s zClassManager._serializez(Union[Literal[False], InstanceState[_O]]c C sT t || j�rdS | j|jk r4| jr4| �|j��|�S | �|| �}| �||� |S dS )z�Install a default InstanceState if none is present. A private convenience method used by the __init__ decorator. FN) �hasattrr� r@ � __class__r{ r� �_new_state_if_noner� r� r� r4 r4 r6 r� s �� z ClassManager._new_state_if_nonec C s t || j�S r3 )r� r� r� r4 r4 r6 � has_state( s zClassManager.has_state)r r � optimisticr2 c C s | � |�j||d�S )�TODO)r� )r� Z hasparent)r5 r r r� r4 r4 r6 � has_parent+ s zClassManager.has_parentc C s dS )z=All ClassManagers are non-zero regardless of attribute state.Tr4 rO r4 r4 r6 �__bool__1 s zClassManager.__bool__c C s d| j j| jt| �f S )Nz<%s of %r at %x>)r� r9 r@ rw rO r4 r4 r6 �__repr__5 s �zClassManager.__repr__)FNNNNN)N)N)F)F)F)N)N)F)Fr9 r: r; �__doc__�__annotations__r ZDEFAULT_MANAGER_ATTRr� ZDEFAULT_STATE_ATTRr� � staticmethodr Z attrsetterr� rH rJ r rK �property� deprecatedrP �setterre rr rq rx rz r{ r Zmemoized_attributer} r� r� r� Zmemoized_propertyrl r� r� r� r� rt r� rb Z 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� r4 r4 r4 r6 rC _ s� � � / � & ! � � �rC r� c @ s( e Zd ZdZddd�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� )r �dc C s |j | _ |j}|j�||� d S r3 )r@ r� rD �pickle)r5 r r� r� r4 r4 r6 re E s z_SerializeManager.__init__c C sj t | j� |_}|d kr,t�|d| j ��n|jrD|jjsD|j�� |d k rX|� ||� |j � ||� d S )NzxCannot deserialize object of type %r - no mapper() has been configured for this class within the current Python process!) rW r@ r� r ZUnmappedInstanceErrorr{ rl Z configuredZ_check_configurer� rD Zunpickle)r5 r r� r� r� r4 r4 r6 r7 J s �� z_SerializeManager.__call__N)r9 r: r; r� re r7 r4 r4 r4 r6 r� = s r� c @ s\ e Zd ZU dZded< ddd�dd�Zdd d�d d �Zdd d d�dd�Zdd d�dd�ZdS )�InstrumentationFactoryz'Factory for new ClassManager instances.z"dispatcher[InstrumentationFactory]rD r= r> r? c C sb |d k s t �t|�d ks t �| �|�\}}|d kr@t}t|�}n |d k sLt �| �||� ||_|S r3 )ro rW �_locate_extended_factoryrC �_check_conflictsrI )r5 r@ r� rI r4 r4 r6 �create_manager_for_clsd s z-InstrumentationFactory.create_manager_for_clsz<Tuple[Optional[ClassManager[_O]], Optional[_ManagerFactory]]c C s dS )z2Overridden by a subclass to do an extended lookup.)NNr4 rA r4 r4 r6 r� x s z/InstrumentationFactory._locate_extended_factoryz&Callable[[Type[_O]], ClassManager[_O]]r/ )r@ rI r2 c C s dS )z;Overridden by a subclass to test for conflicting factories.Nr4 )r5 r@ rI r4 r4 r6 r� ~ s z'InstrumentationFactory._check_conflictsc C s t |�}|�� | j�|� d S r3 )�manager_of_classr� rD Zclass_uninstrument)r5 r@ r� r4 r4 r6 r� � s z!InstrumentationFactory.unregisterN) r9 r: r; r� r� r� r� r� r� r4 r4 r4 r6 r� _ s r� Tr= rf rg rh ri rj rF r> )r@ rk rl rK rJ rE rG r2 c C s4 t | �}|dkrt�| �}|j||||||d� |S )z_Register class instrumentation. Returns the existing or newly created class manager. N)rl rK rJ rE rG rk )rW ru r� rr )r@ rk rl rK rJ rE rG r� r4 r4 r6 r� � s � r� c C s t �| � dS )z!Unregister class instrumentation.N)ru r� )r@ r4 r4 r6 �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 r4 r4 r6 r� � s �r� c C s� |dkr| j }d}tj|dd�}|| }t|dd�}t|dd�}t� �� }t|d< t||� |d } |j| _|| _ |r||| _ |r�|| _ | 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__r9 re ) re r Zformat_argspec_init�getattr�locals�copyr9 �execr� Z_sa_original_initr� r� ) r@ Z class_managerrH Z func_bodyZ func_varsZ func_textZ func_defaultsZfunc_kw_defaults�envre r4 r4 r6 r� � s$ r� )TNNNNN)Hr� � __future__r �typingr r r r r r r r r r r r r r r rm � r r r r r Z_typingr r� r r �eventr r Z util.typingr r r r! r" r# r$ Z decl_baser% �eventsr& rl r' r( r) r* �symbolr, r- r< r� rC r� r� ru Zinstance_stater� � instance_dictr� r� r� rW Z_default_opt_manager_getterr� r� r� r� r4 r4 r4 r6 �<module> s� � a", �