File: //opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/__pycache__/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 Zclass_�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 r2 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$ �appendr3 � 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) r5 � add_removedr8 r9 r r: r$ �remover4 )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�CollectionHistoryr5 r Z NEVER_SET)r r+ r, r r r r5 � 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. r5 r* �union� intersection�
differencer? rB )r r+ r, r; r< r- Z check_old�popZ_adapt�iterableZ
new_valuesZold_collectionr= ZidsetZ 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.)rN )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- rT 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# rC rD rF r ZINIT_OK)r r+ r- rT 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 r7 � s zDynamicAttributeImpl.appendc C s || k r| � ||||� d S )N)rB )r r+ r, r; r< r- r r r rA � s zDynamicAttributeImpl.removec C s | j |||||d� d S )N)r- )rA )r r+ r, r; r< r- r r r rJ s zDynamicAttributeImpl.pop)N)N)N) r r r Z uses_objectsZ default_accepts_scalar_loaderZsupports_populationZ
collectionr
r r Z PASSIVE_OFFr. r) r1 r Zmemoized_propertyr3 r4 r? rB r5 rM rQ rR rU rZ r( r7 rA rJ �
__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 r_ r+ r^ 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 r^ � autoflush�flush�orm_utilrF )r �sessr r r �session s
zAppenderMixin.sessionc C s d S )Nr )�srW r r r �<lambda>- � zAppenderMixin.<lambda>c C s> | j }|d kr,t| j�t�| j�tj�j�S t| � |��S d S )N)
rg �iterr_ r( r r: r^ r) r* �_clone)r rf r r r �__iter__/ s
zAppenderMixin.__iter__c C s@ | j }|d kr,| j�t�| j�tj��|�S | �|�� |�S d S )N)
rg r_ r( r r: r^ r) �indexedrl � __getitem__)r �indexrf r r r ro ; s
zAppenderMixin.__getitem__c C s> | j }|d kr,t| j�t�| j�tj�j�S | � |��
� S d S )N) rg �lenr_ r( r r: r^ r) r* rl �count)r rf r r r rr 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)rg )r^ r �orm_excZDetachedInstanceErrorre Z instance_strr_ r# r r �queryra r` rb )r rf r^ rt r r r rl Q s zAppenderMixin._clonec C s4 x.|D ]&}| j �t�| j�t�| j�|d � qW d S )N)r_ r7 r r: r^ �
instance_dict)r �iterator�itemr r r �extendk s
zAppenderMixin.extendc C s& | j �t�| j�t�| j�|d � d S )N)r_ r7 r r: r^ ru )r rw r r r r7 t s
zAppenderMixin.appendc C s& | j �t�| j�t�| j�|d � d S )N)r_ rA r r: r^ ru )r rw r r r rA | s
zAppenderMixin.remove)N)r r r r r rg �propertyrm ro rr rl rx r7 rA 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 )rD 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 rc r ZOrderedIdentitySet�unchanged_itemsr* �
deleted_items�_reconcile_collection)r r_ r+ r[ Zcollr r r r � s
zCollectionHistory.__init__c C s t | j�| j��S )N)rE r* rG r~ )r r r r r/ � s z&CollectionHistory.added_plus_unchangedc C s t | j�| j��| j��S )N)rE r* rG r~ r )r r r r rY � 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* rI r~ r rH r ZHistoryrE )r ZaddedZdeletedZ unchangedr r r rS � s z CollectionHistory.as_historyc C s t | j�| S )N)rE r* )r rp r r r rn � s zCollectionHistory.indexedc C s | j �|� d S )N)r* �add)r r; r r r r6 � s zCollectionHistory.add_addedc C s( || j kr| j �|� n | j�|� d S )N)r* rA r r� )r r; r r r r@ � s
z CollectionHistory.add_removed)N) r r r rz r ry r/ rY rS rn r6 r@ r r r r rD � s
rD N)rz � r r rs r r r r r r re rt r
r Z class_loggerZRelationshipPropertyZ strategy_forZAbstractRelationshipLoaderr Z
AttributeImplr �objectr r r" rD r r r r �<module>
s* I