Edit File: dependency.cpython-37.pyc
B ��4]ܵ � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z d dlmZ d d lm Z d dlmZ G d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Ze ee ee eiZdS )z Relationship dependencies. � )� attributes)�exc)�sync)� unitofwork)�util)� MANYTOMANY)� MANYTOONE)� ONETOMANY� )�sqlc @ s� e Zd Zdd� Zedd� �Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd%d d �Zd d!� Zd"d#� Zd$S )&�DependencyProcessorc C s� || _ |j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j | _ | jr^t j | _ nt j | _ | jrvt j | _nt j | _|j| _| j js�t�d| j ��d S )Nz|Can't build a DependencyProcessor for relationship %s. No target attributes to populate between parent and child are present)�prop�cascade�mapper�parent� secondary� direction� post_update�passive_deletes�passive_updates�enable_typechecksr �PASSIVE_NO_INITIALIZE�_passive_delete_flag� PASSIVE_OFF�_passive_update_flag�key�synchronize_pairs�sa_exc� ArgumentError)�selfr � r �L/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/dependency.py�__init__ s* z DependencyProcessor.__init__c C s t |j |�S )N)�_direction_to_processorr )�clsr r r r! �from_relationship6 s z%DependencyProcessor.from_relationshipc C s | j j�| j��|�S )z�return True if the given object instance has a parent, according to the ``InstrumentedAttribute`` handled by this ``DependencyProcessor``. )r Z class_manager�get_implr � hasparent)r �stater r r! r' : s z DependencyProcessor.hasparentc C s |� | d� dS )z�establish actions and dependencies related to a flush. These actions will operate on all relevant states in the aggregate. TN)�register_preprocessor)r �uowr r r! �per_property_preprocessorsB s z.DependencyProcessor.per_property_preprocessorsc C sz t �|| dd�}t �|| dd�}t �|| jj�}t �|| jj�}t �|| jj�}t �|| jj�}| �|||||||� d S )NFT)r � ProcessAll� SaveUpdateAllr �primary_base_mapperr � DeleteAll�per_property_dependencies)r r* � after_save� before_delete� parent_saves� child_saves�parent_deletes� child_deletesr r r! �per_property_flush_actionsK s$ z.DependencyProcessor.per_property_flush_actionsc C s | j j}t�||�}t�||�}|r<t�|| dd�}d|_nt�|| dd�}d|_||jkr�||jksjt�|df|dfg} d} nd} |s�t�|| j j �} d } }| |jkr�d} n&t�|| j j �} d } }| |jkr�d} �x&|D �] }|j | j j �||j|�r| jntj�}|�sq�|�r>t�|| d|�}| �r`t�||�} n"t�|| d|�}| �r`t�||�} | �r�g } xd|D ]\\}}||jk�r�d}n6|j| \}}|�r�t�||�df}nt�||�df}| �|� �qpW x*| D ]"\}}| �|| | |||||� �q�W q�W dS )z�establish actions and dependencies related to a flush. These actions will operate on all relevant states individually. This occurs only if there are cycles in the 'aggregated' version of events. TFN)NN)r r. r r- r/ r, �disabledZcycles�AssertionErrorr � base_mapper�managerr �implZget_all_pending�dictr r r Z ProcessStateZ DeleteStateZSaveUpdateState�states�append�per_state_dependencies)r r* r> �isdeleteZchild_base_mapperr4 r6 r2 r1 Z child_actionsZchild_in_cyclesr3 r5 Zparent_in_cyclesr( Zsum_Z child_state�child� child_action�deletedZlistonly� childisdeleter r r! �per_state_flush_actionsg sx z+DependencyProcessor.per_state_flush_actionsc C s dS )NFr )r � uowcommitr> r r r! �presort_deletes� s z#DependencyProcessor.presort_deletesc C s dS )NFr )r rG r> r r r! � presort_saves� s z!DependencyProcessor.presort_savesc C s d S )Nr )r rG r> r r r! �process_deletes� s z#DependencyProcessor.process_deletesc C s d S )Nr )r rG r> r r r! � process_saves� s z!DependencyProcessor.process_savesc C sv |r | j rtj}n| jtkr$tj}ntj}xF|D ]$}|�|| j|�}|r0|� � s0dS q0W |op| j j op| j |j kS d S )NT)r r r r r ZPASSIVE_NO_FETCH_RELATEDr �get_attribute_historyr �emptyr Z_is_self_referentialr Zmappers)r rG r> rA �passive�s�historyr r r! �prop_has_changes� s z$DependencyProcessor.prop_has_changesc C s� | j jr$|d kr$t�d| j f ��nn|d k r�| jj|| j d�s�| jj|dd�rtt�d|j| j | jj| jd� ��n t�d|j| j | jjd� ��d S )Nz-Can't flush None value found in collection %s)Zallow_subtypesTaE Attempting to flush an item of type %(x)s as a member of collection "%(y)s". Expected an object of type %(z)s or a polymorphic subclass of this type. If %(x)s is a subclass of %(z)s, configure mapper "%(zm)s" to load this subtype polymorphically, or set enable_typechecks=False to allow any subtype to be accepted for flush. )�x�y�zZzmz�Attempting to flush an item of type %(x)s as a member of collection "%(y)s". Expected an object of type %(z)s or a polymorphic subclass of this type.)rR rS rT )r �uselistr Z FlushErrorr Z_canloadr �class_)r r( r r r! �_verify_canload� s$ z#DependencyProcessor._verify_canloadc C s t � �d S )N)�NotImplementedError)r r( rB �associationrow� clearkeysrG r r r! � _synchronize" s z DependencyProcessor._synchronizec C s> | j js d S tt| jgdd� | j jD � ��}|�d|ft�S )Nc S s g | ] }|j �qS r )r )�.0�pr r r! � <listcomp>* s zCDependencyProcessor._get_reversed_processed_set.<locals>.<listcomp>Z reverse_key)r �_reverse_property�tuple�sortedr �memo�set)r r* Z process_keyr r r! �_get_reversed_processed_set% s z/DependencyProcessor._get_reversed_processed_setFc C s: x4|D ],}|r|d k r|� |dd� | jjD �� P qW d S )Nc S s g | ] \}}|�qS r r )r\ �l�rr r r! r^ 2 s z4DependencyProcessor._post_update.<locals>.<listcomp>)Zregister_post_updater r )r r( rG �related� is_m2o_deleterR r r r! � _post_update. s z DependencyProcessor._post_updatec C s t � �d S )N)rX )r rG r( r r r! � _pks_changed6 s z DependencyProcessor._pks_changedc C s d| j j| jf S )Nz%s(%s))� __class__�__name__r )r r r r! �__repr__9 s z DependencyProcessor.__repr__N)F)rl � __module__� __qualname__r" � classmethodr% r' r+ r7 rF rH rI rJ rK rQ rW r[ rd ri rj rm r r r r! r s" p) r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� OneToManyDPc C s� | j r\t�|| jjd�}t�|| jjd�} |j�||f||f||f|| f| |f| |fg� n6|j�||f||f||f||f||f||f||fg� d S )NFT)r r � PostUpdateAllr r. � dependencies�update) r r* r3 r4 r5 r6 r1 r2 �child_post_updates�child_pre_updatesr r r! r0 > s( z%OneToManyDP.per_property_dependenciesc C s� | j r�t�|| jjd�} t�|| jjd�} |sl|rL|j�||f|| fg� q�|j�||f||f|| fg� q�|r�|j�|| f| |fg� q�|j�|| f| |fg� n<|s�|j�||f||f||fg� n|j�||f||fg� d S )NFT)r r rr r r. rs rt ) r r* � save_parent� delete_parentrC r1 r2 rA rE ru rv r r r! r@ g s8 z"OneToManyDP.per_state_dependenciesc C s� | j j o| jdk }x�|D ]�}|�|| j| j�}|r xF|jD ]<}|d k r>| �|�dkr>| j jrp|j |dd� q>|� |� q>W |r x(|j D ] }|d k r�|j |d| j d� q�W q W d S )N�allFT)rA �delete)� operationr ) r rz r rL r r rD r' � delete_orphan�register_object� unchangedr )r rG r> �should_null_fksr( rP rB r r r! rH � s zOneToManyDP.presort_deletesc C sn |� d| ft�}| jj o$| jdk }�x@|D �]6}| �||�}|rJ| jrRtj}ntj }|� || j |�}|�r*x*|j D ] } | d k rv|j | dd| jd� qvW |�|j � x�|jD ]x} | jjs�|r�|j | dd| jd� q�| �| �dkr�|j | dd| jd� x.| j�d| �D ] \} } } } |j | dd � �qW q�W |r.|r.x2|jD ](} | d k �r:|j | d| jd | jd � �q:W q.W d S ) N�children_addedry T�add)Z cancel_deleter{ r Frz )rA r{ r )rA z pk change)r{ r )rb rc r r| r rj r r r r rL r �addedr} r rt rD r' r �cascade_iteratorr~ )r rG r> r� r r( � pks_changedrN rP rB �c�m�st_�dct_r r r! rI � sZ zOneToManyDP.presort_savesc C s� | j s| jdks�|�d| ft�}x�|D ]�}|�|| j| j�}|r&xR|jD ]H}|d k rH| �|�dkrH| � ||d d|d� | j rH|rH| � |||g� qHW | j s�| j j s&xNt|j ��|�D ]:}|d k r�| � ||d d|d� | j r�|r�| � |||g� q�W q&W d S )Nry r� FT)r r rb rc rL r r rD r' r[ ri r rz r~ � difference)r rG r> r� r( rP rB r r r! rJ s, zOneToManyDP.process_deletesc C s� | j j o| jdk }x�|D ]�}|�|| jtj�}|r x@|jD ]6}| �||d d|d� |d k r>| j r>| � |||g� q>W x8|j D ].}|r�| j js�| � |�s�| �||d d|d� q�W | � ||�r x"|jD ]}| �||d d|d� q�W q W d S )Nry FT)r r| r rL r r r r� r[ r ri rD r' rj r~ )r rG r> r r( rP rB r r r! rK / s* zOneToManyDP.process_savesc C sp |}|}| � |� |d ks*| js.|�|�r.d S |rHt�|| j| jj� n$t�|| j || j| jj|| j oh|� d S )N) rW r � is_deletedr �clearr r r �populater r ) r r( rB rY rZ rG r� �source�destr r r! r[ Q s zOneToManyDP._synchronizec C s t �||| j| jj�S )N)r �source_modifiedr r r )r rG r( r r r! rj h s zOneToManyDP._pks_changedN) rl rn ro r0 r@ rH rI rJ rK r[ rj r r r r! rq = s )C>,"rq c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z ddd�Z dS )� ManyToOneDPc C s2 t �| |� x | jjD ]}|j�t|�� qW d S )N)r r" r Zself_and_descendantsZ_dependency_processorsr? �DetectKeySwitch)r r r r r r! r" o s zManyToOneDP.__init__c C s� | j rbt�|| jjd�}t�|| jjd�} |j�||f||f||f|| f|| f| |f| |fg� n$|j�||f||f||f||fg� d S )NFT)r r rr r r. rs rt ) r r* r3 r4 r5 r6 r1 r2 �parent_post_updates�parent_pre_updatesr r r! r0 t s$ z%ManyToOneDP.per_property_dependenciesc C s� | j r�|sZt�|| jjd�} |r:|j�|| f| |fg� q�|j�||f||f|| fg� q�t�|| jjd�} |j�|| f| |f| |fg� nL|s�|s�|j�||f||fg� q�|j�||fg� n|r�|j�||fg� d S )NFT)r r rr r r. rs rt ) r r* rw rx rC r1 r2 rA rE r� r� r r r! r@ � s2 z"ManyToOneDP.per_state_dependenciesc C s� | j js| j jr�x�|D ]�}|�|| j| j�}|r| j jrB|�� }n|�� }x\|D ]T}|d kr^qP|j|dd| j d� | j � d|�}x"|D ]\}} } } |j| dd� q�W qPW qW d S )NTrz )rA r{ r )rA ) r rz r| rL r r �sumZ non_deletedr} r r r� ) r rG r> r( rP ZtodeleterB �tr� r� r� r� r r r! rH � s& zManyToOneDP.presort_deletesc C s� x�|D ]�}|j |d| jd� | jjr|�|| j| j�}|rxb|jD ]X}| �|�dkrB|j |dd| jd� | j � d|�}x"|D ]\}}} } |j | dd� q|W qBW qW d S )Nr� )r{ r FTrz )rA r{ r )rA ) r} r r r| rL r r rD r' r r� ) r rG r> r( rP rB r� r� r� r� r� r r r! rI � s zManyToOneDP.presort_savesc C sr | j rn| jjsn| jdksnxT|D ]L}| �|d d d|� |r | j r |�|| j| j�}|r | j|||� � dd� q W d S )Nry T)rh ) r r r| r r[ rL r r ri r� )r rG r> r( rP r r r! rJ s zManyToOneDP.process_deletesc C s� x�|D ]x}|� || jtj�}|r|jrLx>|jD ]}| �||d d|d� q.W n|jrf| �|d d d|d� | jr| �|||� � � qW d S )NFr� Trz ) rL r r r r� r[ rD r ri r� )r rG r> r( rP rB r r r! rK s zManyToOneDP.process_savesNc C s� |d ks| j s |�|�r d S |d k rX|d k rX|j�|�sXt�dt�|�|| jf � d S |sd|d krzt � || j | jj � n(| � |� t �|| j|| j | jj |d� d S )NzGObject of type %s not in session, %s operation along '%s' won't proceedF)r r� �session�_contains_stater �warn� mapperutil�state_class_strr r r� r r rW r� r )r r( rB rY rZ rG r{ r r r! r[ * s* zManyToOneDP._synchronize)N) rl rn ro r" r0 r@ rH rI rJ rK r[ r r r r! r� n s (:r� c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )r� a� For many-to-one relationships with no one-to-many backref, searches for parents through the unit of work when a primary key has changed and updates them. Theoretically, this approach could be expanded to support transparent deletion of objects referenced via many-to-one as well, although the current attribute system doesn't do enough bookkeeping for this to be efficient. c C s< | j jr,| jrd S ddd� | j jD �kr,d S |�| d� d S )NFc s s | ] }|j V qd S )N)r )r\ r r r r! � <genexpr>e s z=DetectKeySwitch.per_property_preprocessors.<locals>.<genexpr>)r r_ r r) )r r* r r r! r+ _ s z*DetectKeySwitch.per_property_preprocessorsc C s6 t �|| jj�}t �|| dd�}|j�||fg� d S )NF)r r- r r: r, rs rt )r r* r3 r1 r r r! r7 l s z*DetectKeySwitch.per_property_flush_actionsc C s d S )Nr )r r* r> rA r r r! rF q s z'DetectKeySwitch.per_state_flush_actionsc C s d S )Nr )r rG r> r r r! rH t s z DetectKeySwitch.presort_deletesc C s | j s| �||� d S )N)r �_process_key_switches)r r* r> r r r! rI w s z DetectKeySwitch.presort_savesc C s" |s | j r | �||�}t|�S dS )NF)r �_key_switchers�bool)r r* r> rA �dr r r! rQ } s z DetectKeySwitch.prop_has_changesc C s dst �d S )NF)r9 )r rG r> r r r! rJ � s z DetectKeySwitch.process_deletesc C s | j s t�| �||� d S )N)r r9 r� )r rG r> r r r! rK � s z DetectKeySwitch.process_savesc C s^ |� d| fdd� �\}}|�|�}x6|D ].}||kr(| �||�rL|�|� q(|�|� q(W |S )NZ pk_switchersc S s t � t � fS )N)rc r r r r! �<lambda>� � z0DetectKeySwitch._key_switchers.<locals>.<lambda>)rb �unionrj r� )r r* r> ZswitchedZ notswitchedZ allstatesrO r r r! r� � s z DetectKeySwitch._key_switchersc C s� | � ||�}|r�x�|jj�� D ]�}t|j| jj�s4q |j}|�| j �j ||| j d�}|t j k r |d k r | jjr~|stq |d }n|}t �|�}||kr |�|d| j� t�|| j|| j| jj|| j� q W d S )N)rN � F)r� r� Z identity_mapZ all_states� issubclassrV r r= r& r �getr r ZPASSIVE_NO_RESULTr rU Zinstance_stater} r r r� r r ) r ZdeplistrG Z switchersr( Zdict_rg Z related_objZ related_stater r r! r� � s6 z%DetectKeySwitch._process_key_switchesc C s t |j�o t�||| j| jj�S )N)r� r r r� r r r )r rG r( r r r! rj � s z DetectKeySwitch._pks_changedN)rl rn ro �__doc__r+ r7 rF rH rI rQ rJ rK r� r� rj r r r r! r� S s %r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� ManyToManyDPc C s: |j �||f||f||f||f||f||f||fg� d S )N)rs rt )r r* r3 r4 r5 r6 r1 r2 r r r! r0 � s z&ManyToManyDP.per_property_dependenciesc C sX |s<|r"|j �||f||fg� qT|j �||f||fg� n|j �||f||fg� d S )N)rs rt ) r r* rw rx rC r1 r2 rA rE r r r! r@ � s z#ManyToManyDP.per_state_dependenciesc C s* | j s&x |D ]}|�|| j| j� q W d S )N)r rL r r )r rG r> r( r r r! rH � s z ManyToManyDP.presort_deletesc C s� | j s2x*|D ]"}| �||�r |�|| jtj�}q W | jjs>d S x�|D ]z}|�|| jtj�}|rDx^|j D ]T}| � |�dkrf|j |dd| j d� x,| j �d|�D ]\}}}} |j |dd� q�W qfW qDW d S )NFTrz )rA r{ r )rA )r rj rL r r r r r| r rD r' r} r r r� ) r rG r> r( rP rB r� r� r� r� r r r! rI s* zManyToManyDP.presort_savesc s� g }g }g }| � |�}t� }x�|D ]�� |�� | j| j�}|r"xR|�� D ]F} | d ksF|d k rh� | f|krhqFi } | �� | | d|d�s�qF|�| � qFW |�� fdd�|�� D �� q"W |d k r�|�|� | � ||||� d S )NFrz c 3 s | ]}|� fV qd S )Nr )r\ r� )r( r r! r� K s z/ManyToManyDP.process_deletes.<locals>.<genexpr>) rd rc rL r r Z non_addedr[ r? rt � _run_crud) r rG r> �secondary_delete�secondary_insert�secondary_update� processed�tmprP rB rY r )r( r! rJ , s8 z ManyToManyDP.process_deletesc s� g }g }g }| � |�}t� }�xP|D �]F� | j o<| �|� �}|rJtj} ntj} |�� | j| �} | r$xH| j D ]>} |d k r�� | f|kr�qli } | � � | | d|d�s�ql|� | � qlW xH| j D ]>} |d k rЈ | f|kr�q�i } | � � | | d|d�s�q�|� | � q�W |� � fdd�| j | j D �� |r$xN| jD ]D} i } t� � | j| d| jj� t� | | j| d| jj� |� | � �q$W q$W |d k �r�|� |� | �||||� d S )NFr� rz c 3 s | ]}|� fV qd S )Nr )r\ r� )r( r r! r� ~ s z-ManyToManyDP.process_saves.<locals>.<genexpr>�old_)rd rc r rj r r r rL r r� r[ r? rD rt r~ r r r r r �secondary_synchronize_pairsr� ) r rG r> r� r� r� r� r� Zneed_cascade_pksrN rP rB rY r )r( r! rK T sn zManyToManyDP.process_savesc s |j �| j�}|r~|d � | j�tj� fdd�| jjD �� �}|�||�}|� � r~|j t |�kr~t � d| jjt |�|j f ��|r�|d � | j�tj� fdd�| jjD �� �}|�||�}|� � r�|j t |�kr�t � d| jjt |�|j f ��|�r | j�� }|�||� d S )Nr� c s, g | ]$}|j � kr|tj|j |jd �k�qS ))�type_)r r � bindparam�type)r\ r� )rY r r! r^ � s z*ManyToManyDP._run_crud.<locals>.<listcomp>zRDELETE statement on table '%s' expected to delete %d row(s); Only %d were matched.c s0 g | ](}|j � kr|tjd |j |jd�k�qS )r� )r� )r r r� r� )r\ r� )rY r r! r^ � s zRUPDATE statement on table '%s' expected to update %d row(s); Only %d were matched.)Z transaction� connectionr r rz r �and_r� �executeZ supports_sane_multi_rowcountZrowcount�lenr ZStaleDataError� descriptionrt �insert)r rG r� r� r� r� � statement�resultr )rY r! r� � s@ zManyToManyDP._run_crudc C s� | � |� |d krdS |d k rP|j�|�sP|jsLt�dt�|�|| jf � dS t � || j || jj � t � || j || jj� dS )NFzGObject of type %s not in session, %s operation along '%s' won't proceedT)rW r� r� rD r r� r� r� r r Z populate_dictr r r r� )r r( rB rY rZ rG r{ r r r! r[ � s" zManyToManyDP._synchronizec C s t �||| j| jj�S )N)r r� r r r )r rG r( r r r! rj � s zManyToManyDP._pks_changedN) rl rn ro r0 r@ rH rI rJ rK r� r[ rj r r r r! r� � s "(H=!r� N)r� � r r r r r r� Z interfacesr r r r r �objectr rq r� r� r� r# r r r r! �<module> s4 & 3 fu ;