Edit File: dynamic.cpython-37.pyc
B ��4]J9 � @ s d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd l m Z d dlmZ d d lmZ d d lm Z ejejjd d�G dd� de j���ZG dd� dej�ZG dd� de�ZG dd� dee �Zdd� ZG dd� de�ZdS ) z�Dynamic collection API. Dynamic collections act like Query() objects for read operations and support basic add/delete mutation. � )� attributes)�exc)� interfaces)� object_mapper)�object_session)� properties)� strategies)�util)�Query� )�log�dynamic)Zlazyc @ s e Zd Zdd� ZdS )� DynaLoaderc C sl d| _ | js t�d| j ��n$| jjtjtjfkrBt � d| j � t j | j|dt | jj| jj| jjd� d S )NTztOn relationship %s, 'dynamic' loaders cannot be used with many-to-one/one-to-one relationships and/or uselist=False.z�On relationship %s, 'dynamic' loaders cannot be used with many-to-one/one-to-one relationships and/or uselist=False. This warning will be an exception in a future release.)Z useobjectZ impl_class� target_mapper�order_by� query_class)Zis_class_levelZuselistr ZInvalidRequestErrorZparent_property� directionr Z ONETOMANYZ MANYTOMANYr �warnr Z_register_attribute�DynamicAttributeImpl�mapperr r )�selfr � r �I/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/dynamic.py�init_class_attribute s& z DynaLoader.init_class_attributeN)�__name__� __module__� __qualname__r r r r r r s r c s e Zd ZdZdZdZdZdZd&� fdd� Ze j fdd�Z de j fdd �Z ejd d � �Zejd d � �Zd'dd�Zd(dd�Zdd� Zde j dddfdd�Zdd� Zdd� Ze j fdd�Ze j fd d �Ze j fd d �Ze j fd d!�Ze j fd"d#�Z e j fd$d%�Z � Z S ))r TFNc sT t t| �j||||f|� || _|| _|s2t| _n t|�� krF|| _n t |�| _d S )N) �superr �__init__r r � AppenderQueryr � AppenderMixin�mro�mixin_user_query) r �class_�keyZ typecallable�dispatchr r r �kw)� __class__r r r E s z DynamicAttributeImpl.__init__c C s* |t j@ s| �|t j�jS | �| |�S d S )N)r �SQL_OK�_get_collection_history�PASSIVE_NO_INITIALIZE� added_itemsr )r �state�dict_�passiver r r �get\ s zDynamicAttributeImpl.getc C s. |t j@ s| �||�jS | �||�}|jS d S )N)r r( r) r+ �added_plus_unchanged)r r, r- Z user_datar. �historyr r r �get_collectiond s z#DynamicAttributeImpl.get_collectionc C s t �| t j�S )N)r �EventZ OP_APPEND)r r r r � _append_tokenq s z"DynamicAttributeImpl._append_tokenc C s t �| t j�S )N)r r3 Z OP_REMOVE)r r r r � _remove_tokenu s z"DynamicAttributeImpl._remove_tokenc C sh |d kr| � ||�}|�|� x"| jjD ]}||||p:| j�}q(W | jrd|d k rd| �t�|�|d� d S )NT) �_modified_event� add_addedr% �appendr4 � trackparent� sethasparentr �instance_state)r r, r- �value� initiator�collection_history�fnr r r �fire_append_eventy s z&DynamicAttributeImpl.fire_append_eventc C sh |d kr| � ||�}|�|� | jr@|d k r@| �t�|�|d� x"| jjD ]}||||p\| j� qJW d S )NF) r6 � add_removedr9 r: r r; r% �remover5 )r r, r- r< r= r> r? r r r �fire_remove_event� s z&DynamicAttributeImpl.fire_remove_eventc C sD | j |jkr t| |�|j| j < |�|| tj� d|| j < |j| j S )NT)r$ �committed_state�CollectionHistoryr6 r Z NEVER_SET)r r, r- r r r r6 � s z$DynamicAttributeImpl._modified_eventc C s� |r|j | j krd S |r$|d kr$d S |} t| �} |jrHt�| �||��} | �||�} |jsb| j} n | �| j�} tj} | � | �}| | �� |�}| � |�}x(| D ] }||kr�| j |||d | d� q�W x |D ]}| j |||d | d� q�W d S )N)r> ) Z parent_token�list� has_identityr Z IdentitySetr/ r6 r+ �union� intersection� differencer@ rC )r r, r- r< r= r. Z check_old�popZ_adapt�iterable� new_valuesZold_collectionr> Zidset� constantsZ additionsZremovals�memberr r r �set� s@ zDynamicAttributeImpl.setc O s t � �d S )N)�NotImplementedError)r �args�kwargsr r r �delete� s zDynamicAttributeImpl.deletec C s t d��d S )Nz7Dynamic attributes don't support collection population.)rQ )r r, r- r< r r r �set_committed_value� s z(DynamicAttributeImpl.set_committed_valuec C s | � ||�}|�� S )N)r) � as_history)r r, r- r. �cr r r � get_history� s z DynamicAttributeImpl.get_historyc C s | � ||�}dd� |jD �S )Nc S s g | ]}t �|�|f�qS r )r r; )�.0�xr r r � <listcomp>� s z8DynamicAttributeImpl.get_all_pending.<locals>.<listcomp>)r) � all_items)r r, r- r. rW r r r �get_all_pending� s z$DynamicAttributeImpl.get_all_pendingc C sJ | j |jkr|j| j }n t| |�}|jrB|tj@ rBt| ||d�S |S d S )N)�apply_to)r$ rD rE rG r ZINIT_OK)r r, r. rW r r r r) � s z,DynamicAttributeImpl._get_collection_historyc C s || k r| � ||||� d S )N)r@ )r r, r- r< r= r. r r r r8 � s zDynamicAttributeImpl.appendc C s || k r| � ||||� d S )N)rC )r r, r- r< r= r. r r r rB � s zDynamicAttributeImpl.removec C s | j |||||d� d S )N)r. )rB )r r, r- r< r= r. r r r rK s zDynamicAttributeImpl.pop)N)N)N) r r r Z uses_objectsZ default_accepts_scalar_loaderZsupports_populationZ collectionr r r Z PASSIVE_OFFr/ r* r2 r Zmemoized_propertyr4 r5 r@ rC r6 rP rT rU rX r] r) r8 rB rK � __classcell__r r )r' r r > s6 - r c sp e Zd ZdZ� fdd�Zdd� Zeedd� �Zdd � Zd d � Zd d � Z ddd�Z dd� Z dd� Z dd� Z � ZS )r Nc s~ t t| ��|jd � |�� | _}|| _t|�}|j| jj }|j d k rX|j j |j f| _ |j|dd�| _| jjrz| jj| _d S )NF)Zalias_secondary)r r r r �obj�instance�attrr Z_propsr$ Z secondaryr Z selectable� _from_objZ _with_parent� _criterionr � _order_by)r rb r, ra r Zprop)r' r r r s zAppenderMixin.__init__c C sH t | j�}|d k r0| jr0|jr0| j|kr0|�� t�| j�s@d S |S d S )N)r ra � autoflush�flush�orm_utilrG )r �sessr r r �session s zAppenderMixin.sessionc C s d S )Nr )�srZ r r r �<lambda>- � zAppenderMixin.<lambda>c C s> | j }|d kr,t| j�t�| j�tj�j�S t| � |��S d S )N) rj �iterrb r) r r; ra r* r+ �_clone)r ri r r r �__iter__/ s zAppenderMixin.__iter__c C s@ | j }|d kr,| j�t�| j�tj��|�S | �|�� |�S d S )N) rj rb r) r r; ra r* �indexedro � __getitem__)r �indexri r r r rr ; s zAppenderMixin.__getitem__c C s> | j }|d kr,t| j�t�| j�tj�j�S | � |�� � S d S )N) rj �lenrb r) r r; ra r* r+ ro �count)r ri r r r ru E s zAppenderMixin.countc C s~ | j }|d kr:t|�}|d kr:t�dt�|�| jjf ��| jrT| j| jj |d�}n|� | jj �}| j |_ | j |_ | j |_ |S )Nz�Parent instance %s is not bound to a Session, and no contextual session is established; lazy load operation of attribute '%s' cannot proceed)rj )ra r �orm_excZDetachedInstanceErrorrh Z instance_strrb r$ r r �queryrd rc re )r ri ra rw r r r ro Q s zAppenderMixin._clonec C s4 x.|D ]&}| j �t�| j�t�| j�|d � qW d S )N)rb r8 r r; ra � instance_dict)r �iterator�itemr r r �extendk s zAppenderMixin.extendc C s& | j �t�| j�t�| j�|d � d S )N)rb r8 r r; ra rx )r rz r r r r8 t s zAppenderMixin.appendc C s& | j �t�| j�t�| j�|d � d S )N)rb rB r r; ra rx )r rz r r r rB | s zAppenderMixin.remove)N)r r r r r rj �propertyrp rr ru ro r{ r8 rB r_ r r )r' r r s r c @ s e Zd ZdZdS )r zBA dynamic query that supports basic collection storage operations.N)r r r �__doc__r r r r r � s r c C s d| j }t|t| fd| i�S )zAReturn a new class with AppenderQuery functionality layered over.ZAppenderr )r �typer )�cls�namer r r r" � s r" c @ sR e Zd ZdZddd�Zedd� �Zedd� �Zd d � Zd d � Z d d� Z dd� Z dS )rE zDOverrides AttributeHistory to receive append/remove events directly.Nc C s` |r8t ||��d�}t�|�| _|j| _|j| _d| _n$t�� | _t�� | _t�� | _d| _d S )NFT)r rf r ZOrderedIdentitySet�unchanged_itemsr+ � deleted_items�_reconcile_collection)r rb r, r^ Zcollr r r r � s zCollectionHistory.__init__c C s t | j�| j��S )N)rF r+ rH r� )r r r r r0 � s z&CollectionHistory.added_plus_unchangedc C s t | j�| j��| j��S )N)rF r+ rH r� r� )r r r r r\ � s zCollectionHistory.all_itemsc C s` | j r0| j�| j�}| j�| j�}| j�|�}n| j| j| j }}}t�t|�t|�t|��S )N) r� r+ rJ r� r� rI r ZHistoryrF )r ZaddedZdeletedZ unchangedr r r rV � s z CollectionHistory.as_historyc C s t | j�| S )N)rF r+ )r rs r r r rq � s zCollectionHistory.indexedc C s | j �|� d S )N)r+ �add)r r< r r r r7 � s zCollectionHistory.add_addedc C s( || j kr| j �|� n | j�|� d S )N)r+ rB r� r� )r r< r r r rA � s z CollectionHistory.add_removed)N) r r r r} r r| r0 r\ rV rq r7 rA r r r r rE � s rE N)r} � r r rv r r r r r r rh rw r r Z class_loggerZRelationshipPropertyZ strategy_forZAbstractRelationshipLoaderr Z AttributeImplr �objectr r r" rE r r r r �<module> s* I