Edit File: unitofwork.cpython-37.pyc
B ��4]�` � @ s d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z d d � Z G d d � d e �Z G d d� de �Z G dd� de �ZG dd� de �ZG dd� de e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�ZG d d � d e�Zd!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. � )� attributes)�exc)� persistence)�util� )�event)� topologicalc sj |j � � fdd�}� fdd�}� fdd�}tj| d|ddd � tj| d |ddd � tj| d |ddd � d S ) z\Establish event listeners on object attributes which handle cascade-on-set/append. c sn |d kr d S | j }|rj|jr&|�d� | jjj� }t�|�}|jj rj|j sV� |j krj|� |�sj|� |� |S )Nzcollection append)�session�_warn_on_events�_flush_warning�manager�mapper�_propsr �instance_state�_cascade� save_update�cascade_backrefs�key�_contains_state�_save_or_update_state)�state�item� initiator�sess�prop� item_state)r � �L/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/orm/unitofwork.py�append 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_session)r r r r r r )r r r �remove6 s" z$track_cascade_events.<locals>.removec s� ||kr |S | j }|r�|jr&|�d� | jjj� }|d k rrt�|�}|jj rr|j s^� |j krr|� |�sr|� |� |d k r�|tjk 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 r! r# r" r$ )r ZnewvalueZoldvaluer r r Znewvalue_stateZoldvalue_state)r r r �set_X s, z"track_cascade_events.<locals>.set_r T)�raw�retvalr% �setN)r r �listen)Z descriptorr r r% r& r )r r �track_cascade_events s " %r+ c @ s� e Zd Zdd� Zedd� �Zdd� Zdd� Zd d � Zd d � Z e j fd d�Z dd� Z dd� Zd'dd�Zdd� Zdd� Zejdd � �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� ZdS )(�UOWTransactionc C sR || _ i | _t�t�| _t�t�| _i | _i | _t� | _ i | _ t�dd� �| _ d S )Nc S s t � t � fS )N)r) r r r r �<lambda>� � z)UOWTransaction.__init__.<locals>.<lambda>) r r r � defaultdictr) �deps�mappers�presort_actions�postsort_actions� dependencies�states�post_update_states)�selfr r r r �__init__� s zUOWTransaction.__init__c C s t | j�S )N)�boolr5 )r7 r r r �has_work� s zUOWTransaction.has_workc C sB |j r>y|�|tj� W n$ tjk r< | j�|g� dS X dS )zVreturn true if the given state is expired and was deleted previously. TF)ZexpiredZ _load_expiredr � PASSIVE_OFF�orm_excZObjectDeletedErrorr �_remove_newly_deleted)r7 r r r r �was_already_deleted� s z"UOWTransaction.was_already_deletedc C s || j ko| j | d S )zWreturn true if the given state is marked as deleted within this uowtransaction.� )r5 )r7 r r r r � is_deleted� s zUOWTransaction.is_deletedc C s, || j kr| j | S |� | j |< }|S d S )N)r )r7 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? TN)r5 )r7 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 �}|rp|jrp|� � }n|}|||f| j |< nJ|j| j}|�||j|t jB �}|r�|jr�|� � }n|}|||f| j |< |S )zOfacade to attributes.get_state_history(), including caching of results.�history) r ZSQL_OKr �implZ get_history�dictr; ZLOAD_AGAINST_COMMITTEDZ uses_objectsZas_state) r7 r r ZpassiveZhashkeyrE Z state_historyZcached_passiverF r r r �get_attribute_history� s, z$UOWTransaction.get_attribute_historyc C s |df| j kS )NT)r2 )r7 � processorr r r �has_dep s zUOWTransaction.has_depc C s& ||f}|| j kr"t||�| j |< d S )N)r2 � Preprocess)r7 rI � fromparentr r r r �register_preprocessor s z$UOWTransaction.register_preprocessorFNc 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_strr5 r r r1 �_per_mapper_flush_actions�add)r7 r rC �listonlyZ cancel_deleteZ operationr r r r r �register_object s z UOWTransaction.register_objectc C s0 |j jj}| j| \}}|�|� |�|� d S )N)r r � base_mapperr6 rQ �update)r7 r Zpost_update_colsr r5 �colsr r r �register_post_update- s z#UOWTransaction.register_post_updatec C sn t | |j�}t| |j�}| j�||f� x|jD ]}|�| � q0W x&|jD ] }|jrVqJ|j }|�| � qJW d S )N) � SaveUpdateAllrT � DeleteAllr4 rQ Z_dependency_processorsZper_property_preprocessorsZ relationshipsZviewonlyZ_dependency_processor)r7 r ZsavesZdeletes�depr r r r rP 3 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- L r. z0UOWTransaction._mapper_for_dep.<locals>.<lambda>)r Z PopulateDict)r7 r r r �_mapper_for_depA 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)rZ �mapper_for_depr r � <listcomp>U s z8UOWTransaction.filter_states_for_dep.<locals>.<listcomp>)r] )r7 rZ r5 r )rZ r` r �filter_states_for_depO s z$UOWTransaction.filter_states_for_depc c sF ||f}x8|j jD ],}x&| j| D ]}| j| |kr"|V q"W qW d S )N)rT �self_and_descendantsr1 r5 )r7 r rC rR Zchecktupr r r r �states_for_mapper_hierarchyW s z*UOWTransaction.states_for_mapper_hierarchyc s^ x4d}x$t � j�� �D ]}|�� �rd}qW |sP qW t�� jt � j�� �� � _}|�r@t � fdd�|D ��}x�t � j�D ]�}d|ks�|d j s�|d j s�|� |�r�� j� |� q||d |kr�� j� |� xn||d D ]}� j� ||d f� q�W q||d |kr|� j� |� x*||d D ]}� j� |d |f� �q W q|W tdd � � j�� D ���|�S ) z}Generate the full, unsorted collection of PostSortRecs as well as dependency pairs for this UOWTransaction. FTc 3 s | ]}|t |�� ��fV qd S )N)r) �per_state_flush_actions)r^ �rec)r7 r r � <genexpr>w s z3UOWTransaction._generate_actions.<locals>.<genexpr>Nr? r c S s g | ]}|j s|�qS r )�disabled)r^ �ar r r ra � s z4UOWTransaction._generate_actions.<locals>.<listcomp>)�listr2 �values�executer Z find_cyclesr4 r3 �cyclesrG rh � issupersetr% rQ r) � difference)r7 rA �actionrm �convertZedgerZ r )r7 r �_generate_actions^ s: z UOWTransaction._generate_actionsc C sl | � � }| jrFxXt�| j|�D ]"}x |r>|�� }|�| |� q$W q W n"x t�| j|�D ]}|�| � qVW d S )N) rr rm r Zsort_as_subsetsr4 �pop�execute_aggregate�sortrl )r7 r3 r& �nrf r r r rl � s zUOWTransaction.executec C sZ | j s dS t| j �}tdd� | j �� D ��}|�|�}|rF| j�|� |rV| 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 | ]\}\}}|r|V qd S )Nr )r^ r_ rC rR r r r rg � s z8UOWTransaction.finalize_flush_changes.<locals>.<genexpr>)r5 r) �itemsro r r= Z_register_persistent)r7 r5 Zisdel�otherr r r �finalize_flush_changes� s z%UOWTransaction.finalize_flush_changes)FFFNN)�__name__� __module__� __qualname__r8 �propertyr: r> r@ rB rD r ZPASSIVE_NO_INITIALIZErH rJ rM rS rW rP r Zmemoized_propertyr] rb rd rr rl ry r r r r r, � s, - . 4r, c @ s e Zd Zdd� ZdS )�IterateMappersMixinc s2 � j r$t� �fdd�� jjjD ��S � jjjS d S )Nc 3 s" | ]}�j |� jf r|V qd S )N)r] �dependency_processor)r^ �m)r7 �uowr r rg � s z/IterateMappersMixin._mappers.<locals>.<genexpr>)rL �iterr �parentrc r )r7 r� r )r7 r� r �_mappers� s z IterateMappersMixin._mappersN)rz r{ r| r� r r r r r~ � s r~ c @ s e Zd ZdZdd� Zdd� ZdS )rK )r rL � processed�setup_flush_actionsc C s || _ || _t� | _d| _d S )NF)r rL r) r� r� )r7 r rL r r r r8 � s zPreprocess.__init__c C s� t � }t � }xZ| �|�D ]L}xF|j| �| j�D ]0}|j| \}}|s0|rV|�|� q0|�|� q0W qW |r�| j�||� | j� |� |r�| j� ||� | j� |� |s�|r�| j s�| j� ||d�s�| j� ||d�r�| j� |� d| _ dS dS d S )NTF)r) r� r1 ro r� r5 rQ r Zpresort_deletesrU Z presort_savesr� Zprop_has_changesZper_property_flush_actions)r7 r� Z delete_statesZ save_statesr r rC rR r r r rl � s2 zPreprocess.executeN)rz r{ r| � __slots__r8 rl r r r r rK � s rK c @ s e Zd ZdZdd� Zdd� ZdS )� PostSortRec)rh c G s@ | f| }||j kr |j | S t�| � |j |< }d|_|S d S )NF)r3 �object�__new__rh )�clsr� �argsr rA r r r r� � s zPostSortRec.__new__c C s | � |� d S )N)rl )r7 r� �recsr r r rt s z PostSortRec.execute_aggregateN)rz r{ r| r� r� rt r r r r r� � s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � ProcessAll)r rC rL c C s* || _ || _|| _|j|jj �|� d S )N)r rC rL r0 r� rT rQ )r7 r� r rC rL r r r r8 s zProcessAll.__init__c C s2 | � |�}| jr | j�||� n| j�||� d S )N)� _elementsrC r �process_deletes� process_saves)r7 r� r5 r r r rl s zProcessAll.executec C s t g �S )N)r� )r7 r� r r r re s z"ProcessAll.per_state_flush_actionsc C s d| j j| j| jf S )Nz%s(%s, isdelete=%s))� __class__rz r rC )r7 r r r �__repr__% s zProcessAll.__repr__c c sN xH| � |�D ]:}x4|j| D ]&}|j| \}}|| jkr |s |V q W q W d S )N)r� r1 r5 rC )r7 r� r r rC rR r r r r� , s zProcessAll._elementsN) rz r{ r| r� r8 rl re r� r� r r r r r� s r� c @ s e Zd ZdZdd� Zdd� ZdS )� PostUpdateAll)r rC c C s || _ || _d S )N)r rC )r7 r� r rC r r r r8 7 s zPostUpdateAll.__init__c s: �j � j \}}� �fdd�|D �}t�� j|�|� d S )Nc s$ g | ] }�j | d � jkr|�qS )r? )r5 rC )r^ r_ )r7 r� r r ra = s z)PostUpdateAll.execute.<locals>.<listcomp>)r6 r r Z post_update)r7 r� r5 rV r )r7 r� r rl ; s zPostUpdateAll.executeN)rz r{ r| r� r8 rl r r r r r� 4 s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) rX )r c C s || _ ||jkst�d S )N)r rT �AssertionError)r7 r� r r r r r8 E s zSaveUpdateAll.__init__c C s t �| j|�| jdd�|� d S )NF)r �save_objr rd )r7 r� r r r rl I s zSaveUpdateAll.executec c s� t |�| jdd��}| jj}t||�}x,|D ]$}t||�}|j�||f� |V q,W x.|j| j D ] }|� ||�}|� ||d� qbW d S )NF) rj rd r rT rY �SaveUpdateStater4 rQ r0 rb re ) r7 r� r5 rT Z delete_allr rp rZ �states_for_propr r r re P s z%SaveUpdateAll.per_state_flush_actionsc C s d| j j| jf S )Nz%s(%s))r� rz r )r7 r r r r� a s zSaveUpdateAll.__repr__N)rz r{ r| r� r8 rl re r� r r r r rX B s rX c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) rY )r c C s || _ ||jkst�d S )N)r rT r� )r7 r� r r r r r8 h s zDeleteAll.__init__c C s t �| j|�| jdd�|� d S )NTF)r � delete_objr rd )r7 r� r r r rl l s zDeleteAll.executec c s� t |�| jdd��}| jj}t||�}x,|D ]$}t||�}|j�||f� |V q,W x.|j| j D ] }|� ||�}|� ||d� qbW d S )NTF) rj rd r rT rX � DeleteStater4 rQ r0 rb re ) r7 r� r5 rT Zsave_allr rp rZ r� r r r re s s z!DeleteAll.per_state_flush_actionsc C s d| j j| jf S )Nz%s(%s))r� rz r )r7 r r r r� � s zDeleteAll.__repr__N)rz r{ r| r� r8 rl re r� r r r r rY e s rY c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ProcessState)r rC r c C s || _ || _|| _d S )N)r rC r )r7 r� r rC r r r r r8 � 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 rC )r^ �r)�cls_r rC r r ra � s z2ProcessState.execute_aggregate.<locals>.<listcomp>c S s g | ] }|j �qS r )r )r^ r� r r r ra � s )r� r rC �difference_updater r� r� )r7 r� r� �our_recsr5 r )r� r rC r rt � s z ProcessState.execute_aggregatec C s d| j j| jt�| j�| jf S )Nz%s(%s, %s, delete=%s))r� rz r rO � state_strr rC )r7 r r r r� � s zProcessState.__repr__N)rz r{ r| r� r8 rt r� r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r� )r r c C s || _ |jj| _d S )N)r r rT )r7 r� r r r r r8 � s zSaveUpdateState.__init__c sN | j � | j�� �fdd�|D �}|�|� t��| jgdd� |D � |� d S )Nc s$ g | ] }|j � kr|j�kr|�qS r )r� r )r^ r� )r� r r r ra � s z5SaveUpdateState.execute_aggregate.<locals>.<listcomp>c S s g | ] }|j �qS r )r )r^ r� r r r ra � s )r� r r� r r� r )r7 r� r� r� r )r� r r rt � s z!SaveUpdateState.execute_aggregatec C s d| j jt�| j�f S )Nz%s(%s))r� rz rO r� r )r7 r r r r� � s zSaveUpdateState.__repr__N)rz r{ r| r� r8 rt r� r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r� )r r c C s || _ |jj| _d S )N)r r rT )r7 r� r r r r r8 � s zDeleteState.__init__c s` | j � | j�� �fdd�|D �}|�|� | jgdd� |D � }t���fdd�|D ��� d S )Nc s$ g | ] }|j � kr|j�kr|�qS r )r� r )r^ r� )r� r r r ra � s z1DeleteState.execute_aggregate.<locals>.<listcomp>c S s g | ] }|j �qS r )r )r^ r� r r r ra � s c s g | ]}� j | d r|�qS )r? )r5 )r^ r_ )r� r r ra � s )r� r r� r r r� )r7 r� r� r� r5 r )r� r r� r rt � s z DeleteState.execute_aggregatec C s d| j jt�| j�f S )Nz%s(%s))r� rz rO r� r )r7 r r r r� � s zDeleteState.__repr__N)rz r{ r| r� r8 rt r� r r r r r� � s r� N)�__doc__� r r r< r r rO r r r+ r� r, r~ rK r� r� r� rX rY r� r� r� r r r r �<module> s* i > 2(###