Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
local
/
lib
/
python3.10
/
dist-packages
/
sqlalchemy
/
orm
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: persistence.cpython-310.pyc
o [�h� � @ s d Z ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlmZ dd lm Z dd lm Z dd l m Z dd l mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ dd� Zdd� ZdTdd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Z dUd#d$�Z dTd%d&�Z d'd(� Z d)d*� Z +dVd,d-�Z! +dVd.d/�Z"d0d1� Z#d2d3� Z$d4d5� Z%d6d7� Z&d8d9� Z'd:d;� Z(d<d=� Z)d>d?� Z*d@dA� Z+G dBdC� dCe,�Z-G dDdE� dEe-�Z.G dFdG� dGe-�Z/G dHdI� dIe-�Z0G dJdK� dKe-�Z1G dLdM� dMe.e0�Z2G dNdO� dOe.e1�Z3G dPdQ� dQe/e0�Z4G dRdS� dSe/e1�Z5dS )Wz�private module containing functions used to emit INSERT, UPDATE and DELETE statements on behalf of a :class:`.Mapper` and its descending mappers. The functions here are called only by the unit of work functions in unitofwork.py. � )�chain)�groupbyN� )� attributes)� evaluator)�exc)�loading)�sync)�_entity_descriptor)� state_str� )�sql)�util)� expression)� _from_objectsc s" �j }t|�}|jjrtd��|r*|r"dd� |D �}dd� |D �}n dd� |D �}nt|�}|�|�� |j�� D ].\} } �� | �sBq8� �fdd�t | � �fdd�|D �d ||d �D �} t |d || | | |d � q8|r�|r��j } d d� �j D �} |D ]\}�| t�fd d�| D ��f|_qxd S d S d S )NzJconnection_callable / per-instance sharding not supported in bulk_insert()c S s g | ]}||j f�qS � ��dict��.0�stater r �E/usr/local/lib/python3.10/dist-packages/sqlalchemy/orm/persistence.py� <listcomp>7 � z _bulk_insert.<locals>.<listcomp>c S s g | ]\}}|�qS r r )r r �dict_r r r r 8 � c S � g | ]}|j �qS r r r r r r r : � c 3 s4 � | ]\}}}}}}}}d ||�� |||fV qd S �Nr ) r r � state_dict�params�mp�conn� value_params� has_all_pks�has_all_defaults�� connection�mapperr r � <genexpr>C s* � �� �z _bulk_insert.<locals>.<genexpr>c 3 s � | ] }d |�� fV qd S r r �r �mappingr& r r r) Y s � T)�bulk�return_defaults� render_nulls�� bookkeepingc S r r ��key�r �pr r r r k r c s g | ]}� | �qS r r )r r2 )r r r r o r )� base_mapper�_cached_connection_dict�session�connection_callable�NotImplementedError�listr' �_sorted_tables�items�isa�_collect_insert_commands�_emit_insert_statements�_identity_class�_identity_key_props�tupler2 )r( �mappings�session_transaction�isstatesr- r. r5 �cached_connections�states�table� super_mapper�records� identity_cls�identity_propsr r )r' r r( r � _bulk_insert# sV � �� � ��rM c s� �j }t|�}�j��jr�jjh�����fdd�� |r2|r*� �fdd�|D �}n dd� |D �}nt|�}|jjr>t d��|� |��|j � � D ]%\}}�� |�sRqHtd |��fdd�|D �d d �} t|d |||| d d � qHd S ) Nc s t �� fdd�� j �� D ��S )Nc 3 s. � | ]\}}|�j v s|� v r||fV qd S r )�committed_state�r �k�v)� search_keysr r r r) s � ��z6_bulk_update.<locals>._changed_dict.<locals>.<genexpr>)r r< )r( r )rR �r r � _changed_dict~ s �z#_bulk_update.<locals>._changed_dictc s g | ]}� �|��qS r r r )rT r( r r r � r z _bulk_update.<locals>.<listcomp>c S r r r r r r r r � r zJconnection_callable / per-instance sharding not supported in bulk_update()c 3 s0 � | ]}d |�� �j r|�j j nd fV qd S r )�_version_id_propr2 r* r& r r r) � s � ��� �z _bulk_update.<locals>.<genexpr>T)r, Fr/ )r5 r6 �_primary_key_propkeysrU r2 �unionr: r7 r8 r9 r' r; r< r= �_collect_update_commands�_emit_update_statements) r( rC rD rE �update_changed_onlyr5 rF rH rI rJ r )rT r' r( rR r � _bulk_updates sJ � ����r[ Fc C s |s| j st| |�D ] }t| |g|dd� q dS g }g }t| �}t| ||�D ]!\}}} } } } } | s3| r>|�||| | | f� q&|�||| | f� q&| j�� D ]'\}} || jvrWqMt ||�}t |||�}t | ||| ||� t | ||| ||� qMt | |tdd� |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 )FNr )r r r r( r' r r r r) � s � � �zsave_obj.<locals>.<genexpr>c s s( � | ]\}}}}}||||d fV qdS )TNr )r r r r( r' �update_version_idr r r r) s � � �)�batch� _sort_states�save_objr6 �_organize_states_for_save�appendr; r< � _pks_by_tabler> rX rY r? � _finalize_insert_update_commandsr )r5 rG �uowtransactionr\ r �states_to_update�states_to_insertrF r r( r' � has_identity� row_switchr] rH �insert�updater r r r` � sp � � �� � ����r` c sr t | �}tt| ||��}| j�� D ]%\�� �� jvrq� �fdd�|D �}t| |�||�}t| ||� �|� qdS )z\Issue UPDATE statements on behalf of a relationship() which specifies post_update. c 3 sJ � | ] \}}}}�|j v r||||� jd ur � �||� j�nd fV qd S r )rc �version_id_col�#_get_committed_state_attr_by_column)r r r � sub_mapperr' �r( rH r r r) s � � ��� �z post_update.<locals>.<genexpr>N)r6 r: � _organize_states_for_post_updater; r< rc � _collect_post_update_commands� _emit_post_update_statements)r5 rG re �post_update_colsrF rf rk r ro r � post_update s, � � ���rt c C s� t | �}tt| ||��}| j}tt|�� ��D ]#}|| }||jvr#q|jr*|jr*qt | |||�}t | |||||� q|D ]\} } }} } |j � || | � q=dS )z�Issue ``DELETE`` statements for a list of objects. This is called within the context of a UOWTransaction during a flush operation. N) r6 r: �_organize_states_for_deleter; �reversed�keysrc �inherits�passive_deletes�_collect_delete_commands�_emit_delete_statements�dispatch� after_delete) r5 rG re rF �states_to_delete�table_to_mapperrH r( �deleter r r' r] r r r � delete_obj; s> � ����r� c c s@ � t | ||�D ]�\}}}}t|j�}|jp|�|�}d } } |s)|j�|||� n|j�|||� |jr;|�|||� |sy||jj v ry|jj | } t � | �} |� | �sy|� | �sft�dt|�|t| �f ��| �d|t|�t| �� |�| � | } |s}| r�|jdur�|�| r�| n|| r�| jn||j�} |||||| | fV qdS )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�boolr2 �_identity_key_from_stater| � before_insert� before_update� _validate_polymorphic_identityr7 � identity_mapr �instance_state�was_already_deleted� is_deleted�orm_exc� FlushErrorr � _log_debug�remove_state_actionsrl rm r ) r5 rG re r r r( r' rh � instance_keyri r] �instance�existingr r r ra i s^ � � � ��� ���ra 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. )r� )r5 rG re r r r rp � s rp c c s` � t | ||�D ]&\}}}}|j�|||� |jdur#|�|||j�}nd}|||||fV qdS )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)r� r| � before_deleterl rm )r5 rG re r r r( r' r] r r r ru � s �� ��ru c c sx � |D ]�\}}}}| |j vrqi } i } |j| } |j| } t| ��|�D ]5} || }| | }|du r9|| vr9|s9q$|s@t|d�sFt|tj�rTt|d�rO|� � n|| |< q$|| |j < q$|ss|j | � | �� dd� | D ��D ]}d| |< ql|rw|r�|j | �| �}|jjr�|j| �| �}nd}nd }}|jdur�|jdur�|j|j| v r�|�d�| |jj < ||| ||| ||fV qdS )zSIdentify sets of values to use in INSERT statements for a list of states. N�__clause_element__c S r r r1 �r �cr r r r r z,_collect_insert_commands.<locals>.<listcomp>TF)rc �_propkey_to_col� _insert_cols_evaluating_none�set� intersection�hasattr� isinstancer � ClauseElementr� r2 �_insert_cols_as_none� difference�_pk_keys_by_table�issubsetr5 �eager_defaults�_server_default_cols�version_id_generatorrl �_cols_by_table)rH rg r, r- r. r r r( r' r r# �propkey_to_col� eval_none�propkey�value�col�colkeyr$ r% r r r r> � sl � �� � �� � � ���r> c # sH � |D �]�\}�}}}||j vrq|j | }i } |j| � |r9t� �fdd�t� �����|j| �D ��} d} nQi } t� ��|j�D ]7} �| } � | }t| d�sXt | t j �rft| d�ra| � � n| | |< qC|j | j�| |j| �durz| | |j< qC|jjr�|j| �| �} nd} |dur�|j|j| v r�|s�| s�| s�|j�� D ]}|j |j j�|�tj�}|j r� nq�q|j}| o�| }|| |j < |s�|j| vr�|j dur�|� |�}|| |j< n|j du r�|r�|| |j< n| s�| s�qd}d}|�r t� �fdd�t� ��|j| �D ��}nfi }|D ]a}|j| j} |j | j�|�tj �}|j �rU|j �r3d||f| jv �rEd}|j d ||j < | �!|jd� n|j d ||j < || v �rTd}n|j"d ||j < ||j du �rnt#�$d ||f ���q| �sv| �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. c 3 s" � | ] }� | j �| fV qd S r r1 �r r� �r� r r r r) Y s � � �z+_collect_update_commands.<locals>.<genexpr>Tr� NFc 3 s$ � | ] }� | j ��|�fV qd S r )�_label�getr� r� r r r) � s � � �� pk_cascadedr zCCan't update table %s using NULL for primary key value on column %s)*rc r� r r� r� r� �_pk_attr_keys_by_tablerN r� r� r r� r� �manager�impl�is_equalr2 r5 r� � _server_onupdate_default_colsr� rl r� �_columntoproperty�values� get_historyr �PASSIVE_NO_INITIALIZE�addedr� r� � PASSIVE_OFF�deleted�pop� unchangedr� r� rk �_table_to_equatedr �populate�passive_updates)re rH rf r, r r( r' r] �pksr# r r% r� r� r� �prop�history� no_params�valr$ �expect_pk_cascaded� pk_params�m� equated_pairsr r� r rX 7 s �� �� � �� �� �� �� � �� � ��� ���� � �� ��rX c c s" � |D ]�\}}}}} |j | } i } d} |j| D ]>} | | v r,|j||| tjd�| | j< q| |v s5| jdurV|j| }|j|j j � ||tj �}|j rV|j d }|| | j < d} q| r�| dur�|j|j| v r�|j} | | | j< t|j �r�| j | vr�|jdur�|�| �}|| | j < ||||| fV qdS )zrIdentify sets of values to use in UPDATE statements for a list of states within a post_update operation. F)�passiveNr T)rc r� �_get_state_attr_by_columnr r� r� �onupdater� r� r2 r� r� r� r� rl r� r� )r5 re rH rf rs r r r( r' r] r� r �hasdatar� r� r� r� r� r r r rq � sP �� � � � � ��rq c c s� � |D ]G\}}}}}||j vrqi } |j | D ]} |�||| � | | j< } | du r2t�d|| f ��q|durE|j|j| v rE|| |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)rc rm r2 r� r� rl r� ) r5 re rH r~ r r r( r' r] r r� r� r r r rz ( s> � � ������ �rz Tc st � j duo � j � j� v �� ��fdd�}| �d�f|�}t|dd� �D �]\\} } } } } }d}t|�}|}d}| sA|�� }d }n |rP| sP� jjrP|�� }d }n � j dur]|�� j �}d }|sc| jj n| jj }|ol| jj }| oq� }| r�|D ],\}}}� } }} } | � |� |�|�}|r�t� |�||||jjd |d � ||j7 }|}qvno|s�|}|D ])\}}}� } }} } || � ||�}|r�t� |�||||jjd |d � ||j7 }q�n>d d � |D �}|p�|o�t|�d k}|| � ||�}||j7 }|D ] \}}}� } }} } |�rt� |�||||jjd |d � q�|�r+|t|�k�r*t�d �jt|�|f ��q$��r7t�d|jj � q$dS )z`Emit UPDATE statements corresponding to value lists collected by _collect_update_commands().Nc sl t �� } � j� D ]}| j�|t j|j|jd�k� q �r/| j�� jt j� jj� jjd�k� �� | �}|S �N)�type_) r �and_rc �clausesrb � bindparamr� �typerl rk ��clauser� �stmt�r( �needs_version_idrH r r � update_stmt` s ���� z,_emit_update_statements.<locals>.update_stmtrk c S �* | d t | d �t| d �| d | d fS �N� r � � � �r� r� ��recr r r �<lambda>{ � �z)_emit_update_statements.<locals>.<lambda>r FTc S � g | ]}|d �qS �r r �r r� r r r r � r 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.)rl r� �_memor r: r- r5 r� �dialect�supports_sane_rowcount� supports_sane_rowcount_returning� supports_sane_multi_rowcount�executer� � _postfetch�context�compiled_parameters�rowcount�lenr� �StaleDataError� descriptionr �warn�dialect_description) r5 re rF r( rH rk r0 r� � cached_stmtr' � paramkeys�hasvaluer% r$ rJ �rows� statementr- �assert_singlerow�assert_multirow�allow_multirowr r r r# r� �check_rowcount� multiparamsr r� r rY O s ��� ��� ��� � � � �� � � �� � �� ������ ��rY c C s� | � d|f|j�}t|dd� �D ]�\\}} } } } } |}|r*| s&| jr&|jjsi| ri| sit| �} dd� | D �}|| �||�}|rht| |j j �D ]!\\}}}}}}} } }|rat ||||||||d� qFt |||� qFq| ss| jrs|� � }n |jdur~|� |j�}| D ]h\}}}}}}} } |r�|�|�|�|�}n|| �||�}|j j}|dur�t||j| �D ] \}}|j| } |dur�||v s�|�| j�du r�||| j< q�|r�|r�t |||||||j j d |d� q�t |||� q�qdS ) z`Emit INSERT statements corresponding to value lists collected by _collect_insert_commands().rj c S r� r� r� r� r r r r� # r� z)_emit_insert_statements.<locals>.<lambda>c S r� r� r r� r r r r : r z+_emit_insert_statements.<locals>.<listcomp>FNr )r� rj r r� r� �implicit_returningr: r� �zipr� r� r� �_postfetch_bulk_saver- rl r� �inserted_primary_keyrc r� r� r2 ) r5 re rF r( rH rj r0 r� r' �pkeysr� r$ r% rJ r� r r� r r r � mapper_recr" r# �last_inserted_params�result� primary_key�pkr� r� r r r r? s� �� ������ �� � � �� � � � ���r? c s� � j duo � j � j� v �� ��fdd�}| �d�f|�}t|dd� �D ]�\}} d} t| �} |d } � j du r;| jjn| jj} | oD| jj} � pI| }|st| }| D ]"\}}}} }|| � ||�}t ||�||||j j d � | |j 7 } qPn6dd � | D �}| p�| o�t|�d k}|| � ||�}| |j 7 } | D ]\}}}} }t ||�||||j j d � q�|r�| t| �kr�t�d �jt| �| f ��q$�r�t�d |jj � q$dS ) zeEmit UPDATE statements corresponding to value lists collected by _collect_post_update_commands().Nc s� t �� } � j� D ]}| j�|t j|j|jd�k� q �r/| j�� jt j� jj� jjd�k� �� | �}� jd ur?|� � j�}|S r� ) r r� rc r� rb r� r� r� rl rk r- r� r� r r r� � s$ ���� z1_emit_post_update_statements.<locals>.update_stmtrt c S s | d t | d �fS )N� r� )r� r� r r r r� � r z._emit_post_update_statements.<locals>.<lambda>r c S s g | ] \}}}}}|�qS r r )r r r r r" r r r r r � s ��z0_emit_post_update_statements.<locals>.<listcomp>r r� r� )rl r� r� r r: r� r� r� r� r� �_postfetch_post_updater� r� r� r� r� r� r� r r� r� )r5 re rF r( rH rk r� r� r2 rJ r� r' r� r r r r r r r r� r r r� r rr � s� � � ��� � � �� � �������rr c sL � j duo � j � j� v �� ��fdd�}| �d�f|�}t|dd� �D ]\}} dd� | D �} || }t| �} d } d } �rh|jjsh|jjrXd } | D ] }|�||�}| |j 7 } qIn t � d |jj � |�|| � n |�|| �}�srd } |j } | j r�| d kr�| | kr�|jjs�t| �dkr�| r�t � d�j| | f � q$t�d�j| | f ��q$dS )z`Emit DELETE statements corresponding to value lists collected by _collect_delete_commands().Nc sh t �� } � j� D ]}| j�|t j|j|jd�k� q �r/| j�� jt j� jj� jjd�k� �� | �S r� ) r r� rc r� rb r� r2 r� rl r� )r� r� �r( �need_version_idrH r r � delete_stmt s � ��� z,_emit_delete_statements.<locals>.delete_stmtr� c S s | d S )Nr r r� r r r r� ! s z)_emit_delete_statements.<locals>.<lambda>c S s g | ]\}}|�qS r r )r r r' r r r r " r z+_emit_delete_statements.<locals>.<listcomp>���Fr 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.)rl r� r� r r� r� r� r� r� r� r r� r� �confirm_deleted_rowsr� r� r� )r5 re rF r( rH r� r r� r' �recs� del_objects�expected� rows_matched� only_warnr r� r r r r{ sf ��� ��� �� �� ����r{ c s |D ]�\� }}}}|j r"� �� fdd�|j D ��}|r"� �� j|� g }| jr1|�� j�|j�� |j durJ|j du rJ|j j � j v rJ|�|j j g� |r`| �� �� _ tj|j�|�� j � |d� |sk|j�||� � n|j�||� � |j du r�|j dur�||j j du r�t�d��qdS )zzfinalize state on states that have been inserted or updated, including calling after_insert/after_update events. c sB g | ] }|j r|jr |j� jv s |j s |js |j� jvr|j�qS r )�expire_on_flush�deferredr2 r r3 rS r r r j s �� �� � z4_finalize_insert_update_commands.<locals>.<listcomp>NF)� refresh_state�only_load_propsz2Instance does not contain a non-NULL version value)�_readonly_props�unmodified_intersection�_expire_attributesr r� �extend�_unloaded_non_objectr� �&_server_default_plus_onupdate_propkeysrl r� rU r2 �unloadedr� r � load_on_identr7 �queryr| � after_insert� after_updater� r� ) r5 re rG r r( r' rh �readonly� toload_nowr rS r rd a sN ���� � ���rd c s� |� |�rd S |jjj}|jjj}� jd ur&� j� j| v r&t|�� jg }t� j j j �} | r1g } |D ]"} | j |v rU| � j v rU|| j |� j | j < | rU| �� j | j � q3| rc| rc� j j � ||| � |ru|�|j� fdd�|D �� d S d S )Nc �" g | ] }|� j v r� j | j�qS r �r� r2 r� �r( r r r � � �z*_postfetch_post_update.<locals>.<listcomp>)r� r� �compiled�prefetch� postfetchrl r� r: r� � class_managerr| � refresh_flushr2 r� rb r r ) r( re rH r r r r � prefetch_cols�postfetch_colsr3 �load_evt_attrsr� r r- r r � s8 �� ���r c s� |j jj} |j jj} |j jj�� jdur$� j� j| v r$t| �� jg } t� j j j �} | r/g } �r]|j j } | dur]�D ]!}|j rE|j jrEq;� j�|�}|r\| | ||j< | r\| �|j� q;| D ]"}|j|v r�|� jv r�||j |� j| j< | r�| �� j| j� q_| r�| r�� j j � ||| � |r�|r�| ��fdd�|D �� | r�|�|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 |� vr|�qS r )r )r r� )�returning_colsr r r s ��z _postfetch.<locals>.<listcomp>c r+ r r, r� r- r r r r. )r� r/ r0 r1 � returningrl r� r: r� r2 r| r3 �returned_defaultsr �isinsertr� r� r2 rb r! r r r� r r� r� )r( re rH r r r r r# �isupdater4 r5 r3 r6 �rowr� r� r� r� r� r )r( r7 r r� � sn ��� �� �� ��r� c C s& | j | D ] \}}t�|||� qd S r )r� r �bulk_populate_inherit_keys)r( r rH r� r� r r r r 0 s �r c c sb � |j jr |j j}n|j�| �}d}t| |�D ]}|r"|| |�� �}|jj}||j||fV qdS )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) r7 r8 � transactionr' r_ �objr� r( r )r5 re rG r8 r' r r( r r r r� 5 s � �r� c s t �� fdd��S )Nc s | j � jd�S )N)�compiled_cache)�execution_options�_compiled_cache)r" �r5 r r r� R s �z)_cached_connection_dict.<locals>.<lambda>)r � PopulateDictrC r rC r r6 O s �r6 c C st t |�}t dd� |D ��}|�|� z t|| jd�}W n ty. } zt�d| ��d }~ww t|t�d�d�| S )Nc s s � | ] }|j d ur|V qd S r r1 )r �sr r r r) Z s � z _sort_states.<locals>.<genexpr>r1 z^Could not sort objects by primary key; primary key values must be sortable in Python (was: %s)� insert_order) r� �difference_update�sorted�_persistent_sortkey_fn� TypeError�sa_exc�InvalidRequestError�operator� attrgetter)r( rG �pending� persistent�persistent_sorted�errr r r r_ X s& �������r_ c @ sr e Zd ZdZdd� Zdd� Zedd� �Zedd � �Z d d � Z d d � Z dd� Z e �d�dd� �Zdd� Zdd� ZdS )�BulkUDz5Handle bulk update and deletes via a :class:`.Query`.c C s$ |� d�| _| j�� | _| �� d S )NF)�enable_eagerloadsr& � _bind_mapperr( �_validate_query_state��selfr& r r r �__init__o s zBulkUD.__init__c C s� ddd t jfddd t jfdddt jfdd dt jfd d dt jfd d dt jffD ]\}}}}|t| j|�|�s=t�d|f ��q&d S )N�_limitzlimit()�_offsetzoffset()� _order_byz order_by()F� _group_byz group_by()� _distinctz distinct()� _from_objz2join(), outerjoin(), select_from(), or from_self()r zCCan't call Query.update() or Query.delete() when %s has been called)rM �is_�eq�getattrr& rK rL )rX �attr�methname�notset�opr r r rV t s& �� ����z BulkUD._validate_query_statec C s | j jS r )r& r7 �rX r r r r7 � s zBulkUD.sessionc G sD z || }W ||� S t y! t�dd�tdd� |D ��� ��w )Nz3Valid strategies for session synchronization are %sz, c s s � | ]}t |�V qd S r )�repr)r �xr r r r) � � � z"BulkUD._factory.<locals>.<genexpr>)�KeyErrorrK � ArgumentError�joinrH )�cls�lookup�synchronize_session�arg�klassr r r �_factory� s ����zBulkUD._factoryc C s4 | � � | �� | �� | �� | �� | �� d S r )�_do_before_compile�_do_pre�_do_pre_synchronize�_do_exec�_do_post_synchronize�_do_postrg r r r �exec_� s z BulkUD.exec_c C s | j �|| j�| _| jj| _d S r )r& � _execute_crudr( r r� )rX r� r r r � _execute_stmt� s zBulkUD._execute_stmtc C s t � �r )r9 rg r r r rt � s zBulkUD._do_before_compilezsqlalchemy.orm.queryc C s� | j }|�|�| _t|jd |j�r@t� }|jD ]}t||j�s&|�� n |�t |j �� qt |�dkr:t � d��|�� | _n|�d�jj| _|j}|jrT|�� d S d S )Nr r zLThis operation requires only one Table or entity be specified as the target.)r& � QueryContextr� r� � _entities� _ColumnEntityr� �clearrk r �columnr� rK rL r� � primary_table�_only_entity_zeror( � local_tabler7 � _autoflush)rX �querylibr& �tables�entr7 r r r ru � s. � �� �zBulkUD._do_prec C � d S r r rg r r r rv � � zBulkUD._do_pre_synchronizec C r� r r rg r r r rx � r� zBulkUD._do_post_synchronizeN)�__name__� __module__� __qualname__�__doc__rY rV �propertyr7 � classmethodrs rz r| rt r � dependenciesru rv rx r r r r rS l s " rS c @ � e Zd ZdZdd� Zdd� ZdS )� BulkEvaluatezDBulkUD which does the 'evaluate' method of session state resolution.c C r� r r )rX �evaluator_compilerr r r �_additional_evaluators� r� z#BulkEvaluate._additional_evaluatorsc s� | j }|�� j�z t���}|jd ur|�|j�� ndd� � | �|� W n tjy8 } zt � d| ��d }~ww � �fdd�|j j � � D �| _d S )Nc S s dS )NTr )r? r r r �eval_condition� r� z8BulkEvaluate._do_pre_synchronize.<locals>.eval_conditionztCould not evaluate current criteria in Python: "%s". Specify 'fetch' or False for the synchronize_session parameter.c s, g | ]\\}}}}t |��r� |�r|�qS r )� issubclass)r rn r �identity_tokenr? �r� � target_clsr r r � s ���z4BulkEvaluate._do_pre_synchronize.<locals>.<listcomp>)r& � _mapper_zero�class_r �EvaluatorCompiler� whereclause�processr� �UnevaluatableErrorrK rL r7 r� r< �matched_objects)rX r& r� rR r r� r rv � s& ���� �z BulkEvaluate._do_pre_synchronizeN)r� r� r� r� r� rv r r r r r� � s r� c @ � e Zd ZdZdd� ZdS )� BulkFetchzABulkUD which does the 'fetch' method of session state resolution.c C sB | j }|j}|�� }|j�| jj�}|j|| j|j d�� � | _ d S )N)r( r ) r& r7 �_compile_contextr� �with_only_columnsr� r r� r( �_params�fetchall� matched_rows)rX r& r7 r� � select_stmtr r r rv � s � ��z BulkFetch._do_pre_synchronizeN)r� r� r� r� rv r r r r r� � s r� c s\ e Zd ZdZ� fdd�Zedd� �Zdd� Zedd � �Z ed d � �Z d d � Z dd� Z � Z S )� BulkUpdatez BulkUD which handles UPDATEs.c s t t| ��|� || _|| _d S r )�superr� rY r� � update_kwargs)rX r& r� r� �� __class__r r rY s zBulkUpdate.__init__c C s t �tttd�||||�S �N)�evaluate�fetchF)rS rs �BulkUpdateEvaluate�BulkUpdateFetchr� )rn r& rp r� r� r r r �factory s ��zBulkUpdate.factoryc C �< | j jjr| j jjD ]}|| j | �}|d ur|| _ q d S d S r )r& r| �before_compile_update�rX �fn� new_queryr r r rt � ��z BulkUpdate._do_before_compilec C s� g }t | jd�r | j�� n| jD ]:\}}| jrCt|tj�r,t| j|�}|�|� |�� qt|t j �r;|�|� |�� q|� ||f� q|� ||f� q|S )Nr< ) r� r� r< r( r� r � string_typesr r! �_bulk_update_tuplesr �QueryableAttributerb )rX r� rP rQ �descr r r �_resolved_values% s � � zBulkUpdate._resolved_valuesc C s� g }| j D ]G\}}t|tj�r|�|j|f� qt|d�r!|�� }| jrFt|t j �rFz| jj | }W n t j y< Y qw |�|j|f� qt�d| ��|S )Nr� zInvalid expression type: %r)r� r� r r� rb r2 r� r� r( r � ColumnElementr� r� �UnmappedColumnErrorrK rL )rX r� rP rQ rc r r r �"_resolved_values_keys_as_propnames9 s$ ��z-BulkUpdate._resolved_values_keys_as_propnamesc C sH | j }| j�dd�st|�}tj| j| jj|fi | j��}| � |� d S )N�preserve_parameter_orderF) r� r� r� r r rk r� r� r� r| )rX r� r� r r r rw P s ��zBulkUpdate._do_execc C � | j j}|j�| � d S r )r& r7 r| �after_bulk_update�rX r7 r r r ry _ � zBulkUpdate._do_post)r� r� r� r� rY r� r� rt r� r� r� rw ry � __classcell__r r r� r r� s r� c sD e Zd ZdZ� fdd�Zedd� �Zdd� Zdd � Zd d � Z � Z S ) � BulkDeletez BulkUD which handles DELETEs.c s t t| ��|� d S r )r� r� rY rW r� r r rY g s zBulkDelete.__init__c C s t �tttd�||�S r� )rS rs �BulkDeleteEvaluate�BulkDeleteFetchr� )rn r& rp r r r r� j s ��zBulkDelete.factoryc C r� r )r& r| �before_compile_deleter� r r r rt v r� z BulkDelete._do_before_compilec C s t �| j| jj�}| �|� d S r )r r� r� r� r� r| )rX r r r r rw } s zBulkDelete._do_execc C r� r )r& r7 r| �after_bulk_deleter� r r r ry � r� zBulkDelete._do_post) r� r� r� r� rY r� r� rt rw ry r� r r r� r r� d s r� c @ r� )r� zSBulkUD which handles UPDATEs using the "evaluate" method of session resolution.c C s4 i | _ | j}|D ]\}}|�t�|��| j |< qd S r )�value_evaluatorsr� r� r �_literal_as_binds)rX r� r� r2 r� r r r r� � s ��z)BulkUpdateEvaluate._additional_evaluatorsc C s� | j j}t� }t| j�� �}| jD ]B}t�|�t� |�}}|j � |�}|D ] }| j| |�||< q&|j j �|d |� |�|t|�� |�|t|��|�� |�|� q|�|� d S r )r& r7 r� r: r� rw r� r r� � instance_dict� unmodifiedr� r� r| �refresh�_commitr r� �add�_register_altered) rX r7 rG �evaluated_keysr? r r � to_evaluater2 r r r rx � s" � � z'BulkUpdateEvaluate._do_post_synchronizeN)r� r� r� r� r� rx r r r r r� � s r� c @ r� )r� zSBulkUD which handles DELETEs using the "evaluate" method of session resolution.c C s | j j�dd� | jD �� d S )Nc S s g | ]}t �|��qS r )r r� )r r? r r r r � r z;BulkDeleteEvaluate._do_post_synchronize.<locals>.<listcomp>)r& r7 �_remove_newly_deletedr� rg r r r rx � s �z'BulkDeleteEvaluate._do_post_synchronizeN�r� r� r� r� rx r r r r r� � � r� c @ r� )r� zPBulkUD which handles UPDATEs using the "fetch" method of session resolution.c s� | j j� | j �� �t� fdd��fdd�| jD �D ��}| j}tdd� |D ��}|D ]}|�|j�}|r:� �||� q*� � |� d S )Nc s& g | ]}|� j v rt�� j | ��qS r )r� r r� )r � identity_key)r7 r r r � s �z8BulkUpdateFetch._do_post_synchronize.<locals>.<listcomp>c s g | ] }� � t|���qS r )� identity_key_from_primary_keyr: )r r )� target_mapperr r r � s ���c s s � | ]\}}|V qd S r r rO r r r r) � rj z7BulkUpdateFetch._do_post_synchronize.<locals>.<genexpr>) r& r7 r� r� r� r� r� r � _expire_stater� )rX rG r� �attribr � to_expirer )r7 r� r rx � s" ��� �z$BulkUpdateFetch._do_post_synchronizeNr� r r r r r� � r� r� c @ r� )r� zPBulkUD which handles DELETEs using the "fetch" method of session resolution.c C sR | j j}| j �� }| jD ]}|�t|��}||jv r&|�t� |j| �g� q d S r ) r& r7 r� r� r� r: r� r� r r� )rX r7 r� r r� r r r rx � s � �����z$BulkDeleteFetch._do_post_synchronizeNr� r r r r r� � r� r� )F)FFF)T)6r� � itertoolsr r rM � r r r r� r r �baser r rK r r r �sql.baser rM r[ r` rt r� ra rp ru r> rX rq rz rY r? rr r{ rd r r� r r� r6 r_ �objectrS r� r� r� r� r� r� r� r� r r r r �<module> sr P EV-.Q �Z � :8. � I � s]F)` e(\#(
Save
Cancel