관리-도구
편집 파일: dependency.cpython-38.pyc
U -?�f� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd l m Z dd l m Z dd l m Z d dlmZ d d lmZ d dlmZ G dd� d�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Ze ee ee eiZdS )z Relationship dependencies. � )� annotations� )� 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 | _d| jj|jf | _|j| _| j js�t�d| j ��d S )Nz%s_%sz|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� _sort_key�key�sort_key�synchronize_pairs�sa_exc� ArgumentError)�selfr � r$ �I/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/dependency.py�__init__ s0 ��z DependencyProcessor.__init__c C s t |j |�S �N)�_direction_to_processorr )�clsr r$ r$ r% �from_relationship; 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_preprocessorsG 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# r1 � after_save� before_delete� parent_saves� child_saves�parent_deletes� child_deletesr$ r$ r% �per_property_flush_actionsP s6 � � � ��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} |D �]}|j | j j �||j|r�| jntj�}|�sq�|�r8t�|| d|�}| �rZt�||�} n"t�|| d|�}| �rZt�||�} | �r�g } |D ]\\}}||jk�r�d}n6|j| \}}|�r�t�||�df}nt�||�df}| �|� �qh| D ]"\}}| �|| | |||||� �q�q�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 r6 r r5 r7 r4 �disabledZcycles�AssertionErrorr � base_mapper�managerr �implZget_all_pending�dictr r r Z ProcessStateZ DeleteStateZSaveUpdateState�states�append�per_state_dependencies)r# r1 rF �isdeleteZchild_base_mapperr< r> r: r9 Z child_actionsZchild_in_cyclesr; r= Zparent_in_cyclesr. Zsum_Z child_state�child� child_action�deletedZlistonly� childisdeleter$ r$ r% �per_state_flush_actionsl s� � �� � � �z+DependencyProcessor.per_state_flush_actionsc C s dS �NFr$ �r# � uowcommitrF r$ r$ r% �presort_deletes� s z#DependencyProcessor.presort_deletesc C s dS rO r$ rP r$ r$ r% � presort_saves� s z!DependencyProcessor.presort_savesc C s d S r' r$ rP r$ r$ r% �process_deletes� s z#DependencyProcessor.process_deletesc C s d S r' r$ rP r$ r$ r% � process_saves� s z!DependencyProcessor.process_savesc C s| |r | j rtjtjB }n | jtkr*tj}n tjtjB }|D ]&}|�|| j |�}|r:|� � s: dS q:|oz| j j oz| j |jkS )NT)r r r �INCLUDE_PENDING_MUTATIONSr r ZPASSIVE_NO_FETCH_RELATEDr �get_attribute_historyr �emptyr Z_is_self_referentialr Zmappers)r# rQ rF rI �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.)r] r^ r_ )r �uselistr Z FlushErrorr Z_canloadr �class_r- r$ r$ r% �_verify_canload s8 �� � ������z#DependencyProcessor._verify_canloadc C s t � �d S r' ��NotImplementedError)r# r. rJ �associationrow� clearkeysrQ r$ r$ r% � _synchronize3 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# r1 Z process_keyr$ r$ r% �_get_reversed_processed_set6 s �z/DependencyProcessor._get_reversed_processed_setFc C s8 |D ].}|r|d k r|� |dd� | jjD �� q4qd S )Nc S s g | ] \}}|�qS r$ r$ )rh �l�rr$ r$ r% rj C s z4DependencyProcessor._post_update.<locals>.<listcomp>)Zregister_post_updater r )r# r. rQ �related� is_m2o_deleter] r$ r$ r% � _post_update? s �z DependencyProcessor._post_updatec C s t � �d S r' rc �r# rQ r. r$ r$ r% � _pks_changedG s z DependencyProcessor._pks_changedc C s d| j j| jf S )Nz%s(%s))� __class__�__name__r )r# r$ r$ r% �__repr__J s z DependencyProcessor.__repr__N)F)ry � __module__� __qualname__r& � classmethodr* r, r2 r? rN rR rS rT rU r\ rb rg rp ru rw rz r$ r$ r$ r% r s$ r ) 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 r3 �r r � PostUpdateAllr r6 � dependencies�update) r# r1 r; r< r= r> r9 r: �child_post_updates�child_pre_updatesr$ r$ r% r8 O 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 r3 r ) r# r1 � save_parent� delete_parentrK r9 r: rI rM r� r� r$ r$ r% rH x sZ � ������������z"OneToManyDP.per_state_dependenciesc C s� | j j o| jdk }|D ]�}|�|| j| j�}|r|jD ]<}|d k r:| �|�dkr:| j jrl|j |dd� q:|� |� q:|r|j D ] }|d k r�|j |d| j d� q�qd S )N�allFT�rI �delete�� operationr ) r r� r rW r r rL r, � delete_orphan�register_object� unchangedr �r# rQ rF �should_null_fksr. r[ rJ r$ r$ r% rR � s, � � �zOneToManyDP.presort_deletesc C sd |� d| ft�}| jj o$| jdk }|D �]2}| �||�}|rF| jrTtjtj B }n tj tj B }|� || j |�}|�r&|j D ] } | d k r||j| dd| jd� q||�|j � |jD ]t} | jjs�|r�|j| dd| jd� q�| �| �dkr�|j| dd| jd� | j�d| �D ] \} } } } |j| dd � �qq�|r*|r*|jD ](} | d k �r4|j| d| jd | jd � �q4q*d S ) N�children_addedr� T�add)Z cancel_deleter� r Fr� �rI r� r r� z pk changer� )rn ro r r� r rw r r r rV r rW r �addedr� r r� rL r, r �cascade_iteratorr� )r# rQ rF r� r� r. � pks_changedrY r[ rJ �c�m�st_�dct_r$ r$ r% rS � st � ���� � �� � �zOneToManyDP.presort_savesc C s� | j s| jdks�|�d| ft�}|D ]�}|�|| j| j�}|r$|jD ]H}|d k rD| �|�dkrD| � ||d d|d� | j rD|rD| � |||g� qD| j s�| j j s$t|j ��|�D ]:}|d k r�| � ||d d|d� | j r�|r�| � |||g� q�q$d S )Nr� r� FT)r r rn ro rW r r rL r, rg ru r r� r� � difference)r# rQ rF r� r. r[ rJ r$ r$ r% rT sV � � � � � � �zOneToManyDP.process_deletesc C s� | j j o| jdk }|D ]�}|�|| jtj�}|r|jD ]6}| �||d d|d� |d k r:| j r:| � |||g� q:|j D ].}|rx| j jsx| � |�sx| �||d d|d� qx| � ||�r|jD ]}| �||d d|d� q�qd S )Nr� FT)r r� r rW r r r r� rg r ru rL r, rw r� r� r$ r$ r% rU D s\ � � � ��� � �zOneToManyDP.process_savesc C sp |}|}| � |� |d ks*| js.|�|�r.d S |rHt�|| j| jj� n$t�|| j || j| jj|| j oh|� d S r' ) rb r � is_deletedr �clearr r r �populater r ) r# r. rJ re rf rQ r� �source�destr$ r$ r% rg f s( ���zOneToManyDP._synchronizec C s t �||| j| jj�S r' �r �source_modifiedr r r rv r$ r$ r% rw } s �zOneToManyDP._pks_changedN) ry r{ r| r8 rH rR rS rT rU rg rw r$ r$ r$ r% r~ N s )AD,"r~ c @ sN e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z ddd�Z dS )� ManyToOneDPc C s. t �| |� | jjD ]}|j�t|�� qd S r' )r r& r Zself_and_descendantsZ_dependency_processorsrG �DetectKeySwitch)r# r r r$ r$ r% r& � 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 r3 �r r r� r r6 r� r� ) r# r1 r; r< r= r> r9 r: �parent_post_updates�parent_pre_updatesr$ r$ r% r8 � s8 � ��� ��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 r3 r� ) r# r1 r� r� rK r9 r: rI rM r� r� r$ r$ r% rH � sL ����� ����z"ManyToOneDP.per_state_dependenciesc C s� | j js| j jr�|D ]�}|�|| j| j�}|r| j jr@|�� }n|�� }|D ]P}|d krZqL|j|dd| j d� | j � d|�}|D ]\}} } } |j| dd� q�qLqd S )NTr� r� r� ) r r� r� rW r r �sumZ non_deletedr� r r r� ) r# rQ rF r. r[ ZtodeleterJ �tr� r� r� r� r$ r$ r% rR � s. � �zManyToOneDP.presort_deletesc C s� |D ]�}|j |d| jd� | jjr|�|| j| j�}|r|jD ]T}| �|�dkr>|j |dd| jd� | j � d|�}|D ]\}}} } |j | dd� qvq>qd S )Nr� r� FTr� r� r� ) r� r r r� rW r r rL r, r r� ) r# rQ rF r. r[ rJ r� r� r� r� r� r$ r$ r% rS s( � �zManyToOneDP.presort_savesc C sn | j rj| jjsj| jdksj|D ]L}| �|d d d|� |r | j r |�|| j| j�}|r | j|||� � dd� q d S )Nr� T)rt ) r r r� r rg rW r r ru r� )r# rQ rF r. r[ r$ r$ r% rT s* ��� � �zManyToOneDP.process_deletesc C s~ |D ]t}|� || jtj�}|r|jrF|jD ]}| �||d d|d� q*n|jr`| �|d d d|d� | jr| �|||� � � qd S )NFr� Tr� ) rW r r r r� rg rL r ru r� )r# rQ rF r. r[ rJ r$ r$ r% rU ) s8 � � �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 )N�GObject 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 rb r� r �r# r. rJ re rf rQ r� r$ r$ r% rg ; s< ���� ��� �zManyToOneDP._synchronize)N) ry r{ r| r& r8 rH rR rS rT rU rg r$ r$ r$ r% r� � s '8 �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 r' )r )rh r r$ r$ r% � <genexpr>u s �z=DetectKeySwitch.per_property_preprocessors.<locals>.<genexpr>)r rk r r/ r0 r$ r$ r% r2 p s �z*DetectKeySwitch.per_property_preprocessorsc C s6 t �|| jj�}t �|| dd�}|j�||fg� d S rO )r r5 r rB r4 r� r� )r# r1 r; r9 r$ r$ r% r? } s z*DetectKeySwitch.per_property_flush_actionsc C s d S r' r$ )r# r1 rF rI r$ r$ r% rN � s z'DetectKeySwitch.per_state_flush_actionsc C s d S r' r$ rP r$ r$ r% rR � s z DetectKeySwitch.presort_deletesc C s | j s| �||� d S r' )r �_process_key_switches)r# r1 rF r$ r$ r% rS � s z DetectKeySwitch.presort_savesc C s" |s | j r | �||�}t|�S dS rO )r �_key_switchers�bool)r# r1 rF rI �dr$ r$ r% r\ � s z DetectKeySwitch.prop_has_changesc C s dst �d S rO )rA rP r$ r$ r% rT � s z DetectKeySwitch.process_deletesc C s | j s t�| �||� d S r' )r rA r� rP r$ r$ r% rU � s z DetectKeySwitch.process_savesc C sZ |� d| fdd� �\}}|�|�}|D ].}||kr&| �||�rJ|�|� q&|�|� q&|S )NZ pk_switchersc S s t � t � fS r' )ro r$ r$ r$ r% �<lambda>� � z0DetectKeySwitch._key_switchers.<locals>.<lambda>)rn �unionrw r� )r# r1 rF ZswitchedZ notswitchedZ allstatesrZ r$ r$ r% r� � s � z DetectKeySwitch._key_switchersc C s� | � ||�}|r�|jj�� D ]�}t|j| jj�s2q |j}|�| j �j ||| j d�}|t j k r |d k r | jjr||srq |d }n|}t �|�}||kr |�|d| j� t�|| j|| j| jj|| j� q d S )N)rY r F)r� r� Z identity_mapZ all_states� issubclassra r rE r+ r �getr r ZPASSIVE_NO_RESULTr r` Zinstance_stater� r r r� r r ) r# ZdeplistrQ Z switchersr. Zdict_rs Z related_objZ related_stater$ r$ r% r� � sH ��� ��z%DetectKeySwitch._process_key_switchesc C s t |j�o t�||| j| jj�S r' )r� r r r� r r r rv r$ r$ r% rw � s �z DetectKeySwitch._pks_changedN)ry r{ r| �__doc__r2 r? rN rR rS r\ rT rU r� r� rw r$ r$ r$ r% r� d 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 r' �r� r� )r# r1 r; r< r= r> r9 r: r$ r$ r% r8 � s ��z&ManyToManyDP.per_property_dependenciesc C sX |s<|r"|j �||f||fg� qT|j �||f||fg� n|j �||f||fg� d S r' r� ) r# r1 r� r� rK r9 r: rI rM r$ r$ r% rH � s ���z#ManyToManyDP.per_state_dependenciesc C s& | j s"|D ]}|�|| j| j� q d S r' )r rW r r )r# rQ rF r. r$ r$ r% rR s �z ManyToManyDP.presort_deletesc C s� | j s.|D ]"}| �||�r |�|| jtj�}q | jjs:d S |D ]r}|�|| jtj�}|r>|j D ]P}| � |�dkr^|j |dd| j d� | j �d|�D ]\}}}} |j |dd� q�q^q>d S )NFTr� r� r� )r rw rW r r r r r� r rL r, r� r r r� ) r# rQ rF r. r[ rJ r� r� r� r� r$ r$ r% rS s< � � � �zManyToManyDP.presort_savesc s� g }g }g }| � |�}t� }|D ]�� |�� | j| j�}|r |�� D ]F} | d ksB|d k rd� | f|krdqBi } | �� | | d|d�s~qB|�| � qB|�� fdd�|�� D �� q |d k r�|�|� | � ||||� d S )NFr� c 3 s | ]}|� fV qd S r' r$ �rh r� �r. r$ r% r� [ s z/ManyToManyDP.process_deletes.<locals>.<genexpr>) rp ro rW r r Z non_addedrg rG r� � _run_crud) r# rQ rF �secondary_delete�secondary_insert�secondary_update� processed�tmpr[ rJ re r$ r� r% rT < sN � � �� �z ManyToManyDP.process_deletesc s� g }g }g }| � |�}t� }|D �]F� | j o8| �|� �}|rLtjtjB } n tjtjB } |�� | j | �} | r | j D ]>} |d k r�� | f|kr�qri } | � � | | d|d�s�qr|� | � qr| j D ]>} |d k r҈ | f|kr�q�i } | � � | | d|d�s�q�|� | � q�|�� fdd�| j | j D �� |r | jD ]D} i } t�� | j| d| jj� t�| | j| d| jj� |� | � �q"q |d k �r~|�|� | �||||� d S )NFr� r� c 3 s | ]}|� fV qd S r' r$ r� r� r$ r% r� � s z-ManyToManyDP.process_saves.<locals>.<genexpr>�old_)rp ro r rw r r rV r rW r r� rg rG rL r� r� r r r r r �secondary_synchronize_pairsr� ) r# rQ rF r� r� r� r� r� Zneed_cascade_pksrY r[ rJ re r$ r� r% rU d s� ����� � � �� �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 Z bindparam�typer� �re r$ r% rj � 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% rj � s �zRUPDATE statement on table '%s' expected to update %d row(s); Only %d were matched.)Z transaction� connectionr r r� �wherer �and_r� �executeZ supports_sane_multi_rowcountZrowcount�lenr ZStaleDataError� descriptionr� �insert)r# rQ r� r� r� r� Z statement�resultr$ r� 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 )NFr� T)rb r� r� rL r r� r� r� r r Z populate_dictr r r r� r� r$ r$ r% rg � s0 �� ��zManyToManyDP._synchronizec C s t �||| j| jj�S r' r� rv r$ r$ r% rw s �zManyToManyDP._pks_changedN) ry r{ r| r8 rH rR rS rT rU r� rg rw r$ r$ r$ r% r� � s "(M= r� N)r� � __future__r � r r r r r r� Z interfacesr r r r! r r r~ r� r� r� r( r$ r$ r$ r% �<module> s> 3 7 bu > �