Edit File: instrumentation.cpython-37.pyc
B ��4]�F � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ e�� Z G d d � d e �Z G d d � d e �Z G d d� de �Ze� Zej ZZej ZZej ZZdd� Zdd� Zdd� Zdd� ZdS )a� 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. � )�base)� collections)�exc)� interfaces)�state� )�utilc @ s� e Zd ZdZejZejZe e � e��Z dZ ejZdZdd� Zdd� Zdd� Zed d � �Zed d � �Zed d� �Zedd� �Ze jdd� �ZdZdd�Z d[dd�Z dd� Z dd� Z dd � Z d d � Z!e jd d � �Z"d!d"� Z#d#d$� Z$e j%d%d&� �Z&e j%d'd(� �Z'e j%d)d*� �Z(d\d,d-�Z)d.d/� Z*d0d1� Z+d]d2d3�Z,d4d5� Z-d6d7� Z.d8d9� Z/d:d;� Z0d<d=� Z1d>d?� Z2d@dA� Z3d^dBdC�Z4dDdE� Z5edFdG� �Z6d_dHdI�Z7d`dJdK�Z8dLdM� Z9dNdO� Z:dPdQ� Z;dRdS� Z<dadTdU�Z=dVdW� Z>e>Z?dXdY� Z@dS )b� ClassManagerz,tracks state information at the class level.Nc C s� || _ i | _d | _i | _i | _dd� dd� | j jD �D �| _x| jD ]}| �|� qDW | jj � || � x,|j D ]"}t |�}|d k rn| j� |j� qnW | �� | �� d|jkr�t�d| � d S )Nc S s g | ]}|d k r|�qS )N� )�.0�mgrr r �Q/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/instrumentation.py� <listcomp>B s z)ClassManager.__init__.<locals>.<listcomp>c S s g | ]}t |t�rt|��qS r )� isinstance�type�manager_of_class)r r r r r r D s �__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.)�class_�info�new_init� local_attrs� originals� __bases__Z_bases�update�dispatchZ_eventsZ_new_classmanager_instance�__mro__r Z_update�manage�_instrument_init�__dict__r �warn)�selfr Zbase_Zbaseclsr r r r �__init__: s* zClassManager.__init__c C s t | �S )N)�id)r r r r �__hash__` s zClassManager.__hash__c C s || kS )Nr )r �otherr r r �__eq__c s zClassManager.__eq__c C s d| j kS )N�mapper)r )r r r r � is_mappedf s zClassManager.is_mappedc C s t | �S )N)� frozenset)r r r r � _all_key_setj s zClassManager._all_key_setc C s t dd� | �� D ��S )Nc S s g | ]}|j jr|j�qS r )�implZ collection�key)r �attrr r r r q s z6ClassManager._collection_impl_keys.<locals>.<listcomp>)r( �values)r r r r �_collection_impl_keysn s z"ClassManager._collection_impl_keysc C s t dd� | �� D ��S )Nc S s g | ]}|j jr|j �qS r )r* Zaccepts_scalar_loader)r r, r r r r x s z5ClassManager._scalar_loader_impls.<locals>.<listcomp>)r( r- )r r r r �_scalar_loader_implst s z!ClassManager._scalar_loader_implsc C s t �| j��d S )N)r ZUnmappedClassErrorr )r r r r r& ~ s zClassManager.mapperc c sn |dkrt � }xZ| jjD ]N}xHt |j��|�D ]4}|�|� |j| }t|tj�r.|j r.||fV q.W qW 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`. N) �setr r r � difference�addr r ZInspectionAttrZ is_attribute)r �exclude�superclsr+ �valr r r �_all_sqla_attributes� s z!ClassManager._all_sqla_attributesc C s. x(| j jD ]}||jkr |j| S q W |S dS )z5return an attribute on the class without tripping it.N)r r r )r r+ �defaultr4 r r r �_get_class_attr_mro� s z ClassManager._get_class_attr_mroc C s || ko| | j dk S )z\Return True if the given attribute is fully initialized. i.e. has an impl. N)r* )r r+ r r r �_attr_has_impl� s zClassManager._attr_has_implc C s t |�}|dkrt�|�}|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. N)r �_instrumentation_factory�create_manager_for_cls)r �cls�managerr r r �_subclass_manager� s z ClassManager._subclass_managerc C s* | j j| _t| j | �| _| �d| j� d S )Nr! )r r! � original_init�_generate_initr �install_member)r r r r r � s z ClassManager._instrument_initc C s | j r| �d� d | _ d S )Nr! )r �uninstall_member)r r r r �_uninstrument_init� s z ClassManager._uninstrument_initc C s | j �| | j� tjS )N)r Z first_initr r Z InstanceState)r r r r �_state_constructor� s z ClassManager._state_constructorc C s t | j| j| � dS )z0Mark this instance as the manager for its class.N)�setattrr � MANAGER_ATTR)r r r r r � s zClassManager.managec C s t | j| j� dS )z)Dissasociate this manager from its class.N)�delattrr rF )r r r r �dispose� s zClassManager.disposec C s t S )N)�_default_manager_getter)r r r r �manager_getter� 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_getter)r r r r � state_getter� s zClassManager.state_getterc C s t S )N)�_default_dict_getter)r r r r � dict_getter� s zClassManager.dict_getterFc C sl |r|| j kr*d S n|| j |< | �||� t�| � || |< x*| j�� D ] }| �|�}|�||d� qHW d S )NT)r �install_descriptor�_memoized_key_collection�expire_instancer �__subclasses__r> �instrument_attribute)r r+ �inst� propagatedr< r= r r r rS � s z!ClassManager.instrument_attributec c sT xN| j �� D ]@}t|�}|d k r || k r |V |r x|�d�D ] }|V q>W q W d S )NT)r rR r �subclass_managers)r � recursiver< r �mr r r rV � s z ClassManager.subclass_managersc C s t j�| j|| | � d S )N)r: r Zattribute_instrumentr )r r+ r r r �post_configure_attribute s z%ClassManager.post_configure_attributec C sr || kr d S |r || j kr2d S n| j |= | �|� t�| � | |= x*| j�� D ] }t|�}|rN|�|d� qNW d S )NT)r �uninstall_descriptorrP rQ r rR r �uninstrument_attribute)r r+ rU r< r= r r r r[ s z#ClassManager.uninstrument_attributec C sH | � � d | _| _| j�� x$t| �D ]}|| jkr(| �|� q(W dS )z<remove all instrumentation established by this ClassManager.N)rC r& r r �clear�listr r[ )r r+ r r r � unregister s zClassManager.unregisterc C s. || j | jfkr td| ��t| j||� d S )NzW%r: requested attribute name conflicts with instrumentation attribute of the same name.)� STATE_ATTRrF �KeyErrorrE r )r r+ rT r r r rO " s z ClassManager.install_descriptorc C s t | j|� d S )N)rG r )r r+ r r r rZ * s z!ClassManager.uninstall_descriptorc C sF || j | jfkr td| ��| j�|t| j|d �� t| j||� d S )NzW%r: requested attribute name conflicts with instrumentation attribute of the same name.)r_ rF r` r � setdefault�getattrr rE )r r+ �implementationr r r rA - s zClassManager.install_memberc C s( | j �|d �}|d k r$t| j||� d S )N)r �poprE r )r r+ �originalr r r rB 6 s z ClassManager.uninstall_memberc C s t �|�S )N)r Zprepare_instrumentation)r r+ Zcollection_classr r r �instrument_collection_class; s z(ClassManager.instrument_collection_classc C s" |� }t �| �|�||�}||fS )N)r ZCollectionAdapter�get_impl)r r+ r �factoryZ user_dataZadapterr r r �initialize_collection> s z"ClassManager.initialize_collectionc C s |r || kS || j kS d S )N)r )r r+ �searchr r r �is_instrumentedE s z ClassManager.is_instrumentedc C s | | j S )N)r* )r r+ r r r rg K s zClassManager.get_implc C s t | �� �S )N)�iterr- )r r r r � attributesN s zClassManager.attributesc C s2 | j �| j �}|d kr"| �|| �}| �||� |S )N)r �__new__rD � _state_setter)r r �instancer r r � new_instanceT s zClassManager.new_instancec C s$ |d kr| � || �}| �||� d S )N)rD ro )r rp r r r r �setup_instance[ s zClassManager.setup_instancec C s t || j� d S )N)rG r_ )r rp r r r �teardown_instance` s z ClassManager.teardown_instancec C s t ||�S )N)�_SerializeManager)r r � state_dictr r r � _serializec s zClassManager._serializec 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_nonerD ro )r rp r r r r ry f s z ClassManager._new_state_if_nonec C s t || j�S )N)rw r_ )r rp r r r � has_state| s zClassManager.has_statec C s | � |�j||d�S )ZTODO)� optimistic)rg Z hasparent)r r r+ r{ r r r � has_parent s zClassManager.has_parentc C s dS )z=All ClassManagers are non-zero regardless of attribute state.Tr )r r r r �__bool__� s zClassManager.__bool__c C s d| j j| jt| �f S )Nz<%s of %r at %x>)rx �__name__r r" )r r r r �__repr__� s zClassManager.__repr__)N)N)F)F)F)N)N)F)Ar~ � __module__� __qualname__�__doc__r ZDEFAULT_MANAGER_ATTRrF ZDEFAULT_STATE_ATTRr_ � staticmethodr Z attrsetterro Zdeferred_scalar_loader�objectr! r? rh r# r% �propertyr' rP r) r. r/ Zmemoized_propertyr& r6 r8 r9 r> r rC rD r rH Z hybridmethodrJ rL rN rS rV rY r[ r^ rO rZ rA rB rf ri rk rg rm rq rr rs rv ry rz r| r} � __nonzero__r r r r r r , sf & r c @ s e Zd ZdZdd� Zdd� ZdS )rt z�Provide serialization of a :class:`.ClassManager`. The :class:`.InstanceState` uses ``__init__()`` on serialize and ``__call__()`` on deserialize. c C s |j | _ |j}|j�||� d S )N)r r= r �pickle)r r �dr= r r r r! � 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!) r r r= r ZUnmappedInstanceErrorr' r& Z configuredZ_configure_allrr r Zunpickle)r r rT ru r= r r r �__call__� s z_SerializeManager.__call__N)r~ r� r� r� r! r� r r r r rt � s rt c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �InstrumentationFactoryz'Factory for new ClassManager instances.c C s` |d k s t �t|�d ks t �| �|�\}}|d kr>t}||�}| �||� ||_| j�|� |S )N)�AssertionErrorr �_locate_extended_factoryr �_check_conflictsrh r Zclass_instrument)r r r= rh r r r r; � s z-InstrumentationFactory.create_manager_for_clsc C s dS )z2Overridden by a subclass to do an extended lookup.)NNr )r r r r r r� � s z/InstrumentationFactory._locate_extended_factoryc C s dS )z;Overridden by a subclass to test for conflicting factories.Nr )r r rh r r r r� � s z'InstrumentationFactory._check_conflictsc C s@ t |�}|�� |�� | j�|� tj|jkr<t|tj� d S )N) r r^ rH r Zclass_uninstrumentr rF r rG )r r r= r r r r^ � s z!InstrumentationFactory.unregisterN)r~ r� r� r� r; r� r� r^ r r r r r� � s r� c C s t | �}|dkrt�| �}|S )z_Register class instrumentation. Returns the existing or newly created class manager. N)r r: r; )r r= r r r �register_class� s r� c C s t �| � dS )z!Unregister class instrumentation.N)r: r^ )r r r r �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)rj )r rx rk )rp r+ r r r rk � s rk c C s� | j }|st�d}tj|dd�}|| }tjrHt|d|�}t|dd�}nt|dd�}t|dd�}t� �� } t|| � | d } |j | _ || _ |r�|| _ tjs�|r�|| _ | S ) z>Build an __init__ decorator that triggers ClassManager events.z�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)ZgroupedZim_func� func_defaultsN� __defaults__�__kwdefaults__r! ) r! r� r Zformat_argspec_initZpy2krb �locals�copy�execr� Z_sa_original_initr� r� ) r Z class_managerZoriginal__init__Z func_bodyZ func_varsZ func_text�funcr� Zfunc_kw_defaults�envr! r r r r@ s( r@ N)r� � r r r r r r Z!group_expirable_memoized_propertyrP �dictr r� rt r� r: Zinstance_staterK � instance_dictrM r rI r� r� rk r@ r r r r �<module> s( g")