관리-도구
편집 파일: persistence.cpython-38.pyc
U -?�f� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlZddlm Z dd lm Z dd lm Z dd lm Z dd lmZ d d lm Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ dDdd�Z dd� Z dd� Z d d � Z d d � Z d d!� Z!dddd"d#�d$d%�Z"ddd"d&�d'd(�Z#d)d*� Z$d+d,� Z%d-dd-d.�d/d0�Z&d-ddd1�d2d3�Z'd4d5� Z(d6d7� Z)d8d9� Z*d:d;� Z+d<d=� Z,d>d?� Z-d@dA� Z.dBdC� Z/dS )Ez�private module containing functions used to emit INSERT, UPDATE and DELETE statements on behalf of a :class:`_orm.Mapper` and its descending mappers. The functions here are called only by the unit of work functions in unitofwork.py. � )� annotations)�chain)�groupby)� zip_longestN� )� attributes)�exc)�loading)�sync)� state_str� )�future)�sql)�util)�cursor)� operators)�BooleanClauseList)� LABEL_STYLE_TABLENAME_PLUS_COLFc C s |s0| j s0t| |�D ]}t| |g|dd� qdS g }g }t| ||�D ]B\}}}} } } } | s^| rt|�|||| | f� qD|�|||| f� qD| j�� D ]J\} }| |jkr�q�t| |�}t || |�}t | ||| |� t | ||| |� q�t | |t dd� |D �dd� |D ��� dS )a; Issue ``INSERT`` and/or ``UPDATE`` statements for a list of objects. This is called within the context of a UOWTransaction during a flush operation, given a list of states to be flushed. The base mapper in an inheritance hierarchy handles the inserts/ updates for all descendant mappers. T)�singleNc s s$ | ] \}}}}||||d fV qdS )FN� )�.0�state� state_dict�mapper� connectionr r �J/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/persistence.py� <genexpr>i s �zsave_obj.<locals>.<genexpr>c s s& | ] \}}}}}||||d fV qdS )TNr )r r r r r �update_version_idr r r r m s �)�batch� _sort_states�save_obj�_organize_states_for_save�append�_sorted_tables�items� _pks_by_table�_collect_insert_commands�_collect_update_commands�_emit_update_statements�_emit_insert_statements� _finalize_insert_update_commandsr )� base_mapper�states�uowtransactionr r �states_to_update�states_to_insert�dict_r r � has_identity� row_switchr �table�insert�updater r r r ( sn � � �������r c sh t t| ||��}| j�� D ]H\�� �� jkr.q� �fdd�|D �}t| |�||�}t| |� �|� qdS )z\Issue UPDATE statements on behalf of a relationship() which specifies post_update. c 3 sH | ]@\}}}}�|j kr||||� jd k r:� �||� j�nd fV qd S �N)r% �version_id_col�#_get_committed_state_attr_by_column)r r r Z sub_mapperr �r r3 r r r � s �� ��z post_update.<locals>.<genexpr>N)�list� _organize_states_for_post_updater# r$ r% � _collect_post_update_commands� _emit_post_update_statements)r+ r, r- �post_update_colsr. r5 r r9 r � post_update{ s. � � ��r? c C s� t t| ||��}| j}tt |�� ��D ]F}|| }||jkr@q&n|jrN|jrNq&t| |||�}t | ||||� q&|D ] \}} }} } |j � || |� qrdS )z�Issue ``DELETE`` statements for a list of objects. This is called within the context of a UOWTransaction during a flush operation. N) r: �_organize_states_for_deleter# �reversed�keysr% ZinheritsZpassive_deletes�_collect_delete_commands�_emit_delete_statements�dispatchZ after_delete) r+ r, r- �states_to_deleteZtable_to_mapperr3 r �deleter r r r r r r � delete_obj� s> � ���rH c c sJ t | ||�D �]6\}}}}t|j�}|jp2|�|�}d } } |sR|j�|||� n|j�|||� |jrv|�|||� |s�||jj kr�|jj | } t � | �} |� | �s�|� | �s�t�dt|�|t| �f � n&| �d|t|�t| �� |�| � | } |s�| �r0|jdk �r0|�| �r| n|| �r&| jn||j�} |||||| | fV q dS )aT Make an initial pass across a set of states for INSERT or UPDATE. This includes splitting out into distinct lists for each, calling before_insert/before_update, obtaining key information for each state including its dictionary, mapper, the connection to use for the execution per state, and the identity flag. NzJNew instance %s with identity key %s conflicts with persistent instance %szPdetected row switch for identity %s. will update %s, remove %s from transaction)�_connections_for_states�bool�key�_identity_key_from_staterE Z before_insertZ before_updateZ _validate_polymorphic_identity�sessionZ identity_mapr Zinstance_stateZwas_already_deleted� is_deletedr �warnr Z _log_debugZremove_state_actionsr7 r8 �dict) r+ r, r- r r0 r r r1 Z instance_keyr2 r �instance�existingr r r r! � s` � � � ��� ��r! c C s t | ||�S )z�Make an initial pass across a set of states for UPDATE corresponding to post_update. This includes obtaining key information for each state including its dictionary, mapper, the connection to use for the execution per state. )rI )r+ r, r- r r r r; # s r; c c s^ t | ||�D ]L\}}}}|j�|||� |jdk rD|�|||j�}nd}|||||fV q dS )z�Make an initial pass across a set of states for DELETE. This includes calling out before_delete and obtaining key information for each state including its dictionary, mapper, the connection to use for the execution per state. N)rI rE Z before_deleter7 r8 )r+ r, r- r r0 r r r r r r r@ / s � �r@ r )�bulk�return_defaults� render_nulls�include_bulk_keysc # s� |D �]�\}� }}| |j kr qi } i } |j| } |j| } t| ��� �D ]l} � | }| | }|dkrt|| krt|stqHqH|s�t|d�s�t|tj�r�t|d�r�|� � n|| |< qH|| |j < qH|s�|j | � | �� dd� | D ��D ] }d| |< q�|r�|�r,|j | �| �}|j�|j| ��r&|j| �| �}nd}nd }}|jdk �rp|jdk �rp|j|j| k�rp|�d�| |jj < |�r�|j�r�| �|j|j� |�r�| �� fdd�|D �� |� | ||| ||fV qdS ) zSIdentify sets of values to use in INSERT statements for a list of states. N�__clause_element__c S s g | ] }|j �qS r �rK �r �cr r r � <listcomp>y s z,_collect_insert_commands.<locals>.<listcomp>TFc 3 s | ]}|� | fV qd S r6 r �r �k�r r r r � s z+_collect_insert_commands.<locals>.<genexpr>)r% �_propkey_to_colZ _insert_cols_evaluating_none�set� intersection�hasattr� isinstancer � ClauseElementrW rK Z_insert_cols_as_none� differenceZ_pk_keys_by_table�issubsetr+ �_prefer_eager_defaults�dialectZ_server_default_col_keys�version_id_generatorr7 �_cols_by_tableZ_set_polymorphic_identity� setdefaultZ_polymorphic_attr_keyZpolymorphic_identityr5 )r3 r/ rS rT rU rV r r r �params� value_params�propkey_to_colZ eval_none�propkey�value�colZcolkey� has_all_pks�has_all_defaultsr r^ r r& F s� � �� � � �� ������� ��r& )rS �use_orm_update_stmtrV c # s� |D �]�\}�}}} ||j kr q|j | } |dk r:|j} ni } |j| �|r|��fdd�t������|j| �D �} d} n�i } t���|j�D ]n}�| }�| }t|d�s�t |t j �r�t|d�r�|� � n|| |< q�|j | j�||j| �dk r�|| |j< q�|jjdk�r |j| �| �} nd} | dk �r�|j|j| k�r�|�s�| �s�| �s�|j�� D ].}|j |j j�|�tj�}|j �r\ �q��q\q|j}| �o�| }| | |j < |�s�|j| k�r�|j dk �r�|� | �}|| |j< n |j dk�r |�r | | |j< n| �s | �s qd}d}|�r|��fdd�t���|j| �D �� t j �� �� ��rHt!j"dd �#� fd d �| D ��� d �d d��n�i � | D ]�}|j| j}|j | j�|�tj$�}|j �r|j%�r�d||f| jk�r�d}|j d � |j < | �&|jd� n |j%d � |j < || k�r"d}n|j'd � |j < � |j dk�r�t(�)d||f ���q�|�rf| �*�fdd �|D �� | �sr| �r�| �*� � |�| ||| | |fV q|r|j+| D ]"\}}t,�-|||||| |j.� �q�qdS )at Identify sets of values to use in UPDATE statements for a list of states. This function works intricately with the history system to determine exactly what values should be updated as well as how the row should be matched within an UPDATE statement. Includes some tricky scenarios where the primary key of an object might have been changed. Nc s i | ]}� | j �| �qS r rX �r ro �rn r r r � <dictcomp>� s � z,_collect_update_commands.<locals>.<dictcomp>TrW Fc s i | ]}� | j ��|��qS r )�_label�getru rv r r rw ) s � z,No primary key value supplied for column(s) z, c 3 s$ | ] }� |j d krt|�V qd S r6 )rx �strrY )� pk_paramsr r r 2 s z+_collect_update_commands.<locals>.<genexpr>zY; per-row ORM Bulk UPDATE by Primary Key requires that records contain primary key valuesZbupq)�codeZ pk_cascadedr zCCan't update table %s using NULL for primary key value on column %sc 3 s | ]}|� | fV qd S r6 r r\ r^ r r r \ s )/r% Z_valuesr_ r` ra re Z_pk_attr_keys_by_table�committed_staterb rc r rd rW �manager�implZis_equalrK r+ �eager_defaultsZ!_server_onupdate_default_col_keysrf r7 rj �_columntoproperty�values� get_historyr �PASSIVE_NO_INITIALIZE�addedrx ri r ZNONE_SET�sa_exc�InvalidRequestError�join� PASSIVE_OFFZdeleted�popZ unchanged�orm_exc� FlushErrorr5 �_table_to_equatedr �populate�passive_updates)r- r3 r. rS rt rV r r r r �pksrm rl rs ro rp rq �prop�historyZ no_params�valrr Zexpect_pk_cascaded�m� equated_pairsr )r{ rn r r r' � s* � ���� �� � ��� ���� � ��� ����� ���� �� � �r' c c s, |D �] \}}}}} |j | } i } d} |j| D ]|} | | krX|j||| tjd�| | j< q0| |ksj| jdk r0|j| }|j|j j � ||tj �}|j r0|j d }|| | j < d} q0| r| dk �r|j|j| k�r|j} | | | j< t|j ��r| j | k�r|jdk �r|�| �}|| | j < ||||| fV qdS )zrIdentify sets of values to use in UPDATE statements for a list of states within a post_update operation. F)ZpassiveNr T)r% rj Z_get_state_attr_by_columnr r� rx Zonupdater� r~ rK r r� r� r� r7 rJ ri )r+ r- r3 r. r> r r r r r r� rl Zhasdatarq r� r� rp r� r r r r< | sZ � � � �� ��� r< c c s� |D ]�\}}}}}||j kr qi } |j | D ]6} |�||| � | | j< } | dkr,t�d|| f ��q,|dk r�|j|j| kr�|| |jj< | |fV qdS )zSIdentify values to use in DELETE statements for a list of states to be deleted.NzHCan't delete from table %s using NULL for primary key value on column %s)r% r8 rK r� r� r7 rj ) r+ r- r3 rF r r r r r rl rq rp r r r rC � s: � �� ���� rC T)� bookkeepingrt �enable_check_rowcountc s � j dk o� j � j� k�d| ji}d� ��fdd� } |dk rH| |�} n| �d�f| �} t|dd� �D �]�\\} } } }}}d}t|�}| }|dk r�|��� d ��}d }|s�|j� j� � }d }|r�|s� j j d kr�j r�| j j r�|j� j� � }d }� j�r |�� j �}d }| j j}|�o | j j}| �o.� }| �r�|D ]j\}}}� } }}}| j|�|�||d �}|�r�t� |�||||jjd |d |j� ||j7 }|�o�|} �q:�n |�s |�o�|} |D ]Z\}}}� } }}}| j|||d �}|�r t� |�||||jjd |d |j� ||j7 }�q�n�d d� |D �} |�oF|�pF|�oFt| �dk} | j|| |d �}||j7 }|D ]N\}}}� } }}}|�rft� |�||||jjd |d |jj�s�|jnd� �qf| �r�|t|�k�r�t� d�j t|�|f ��qf�rft � d|j j � qfdS )z`Emit UPDATE statements corresponding to value lists collected by _collect_update_commands().N�compiled_cachec s� t �tj�}� j� D ] }|�|tj|j|j d�k� q�r^|�� j tj� j j� j j d�k� | d k rr| � |�}n�� � � |�}|S �N)�type_) r �_construct_rawr �and_r% �_append_inplacer � bindparamrx �typer7 �wherer5 )Z existing_stmt�clausesrq �stmt�r �needs_version_idr3 r r � update_stmt� s$ ���� z,_emit_update_statements.<locals>.update_stmtr5 c S s* | d t | d �t| d �| d | d fS �N� r � � � �r` rJ ��recr r r �<lambda> s �z)_emit_update_statements.<locals>.<lambda>r )Z_emit_update_tableZ_emit_update_mapperFT��execution_optionsc S s g | ] }|d �qS �r r �r r� r r r r[ � s z+_emit_update_statements.<locals>.<listcomp>r �MUPDATE statement on table '%s' expected to update %d row(s); %d were matched.�MDialect %s does not support updated rowcount - versioning cannot be verified.)N)!r7 rj �_compiled_cache�_memor r: � _annotaterT r% r+ r� �implicit_returningrh Zupdate_returningZ _server_onupdate_default_cols�!_version_id_has_server_side_value�supports_sane_rowcount� supports_sane_multi_rowcount�executer� � _postfetch�context�compiled_parameters�returned_defaults�rowcount�len� executemanyr� �StaleDataError� descriptionr rO �dialect_description) r+ r- r r3 r5 r� rt r� r� r� � cached_stmtr Z paramkeys�hasvaluers rr �records�rows� statementrT �assert_singlerow�assert_multirow�allow_executemanyr r rl rm rZ �check_rowcount� multiparamsr r� r r( � sB � �� ���� ��� � � �� � � � � � � � � �����r( )r� �use_orm_insert_stmtr� c % C s� |dk r,|}t j} t|j�} | r&|jp(|} n$d} |} | �d|f|j�}d| ji} |rdt j�| |�}n| }d} t |dd� �D �]\\} }}}}}|}|dk r�|� ||d��}|r�|s�| � | j |�r�|j r�| j j�sx| �sx|�rx|�sxt|�}dd � |D �}| j|||d �}|�r�t||jj�D ]X\\}}}}}}}}} |�rft||||||| |d|jj�s^|jnd� n t|||� �q qzt|�}| �s�|j �r|�st|�d k�r| �r�| j j�s�| �s�| j j�r�d } n8| �r�| �r�d } nd} t� d| j j � d| � d���nd} nd} |dk�r<|�s<| � | j |��r<|j |j | d|i�}|j dk �rZ|j |j |d�}n| �rr|j |j!d|i�}| �r�dd � |D �}| j|||d �}|dk �r�| dk�r�|} n | �"|�} |�r�t#||jj|j$|j%�p�d�D ]�\\}}}}}}}}} } } | dk�rt&�'d| ��t| |j(| �D ]0\}!}"|j)|" }#|�*|#j+�dk�r |!||#j+< �q |�rtt||||||| |d| � n t|||� �q�qz| �r�t,�|D ]�\}}}}} }}}|�r�| j|�-|�||d �}n| j|||d �}|j.}$|$dk�r�t&�'d|f ��t|$|j(| �D ]:\}!}"|j)|" }#|"|k�s4|�*|#j+�dk�r|!||#j+< �q|�r�|�r�t|||||||jjd |d|jj�sz|jnd� n t|||� �q�qz|dk �r�| dk�r�t/�0� S | S dS )z`Emit INSERT statements corresponding to value lists collected by _collect_insert_commands().NFr4 r� c S s* | d t | d �t| d �| d | d fS r� r� r� r r r r� � s �z)_emit_insert_statements.<locals>.<lambda>)Z_emit_insert_tableZ_emit_insert_mapperc S s g | ] }|d �qS r� r r� r r r r[ s z+_emit_insert_statements.<locals>.<listcomp>r� r Tz+ with RETURNING and sort by parameter orderz with RETURNINGz<Can't use explicit RETURNING for bulk INSERT operation with z backend; executemanyz! is not enabled for this dialect.�sort_by_parameter_order)r� c S s g | ] }|d �qS r� r r� r r r r[ u s r z�Multi-row INSERT statement for %s did not produce the correct number of INSERTed rows for RETURNING. Ensure there are no triggers or special driver issues preventing INSERT from functioning properly.z�Single-row INSERT statement for %s did not produce a new primary key result being invoked. Ensure there are no triggers or special driver issues preventing INSERT from functioning properly.r )1r � EMPTY_DICTrJ Z _returningZ_sort_by_parameter_orderr� r4 r� Z merge_withr r� rg rh r� Zinsert_returningr: r� �zipr� r� r� r� r� �_postfetch_bulk_saver� Z4insert_executemany_returning_sort_by_parameter_orderZ insert_executemany_returningr� r� r� rT Z_server_default_colsr7 � primary_keyZsplice_verticallyr Zinserted_primary_key_rowsZreturned_defaults_rowsr� r� r% r� ry rK �AssertionErrorr� �inserted_primary_key�_cursorZnull_dml_result)%r+ r- r r3 r4 r� r� r� r� Zexec_optZ returning_is_required_anywayZdeterministic_results_reqdZ return_resultr �_r� rr rs r� r� r� �resultr r rl � mapper_rec�connrm Zlast_inserted_paramsZdo_executemany�dtr� r� �pkrq r� r� r r r r) � s �� ��� �� �� ��� � � �� � �� ��� ������ � ���� ��� � �� �� � � ��� �� � �� � � r) c s� d| j i}� jdk o"� j� j� k�� ��fdd�}| �d�f|�}� jrV|�� j�}t|dd� �D �]F\}} d} t| �} |d } | jj } | o�| jj } � p�| }|s�| }| D ]D\}}}} }| j |||d �}t ||�||||j jd � | |j7 } q�nrd d � | D �}| �p| �ot|�d k}| j |||d �}| |j7 } | D ],\}}}} }t ||�||||j jd � �q8|�r�| t| �k�r�t�d �jt| �| f ��qd�rdt�d|jj � qddS )zeEmit UPDATE statements corresponding to value lists collected by _collect_post_update_commands().r� Nc sp t �tj�} � j� D ] }| �|tj|j|j d�k� q�r^| �� j tj� j j� j j d�k� �� � � | �}|S r� ) r r� r r� r% r� r r� rx r� r7 r5 r� )r� rq r� r� r r r� s ����z1_emit_post_update_statements.<locals>.update_stmtr? c S s | d t | d �fS )N� r� )r` r� r r r r� 3 � z._emit_post_update_statements.<locals>.<lambda>r r� c S s g | ]\}}}}}|�qS r r )r r r r� r� rl r r r r[ S s �z0_emit_post_update_statements.<locals>.<listcomp>r r� r� )r� r7 rj r� r� rT r r: rh r� r� r� �_postfetch_post_updater� r� r� r� r� r� r� r rO r� )r+ r- r r3 r5 r� r� r� rK r� r� r r� r� r� r� r r r� rl rZ r� r r� r r= s� � �� � � �� � � ����r= c s` � j dk o� j � j� k�� ��fdd�}| �d�f|�}t|dd� �D �]\}}dd� |D �} d | ji} t| �} d } d } �r�|jjs�|jjr�d } | D ] }|j ||| d �}| |j 7 } q�q�t � d|jj � |j || | d � n |j || | d �}�s�d} |j } | jrH| d krH| | krH|jj�s&t| �dkrH| �rDt � d�j| | f � qHt�d�j| | f ��qHdS )z`Emit DELETE statements corresponding to value lists collected by _collect_delete_commands().Nc sl t �tj�} � j� D ] }| �|tj|j|j d�k� q�r^| �� j tj� j j� j j d�k� �� � � | �S r� ) r r� r r� r% r� r r� rK r� r7 rG r� )r� rq �r Zneed_version_idr3 r r � delete_stmt� s � ���z,_emit_delete_statements.<locals>.delete_stmtrG c S s | d S )Nr r r� r r r r� � r� z)_emit_delete_statements.<locals>.<lambda>c S s g | ] \}}|�qS r r )r rl r r r r r[ � s z+_emit_delete_statements.<locals>.<listcomp>r� ���Fr r� zMDialect %s does not support deleted rowcount - versioning cannot be verified.Tr z�DELETE statement on table '%s' expected to delete %d row(s); %d were matched. Please set confirm_deleted_rows=False within the mapper configuration to prevent this warning.)r7 rj r� r r� r� rh r� r� r� r� r rO r� Zconfirm_deleted_rowsr� r� r� )r+ r- r r3 rG r� r� r ZrecsZ del_objectsr� �expectedZ rows_matchedZ only_warnrl rZ r r� r rD | s~ � �� ��� � ����� � �� ��rD c s, |D �] \� }}}}|j rF� �� fdd�|j D ��}|rF� �� j|� g }| jdkrh|�� j�|j�� |j dk r�|j dkr�|j j � j kr�|�|j j g� |r�| �� �� _ t�|��t�} tj|j| � j � |d� |s�|j�||� � n|j�||� � |j dkr|j dk r||j j dkrt�d��qdS )zzfinalize state on states that have been inserted or updated, including calling after_insert/after_update events. c sB g | ]:}|j r |jr8|j� jks8|j s|js|j� jkr|j�qS r )Zexpire_on_flush�deferredrK rP )r �p�r r r r[ � s �z4_finalize_insert_update_commands.<locals>.<listcomp>TNF)Z refresh_stateZonly_load_propsz2Instance does not contain a non-NULL version value)Z_readonly_propsZunmodified_intersection�_expire_attributesrP r� �extendZ_unloaded_non_objectra Z&_server_default_plus_onupdate_propkeysr7 ri Z_version_id_proprK ZunloadedrL r �selectZset_label_styler r Z load_on_identrM rE Z after_insertZ after_updater� r� ) r+ r- r, r r r r1 �readonlyZ toload_nowr� r r� r r* � sZ �� ���� �� ���r* c s� � j d k o� j � j| k}|�|�s:|jjj}|jjj} n|rHd }} nd S |r`t|�� j g }t� j j j �} | rvg } |D ]D} | j |krz| � j krz|| j |� j | j < | rz| �� j | j � qz| r�| rڈ j j � ||| � | r�|�|j� fdd�| D �� d S )Nr c s" g | ]}|� j kr� j | j�qS r �r� rK rY �r r r r[ T s �z*_postfetch_post_update.<locals>.<listcomp>)r7 rj rN r� �compiled�prefetch� postfetchr: rJ � class_managerrE � refresh_flushrK r� r" r� rP ) r r- r3 r r0 r� rl r� � prefetch_cols�postfetch_colsr� �load_evt_attrsrZ r r� r r� * s@ � � ��r� c s� |j jj} |j jj} |j jj�� jdk rH� j� j| krHt| �� jg } t� j j j �} | r^g } �r�| }|dk r�t |��D ]B\}}|j r�|j jr�qx� j�|�}|rx|||j< | rx| �|j� qx| D ]N}|j|kr�|� jkr�� j| j}||j ||< |j�|d� | r�| �|� q�| �r.| �r.� j j � ||| � |�rR|�rR| ��fdd�|D �� | �rt|�|j� fdd�| D �� � j| D ]"\}}t�||||||� j� �q~dS )z�Expire attributes in need of newly persisted database state, after an INSERT or UPDATE statement has proceeded for that state.Nc s g | ]}|j r|� kr|�qS r )r� )r rq )�returning_colsr r r[ � s �z _postfetch.<locals>.<listcomp>c s" g | ]}|� j kr� j | j�qS r r� rY r� r r r[ � s �) r� r� r� r� Zeffective_returningr7 rj r: rJ r� rE r� r� r� Zisinsertr� ry rK r" r} r� r� r� rP r� r r� r� )r r- r3 r r0 r� rl rm Zisupdater� r� r� r� r� �rowZ row_valuerq r� rZ �pkeyr� r� r )r r� r r� \ st �� � �� �� �r� c C s& | j | D ]\}}t�|||� q d S r6 )r� r Zbulk_populate_inherit_keys)r r0 r3 r� r� r r r r� � s r� c c s` |j jr|j j}n|j�| �}d}t| |�D ].}|rB|| |�� �}|jj}||j||fV q,dS )z�Return an iterator of (state, state.dict, mapper, connection). The states are sorted according to _sort_states, then paired with the connection they should be using for the given unit of work transaction. N) rM �connection_callableZ transactionr r �objr~ r rP )r+ r- r, r� r r r r r r rI � s rI c C s| t |�}dd� |D �}|�|� zt|| jd�}W n2 tk rd } zt�d| �|�W 5 d }~X Y nX t|t�d�d�| S )Nc S s h | ]}|j d k r|�qS r6 rX )r �sr r r � <setcomp>� s z _sort_states.<locals>.<setcomp>rX z^Could not sort objects by primary key; primary key values must be sortable in Python (was: %s)Z insert_order) r` �difference_update�sortedZ_persistent_sortkey_fn� TypeErrorr� r� �operator� attrgetter)r r, �pending� persistentZpersistent_sorted�errr r r r � s( � �����r )F)0�__doc__� __future__r � itertoolsr r r r � r r r� r r �baser r� r r r Zenginer r� r Z sql.elementsr Zsql.selectabler r r? rH r! r; r@ r& r' r< rC r( r) r= rD r* r� r� r� rI r r r r r �<module> sn S-+P � j� R6-� r� FvcK2j