관리-도구
편집 파일: unitofwork.cpython-38.pyc
U -?�f�i � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd l m Z dd l m Z dd l m Z d d l mZ d d l m Z d dl mZ er�ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd� Z G dd� d�Z G dd� d�Z G dd � d e �Z G d d � d �Z G d d � d e e �Z!G d!d"� d"e �Z"G d#d$� d$e �Z#G d%d&� d&e �Z$G d'd(� d(e �Z%G d)d*� d*e �Z&G d+d,� d,e �Z'd-S ).z�The internals for the unit of work system. The session's flush() process passes objects to a contextual object here, which assembles flush tasks based on mappers and their properties, organizes them in order of dependency, and executes. � )� annotations)�Any)�Dict)�Optional)�Set)� TYPE_CHECKING� �� attributes)�exc)�util� )�event)� topological)�DependencyProcessor)�MapperProperty)�Mapper)�Session)�SessionTransaction)� InstanceStatec s� |j � � fdd�}� fdd�}� fdd�}tj| d|ddd � tj| d |dddd � tj| d |dddd � tj| d |dddd � dS )z\Establish event listeners on object attributes which handle cascade-on-set/append. c sh |d kr d S | j }|rd|jr&|�d� | jjj� }t�|�}|jj rd� |j krd|� |�sd|� |� |S )Nzcollection append) �session�_warn_on_events�_flush_warning�manager�mapper�_propsr �instance_state�_cascade� save_update�key�_contains_state�_save_or_update_state��state�item� initiator�kw�sess�propZ item_state�r � �I/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/unitofwork.py�append2 s ��� z$track_cascade_events.<locals>.appendc s� |d kr d S | j }| jjj� }|r>|jr>|�|jr8dnd� |d k r�|tjk r�|tj k r�|j j r�t� |�}|j� |�r�|r�||jkr�|�|� nd|_d S )Nzcollection removezrelated attribute deleteT)r r r r r r Zuselistr � NEVER_SET�PASSIVE_NO_RESULTr � delete_orphanr � _is_orphan�_new�expungeZ _orphaned_outside_of_sessionr" r) r* r+ �removeI s. ������ z$track_cascade_events.<locals>.removec s� ||kr |S | j }|r�|jr&|�d� | jjj� }|d k rlt�|�}|jj rl� |j krl|� |�sl|� |� |d k r�|tj k r�|tjk r�|jjr�t�|�}||jkr�|j�|�r�|�|� |S )Nzrelated attribute set)r r r r r r r r r r r r r! r- r. r/ r1 r0 r2 ) r# ZnewvalueZoldvaluer% r& r' r( Znewvalue_stateZoldvalue_stater) r* r+ �set_k s< ��� ���� � z"track_cascade_events.<locals>.set_Zappend_wo_mutationT)�raw� include_keyr, )r5 �retvalr6 r3 �setN)r r �listen)Z descriptorr( r, r3 r4 r* r) r+ �track_cascade_events+ sF " $ � � � �r: c @ s� e Zd ZU ded< ded< ded< ded< d ed < dd �d d �Zedd� �Zdd� Zdd� Zdd� Z dd� Z e j fdd�Z dd� Zd d � Zd9d d!d!d!d"d#d!d$�d%d&�Zd'd(� Zd)d*� Zejd+d,� �Zd-d.� Zd/d0� Zd1d2� Zd3d4�d5d6�Zd3d4�d7d8�Zd S ):�UOWTransactionr r r Z transactionzDict[str, Any]r z7util.defaultdict[Mapper[Any], Set[DependencyProcessor]]�depsz6util.defaultdict[Mapper[Any], Set[InstanceState[Any]]]�mappers)r c C sR || _ i | _t�t�| _t�t�| _i | _i | _t� | _ i | _ t�dd� �| _ d S )Nc S s t � t � fS �N)r8 r* r* r* r+ �<lambda>� � z)UOWTransaction.__init__.<locals>.<lambda>) r r r � defaultdictr8 r< r= �presort_actions�postsort_actions� dependencies�states�post_update_states)�selfr r* r* r+ �__init__� s zUOWTransaction.__init__c C s t | j�S r> )�boolrE �rG r* r* r+ �has_work� s zUOWTransaction.has_workc C sD |j r@z|�|tj� W n& tjk r> | j�|g� Y dS X dS )zZReturn ``True`` if the given state is expired and was deleted previously. TF)ZexpiredZ _load_expiredr � PASSIVE_OFF�orm_excZObjectDeletedErrorr �_remove_newly_deleted�rG r# r* r* r+ �was_already_deleted� s z"UOWTransaction.was_already_deletedc C s || j ko| j | d S )z[Return ``True`` if the given state is marked as deleted within this uowtransaction.r �rE rO r* r* r+ � is_deleted� s zUOWTransaction.is_deletedc C s, || j kr| j | S |� | j |< }|S d S r> r )rG r Z callable_�retr* r* r+ �memo� s zUOWTransaction.memoc C s | j | d }|df| j |< dS )z;Remove pending actions for a state from the uowtransaction.r TNrQ )rG r# �isdeleter* r* r+ �remove_state_actions� s z#UOWTransaction.remove_state_actionsc C s� d||f}|| j kr�| j | \}}}|t j@ s�|t j@ r�|j| j}|�||jt jt jB t jB �}|rv|j rv|� � }n|}|||f| j |< nP|j| j}|�||j|t jB t jB �}|r�|j r�|� � }n|}|||f| j |< |S )zOFacade to attributes.get_state_history(), including caching of results.�history) r ZSQL_OKr �implZ get_history�dictrL ZLOAD_AGAINST_COMMITTEDZNO_RAISEZ uses_objectsZas_state) rG r# r ZpassiveZhashkeyrW Z state_historyZcached_passiverX r* r* r+ �get_attribute_history� sH �� ��� ��� z$UOWTransaction.get_attribute_historyc C s |df| j kS )NT)rB )rG � processorr* r* r+ �has_dep* s zUOWTransaction.has_depc C s& ||f}|| j kr"t||�| j |< d S r> )rB � Preprocess)rG r[ � fromparentr r* r* r+ �register_preprocessor- s z$UOWTransaction.register_preprocessorFNzInstanceState[Any]rI z Optional[str]zOptional[MapperProperty])r# rU �listonly� cancel_delete� operationr( �returnc C s� | j �|�s8|js4|d k r4t�dt�|�||f � dS || jkr~|jj }|| j kr^| � |� | j | � |� ||f| j|< n|s�|s�|r�|df| j|< dS )NzJObject of type %s not in session, %s operation along '%s' will not proceedFT) r r Zdeletedr �warn�orm_utilZstate_class_strrE r r r= �_per_mapper_flush_actions�add)rG r# rU r` ra rb r( r r* r* r+ �register_object2 s" �� z UOWTransaction.register_objectc C s0 |j jj}| j| \}}|�|� |�|� d S r> )r r � base_mapperrF rg �update)rG r# Zpost_update_colsr rE �colsr* r* r+ �register_post_updateT s z#UOWTransaction.register_post_updatec C sf t | |j�}t| |j�}| j�||f� |jD ]}|�| � q.|jD ] }|jrPqD|j }|�| � qDd S r> ) � SaveUpdateAllri � DeleteAllrD rg Z_dependency_processorsZper_property_preprocessorsZ relationshipsZviewonlyZ_dependency_processor)rG r ZsavesZdeletes�depr( r* r* r+ rf Z s z(UOWTransaction._per_mapper_flush_actionsc C s t �dd� �S )a return a dynamic mapping of (Mapper, DependencyProcessor) to True or False, indicating if the DependencyProcessor operates on objects of that Mapper. The result is stored in the dictionary persistently once calculated. c S s | d j �| d j�| d jkS )Nr r )r �getr r( )�tupr* r* r+ r? s r@ z0UOWTransaction._mapper_for_dep.<locals>.<lambda>)r Z PopulateDictrJ r* r* r+ �_mapper_for_deph s �z UOWTransaction._mapper_for_depc s | j �� �fdd�|D �S )zmFilter the given list of InstanceStates to those relevant to the given DependencyProcessor. c s g | ]}�|j j� f r|�qS r* )r r ��.0�s�ro Zmapper_for_depr* r+ � <listcomp>| s z8UOWTransaction.filter_states_for_dep.<locals>.<listcomp>)rr )rG ro rE r* rv r+ �filter_states_for_depv s z$UOWTransaction.filter_states_for_depc c s> ||f}|j jD ](}| j| D ]}| j| |kr |V q qd S r> )ri �self_and_descendantsr= rE )rG r rU r` Zchecktupr# r* r* r+ �states_for_mapper_hierarchy~ s z*UOWTransaction.states_for_mapper_hierarchyc sB d}t � j�� �D ]}|�� �rd}q|s q.q t�� jt � j�� �� � _}|�r(� fdd�|D �}t � j�D ]�}d|ks�|d j s�|d j s�|� |�r�� j� |� qn|d |kr� j� |� ||d D ]}� j� ||d f� q�qn|d |krn� j� |� ||d D ]}� j� |d |f� �q qndd � � j�� D �� |�S ) z}Generate the full, unsorted collection of PostSortRecs as well as dependency pairs for this UOWTransaction. FTc s i | ]}|t |�� ���qS r* )r8 �per_state_flush_actions)rt �recrJ r* r+ � <dictcomp>� s z4UOWTransaction._generate_actions.<locals>.<dictcomp>Nr r c S s h | ]}|j s|�qS r* ��disabled)rt �ar* r* r+ � <setcomp>� s z3UOWTransaction._generate_actions.<locals>.<setcomp>)�listrB �values�executer Z find_cyclesrD rC �cyclesr � issupersetr3 rg � difference)rG rS �actionr� �convert�edgero r* rJ r+ �_generate_actions� sJ � ����� ��z UOWTransaction._generate_actions�None)rc c C sx | � � }t|dd� d�}| jrVt�| j|�D ]&}t|�}|r,|�� }|�| |� q8q,n t� | j|�D ]}|� | � qdd S )Nc S s | j S r> )�sort_key)r$ r* r* r+ r? � r@ z(UOWTransaction.execute.<locals>.<lambda>r) ) r� �sortedr� r Zsort_as_subsetsrD r8 �pop�execute_aggregate�sortr� )rG rC Zsubsetr4 �nr| r* r* r+ r� � s � �zUOWTransaction.executec C sV | j s dS t| j �}dd� | j �� D �}|�|�}|rB| j�|� |rR| j�|� dS )z�Mark processed objects as clean / deleted after a successful flush(). This method is called within the flush() method after the execute() method has succeeded and the transaction has been committed. Nc S s h | ]\}\}}|r|�qS r* r* )rt ru rU r` r* r* r+ r� � s z8UOWTransaction.finalize_flush_changes.<locals>.<setcomp>)rE r8 �itemsr� r rN Z_register_persistent)rG rE Zisdel�otherr* r* r+ �finalize_flush_changes� s � z%UOWTransaction.finalize_flush_changes)FFFNN)�__name__� __module__� __qualname__�__annotations__rH �propertyrK rP rR rT rV r ZPASSIVE_NO_INITIALIZErZ r\ r_ rh rl rf r Zmemoized_propertyrr rx rz r� r� r� r* r* r* r+ r; � s> - � 5 � " 4r; c @ s e Zd ZdZdd� ZdS )�IterateMappersMixinr* c s2 � j r$t� �fdd�� jjjD ��S � jjjS d S )Nc 3 s" | ]}�j |� jf r|V qd S r> )rr �dependency_processor)rt �m�rG �uowr* r+ � <genexpr>� s �z/IterateMappersMixin._mappers.<locals>.<genexpr>)r^ �iterr� �parentry r r� r* r� r+ �_mappers� s �z IterateMappersMixin._mappersN)r� r� r� � __slots__r� r* r* r* r+ r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )r] )r� r^ � processed�setup_flush_actionsc C s || _ || _t� | _d| _d S �NF)r� r^ r8 r� r� )rG r� r^ r* r* r+ rH s zPreprocess.__init__c C s� t � }t � }| �|�D ]H}|j| �| j�D ]0}|j| \}}|s,|rR|�|� q,|�|� q,q|r~| j�||� | j� |� |r�| j� ||� | j� |� |s�|r�| j s�| j� ||d�s�| j� ||d�r�| j� |� d| _ dS dS d S �NTF)r8 r� r= r� r� rE rg r� Zpresort_deletesrj Z presort_savesr� Zprop_has_changesZper_property_flush_actions)rG r� Z delete_statesZ save_statesr r# rU r` r* r* r+ r� sB �� �� zPreprocess.executeN)r� r� r� r� rH r� r* r* r* r+ r] � s r] c @ s e Zd ZdZdd� Zdd� ZdS )� PostSortRecr~ c G s@ | f| }||j kr |j | S t�| � |j |< }d|_|S d S r� )rC �object�__new__r )�clsr� �argsr rS r* r* r+ r� - s zPostSortRec.__new__c C s | � |� d S r> )r� )rG r� �recsr* r* r+ r� 6 s z PostSortRec.execute_aggregateN)r� r� r� r� r� r� r* r* r* r+ r� * s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � ProcessAll)r� rU r^ r� c C s: || _ d| j j|f| _|| _|| _|j|jj �|� d S )Nr� )r� r� rU r^ r< r� ri rg )rG r� r� rU r^ r* r* r+ rH = s ��zProcessAll.__init__c C s2 | � |�}| jr | j�||� n| j�||� d S r> )� _elementsrU r� �process_deletes� process_saves)rG r� rE r* r* r+ r� J s zProcessAll.executec C s t g �S r> )r� r� r* r* r+ r{ Q s z"ProcessAll.per_state_flush_actionsc C s d| j j| j| jf S )Nz%s(%s, isdelete=%s))� __class__r� r� rU rJ r* r* r+ �__repr__X s �zProcessAll.__repr__c c sF | � |�D ]6}|j| D ]&}|j| \}}|| jkr|s|V qq d S r> )r� r= rE rU )rG r� r r# rU r` r* r* r+ r� _ s zProcessAll._elementsN) r� r� r� r� rH r� r{ r� r� r* r* r* r+ r� : s r� c @ s* e Zd ZdZdd� Ze�d�dd� �ZdS )� PostUpdateAll)r rU r� c C s || _ || _d|j|f| _d S )Nr� )r rU � _sort_keyr� )rG r� r rU r* r* r+ rH j s zPostUpdateAll.__init__�sqlalchemy.orm.persistencec sB t jj}�j� j \}}� �fdd�|D �}|�� j|�|� d S )Nc s$ g | ] }�j | d � jkr|�qS �r )rE rU rs r� r* r+ rw s s z)PostUpdateAll.execute.<locals>.<listcomp>)r � preloaded�orm_persistencerF r Z post_update)rG r� � persistencerE rk r* r� r+ r� o s zPostUpdateAll.executeN)r� r� r� r� rH r �preload_moduler� r* r* r* r+ r� g s r� c @ s: e Zd ZdZdd� Ze�d�dd� �Zdd� Zd d � Z d S ) rm �r r� c C s$ || _ d|jf| _||jks t�d S )Nrm �r r� r� ri �AssertionError�rG r� r r* r* r+ rH { s zSaveUpdateAll.__init__r� c C s$ t jj�| j|�| jdd�|� d S r� )r r� r� �save_objr rz r� r* r* r+ r� � s �zSaveUpdateAll.executec c s� t |�| jdd��}| jj}t||�}|D ]$}t||�}|j�||f� |V q*|j| j D ] }|� ||�}|� ||d� q\d S r� ) r� rz r ri rn �SaveUpdateStaterD rg r< rx r{ ) rG r� rE ri Z delete_allr# r� ro �states_for_propr* r* r+ r{ � s � z%SaveUpdateAll.per_state_flush_actionsc C s d| j j| jf S �Nz%s(%s)�r� r� r rJ r* r* r+ r� � s zSaveUpdateAll.__repr__N� r� r� r� r� rH r r� r� r{ r� r* r* r* r+ rm x s rm c @ s: e Zd ZdZdd� Ze�d�dd� �Zdd� Zd d � Z d S ) rn r� c C s$ || _ d|jf| _||jks t�d S )Nrn r� r� r* r* r+ rH � s zDeleteAll.__init__r� c C s$ t jj�| j|�| jdd�|� d S r� )r r� r� � delete_objr rz r� r* r* r+ r� � s �zDeleteAll.executec c s� t |�| jdd��}| jj}t||�}|D ]$}t||�}|j�||f� |V q*|j| j D ] }|� ||�}|� ||d� q\d S r� ) r� rz r ri rm � DeleteStaterD rg r< rx r{ ) rG r� rE ri Zsave_allr# r� ro r� r* r* r+ r{ � s � z!DeleteAll.per_state_flush_actionsc C s d| j j| jf S r� r� rJ r* r* r+ r� � s zDeleteAll.__repr__Nr� r* r* r* r+ rn � s rn c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ProcessState)r� rU r# r� c C s" || _ d|jf| _|| _|| _d S �Nr� )r� r� rU r# )rG r� r� rU r# r* r* r+ rH � s zProcessState.__init__c sj | j � | j�| j�� ��fdd�|D �}|�|� | jgdd� |D � }�rZ��||� n ��||� d S )Nc s. g | ]&}|j � kr|j�kr|j�kr|�qS r* )r� r� rU �rt �r��cls_r� rU r* r+ rw � s �z2ProcessState.execute_aggregate.<locals>.<listcomp>c S s g | ] }|j �qS r* �r# r� r* r* r+ rw � s )r� r� rU �difference_updater# r� r� )rG r� r� �our_recsrE r* r� r+ r� � s � z ProcessState.execute_aggregatec C s d| j j| jt�| j�| jf S )Nz%s(%s, %s, delete=%s))r� r� r� re � state_strr# rU rJ r* r* r+ r� � s �zProcessState.__repr__N)r� r� r� r� rH r� r� r* r* r* r+ r� � s r� c @ s2 e Zd ZdZdd� Ze�d�dd� �Zdd� Zd S ) r� �r# r r� c C s" || _ |jj| _d| jjf| _d S r� �r# r ri r� r� �rG r� r# r* r* r+ rH � s zSaveUpdateState.__init__r� c sV t jj}| j� | j�� �fdd�|D �}|�|� |��| jgdd� |D � |� d S )Nc s$ g | ] }|j � kr|j�kr|�qS r* �r� r r� �r� r r* r+ rw � s z5SaveUpdateState.execute_aggregate.<locals>.<listcomp>c S s g | ] }|j �qS r* r� r� r* r* r+ rw � s )r r� r� r� r r� r� r# )rG r� r� r� r� r* r� r+ r� � s � �z!SaveUpdateState.execute_aggregatec C s d| j jt�| j�f S r� �r� r� re r� r# rJ r* r* r+ r� � s �zSaveUpdateState.__repr__N� r� r� r� r� rH r r� r� r� r* r* r* r+ r� � s r� c @ s2 e Zd ZdZdd� Ze�d�dd� �Zdd� Zd S ) r� r� c C s" || _ |jj| _d| jjf| _d S )Nr� r� r� r* r* r+ rH s zDeleteState.__init__r� c sh t jj}| j� | j�� �fdd�|D �}|�|� | jgdd� |D � }|���fdd�|D ��� d S )Nc s$ g | ] }|j � kr|j�kr|�qS r* r� r� r� r* r+ rw s z1DeleteState.execute_aggregate.<locals>.<listcomp>c S s g | ] }|j �qS r* r� r� r* r* r+ rw s c s g | ]}� j | d r|�qS r� rQ rs )r� r* r+ rw s )r r� r� r� r r� r# r� )rG r� r� r� r� rE r* )r� r r� r+ r� s � �z DeleteState.execute_aggregatec C s d| j jt�| j�f S r� r� rJ r* r* r+ r� s �zDeleteState.__repr__Nr� r* r* r* r+ r� s r� N)(�__doc__� __future__r �typingr r r r r � r r rM r re r r � dependencyr Z interfacesr r r r r r r# r r: r; r� r] r� r� r� rm rn r� r� r� r* r* r* r+ �<module> sB r O2-%%$