o
    y¬´\àø  ã                   @   s  d Z ddlmZ ddlmZ ddlZddlmZ ddlmZ ddlmZ	 dd	lm
Z
 dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd„ Zdd„ ZdTdd„Zdd„ Zdd„ Zdd„ Zdd „ Zd!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ƒZ5dS )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|ƒ}|jjrtdƒ‚|r*|r"dd„ |D ƒ}dd„ |D ƒ}ndd„ |D ƒ}nt|ƒ}| |¡‰ |j ¡ D ].\}	}
ˆ 	|
¡sBq8‡ ‡fdd„t
|	‡ ‡fdd„|D ƒd	||d
�D ƒ}t|d ||
|	||d� q8|r‹|r�ˆj}dd„ ˆjD ƒ}|D ]\}‰|t‡fd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‘qS © ©Údict©Ú.0Ústater   r   úL/var/www/html/env/lib/python3.10/site-packages/sqlalchemy/orm/persistence.pyÚ
<listcomp>7   ó    z _bulk_insert.<locals>.<listcomp>c                 S   s   g | ]\}}|‘qS r   r   )r   r   Údict_r   r   r   r   8   ó    c                 S   ó   g | ]}|j ‘qS r   r   r   r   r   r   r   :   ó    c           	   	   3   s4   � | ]\}}}}}}}}d ||ˆˆ |||fV  qd 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 |ˆˆ fV  qd 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 | ]}ˆ | ‘qS 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‰ˆjrˆjjh ˆ¡‰‡fdd„‰ |r2|r*‡ ‡fdd„|D ƒ}ndd„ |D ƒ}nt|ƒ}|jjr>t	dƒ‚| 
|¡‰|j ¡ D ]%\}}ˆ |¡sRqHtd |‡‡fdd„|D ƒd	d
�}	t|d ||||	dd� qHd S )Nc                    s   t ‡‡ fdd„ˆ j  ¡ D ƒƒS )Nc                 3   s.   � | ]\}}|ˆj v s|ˆ v r||fV  qd 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 | ]}ˆ ˆ|ƒ‘qS 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 nd fV  qd 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 st|ƒD ]}t| |g|dd� q	dS g }g }t| ƒ}t| ||ƒD ]!\}}}	}
}}}|s2|r=| |||	|
|f¡ q%| |||	|
f¡ q%| j ¡ D ]'\}}	||	jvrVqLt	||ƒ}t
|||ƒ}t| |||	||ƒ t| |||	||ƒ qLt| |tdd„ |D ƒdd„ |D ƒƒƒ dS )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 fV  qdS )FNr   )r   r   r   r(   r'   r   r   r   r)   ü   s
   € 
ÿ
ÿzsave_obj.<locals>.<genexpr>c                 s   s(   � | ]\}}}}}||||d fV  qdS )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 | ƒ}tt| ||ƒƒ}| j ¡ D ]%\‰‰ ˆˆ jvrq‡ ‡fdd„|D ƒ}t| |ˆ||ƒ}t| ||ˆ ˆ|ƒ qdS )z\Issue UPDATE statements on behalf of a relationship() which
    specifies post_update.

    c              
   3   sJ   � | ] \}}}}ˆ|j v r||||ˆ jd urˆ  ||ˆ j¡nd fV  qd 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 | ƒ}tt| ||ƒƒ}| j}tt| ¡ ƒƒD ]#}|| }||jvr#q|jr*|jr*qt	| |||ƒ}t
| |||||ƒ q|D ]\}	}
}}}|j |||	¡ q=dS )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ƒ}|jp| |¡}d }	}
|s)|j |||¡ n|j |||¡ |jr;| |||¡ |sy||jj	v ry|jj	| }t
 |¡}| |¡sy| |¡sft dt|ƒ|t|ƒf ¡‚|  d|t|ƒt|ƒ¡ | |¡ |}	|s}|	r“|jdur“| |	rˆ|	n||	rŽ|	jn||j¡}
||||||	|
fV  qdS )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 |||¡ |jdur#| |||j¡}nd}|||||fV  qdS )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   sp  � |D ]²\}}}}| |j vrqi }	i }
|j|  }|j|  }t|ƒ |¡D ]6}|| }|| }|du r9||vr9|s9q$|s@t|dƒsFt|tjƒrUt|dƒrO| 	¡ n||
|j
< q$||	|j
< q$|so|j|   |	¡ |
¡D ]}d|	|< qh|rs|r‹|j|   |	¡}|jjrˆ|j|   |	¡}nd}nd }}|jdurª|jdurª|j|j|  v rª| d¡|	|jj
< |||	|||
||fV  qdS )zSIdentify sets of values to use in INSERT statements for a
    list of states.

    NÚ__clause_element__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              	   #   s
  � |D �]~\}‰}}}||j vrq|j | }i }	|j| ‰ |r9t‡ ‡fdd„tˆ ƒ ˆ¡ |j| ¡D ƒƒ}
d}nQi }
tˆ ƒ |j¡D ]7}ˆ| }ˆ | }t|dƒsXt	|t
jƒrft|dƒra| ¡ n||	|< qC|j| j ||j| ¡durz||
|j< qC|jjrˆ|j|  |
¡}nd}|duré|j|j| v ré|s·|
s·|	s·|j ¡ D ]}|j|j j |ˆtj¡}|jrµ nq¡q|j}|
 o¿|	 }||
|j< |sÌ|j|
vrÜ|jdurÜ| |¡}||
|j< n|jdu rè|rè||
|j< n|
sî|	sîqd}|�rt‡ ‡fdd„tˆ ƒ |j| ¡D ƒƒ}ndi }|D ]_}|j| j}|j| j |ˆtj¡}|j�rQ|j �r1d||f| jv �rA|jd	 ||j< |
 !|jd¡ n|j d	 ||j< ||	v �rPd}n|j"d	 ||j< ||j du �rjt# $d
||f ¡‚�q|
�sr|	�r‚|
 %|¡ |ˆ|
|||	||fV  qdS )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 ˆ| fV  qd S r   r1   ©r   r¦   ©r¤   r   r   r   r)   Y  s
   € ÿ
ÿz+_collect_update_commands.<locals>.<genexpr>Tr”   NFc                 3   s$   � | ]}ˆ | j ˆ |¡fV  qd 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   )re   rH   rf   r,   r   r(   r'   r]   Úpksr#   r    r%   r¦   r§   r¨   ÚpropÚhistoryÚ	no_paramsÚvalr$   Ú	pk_paramsr   r«   r   rX   7  sê   €ú


üÿÿ
ÿý
ÿý
€þÿÿ
ÿ



€ÿ
þÿÿÿ
€ÿÿÿ
ø€ �òrX   c                 c   s"  � |D ]‹\}}}}}	|j | }
i }d}|j| D ]>}||
v r,|j|||tjd�||j< q||v s5|jdurV|j| }|j|j	 j
 ||tj¡}|jrV|jd }|||j	< d}q|rŽ|	dur†|j|j| v r†|j}|	||j< t|j	ƒr†|j	|vr†|jdur†| |	¡}|||j	< |||||fV  qdS )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 vrqi }	|j | D ]}
| |||
¡ |	|
j< }|du r2t d||
f ¡‚q|durE|j|j| v rE||	|jj< |	|fV  qdS )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                    sn  ˆ j duoˆ j ˆ jˆ v ‰‡ ‡‡fdd„}|  dˆf|¡}t|dd„ ƒD �]\\}	}
}}}}d}t|ƒ}|}d}|sA| ¡ }d	}n|rP|sPˆ jjrP| ¡ }d	}nˆ j dur]| ˆ j ¡}d	}|sc|	jj	n|	jj
}|ol|	jj}|oqˆ }|r£|D ]+\}}}‰ }	}}}|	 | |¡|¡}|rštˆ |ˆ||||jjd |ƒ ||j7 }|}qvnm|sÓ|}|D ](\}}}‰ }	}}}||	  ||¡}|rÌtˆ |ˆ||||jjd |ƒ ||j7 }q©n=d
d„ |D ƒ}|pã|oãt|ƒdk}||	  ||¡}||j7 }|D ]\}}}‰ }	}}}|�rtˆ |ˆ||||jjd |ƒ qó|�r(|t|ƒk�r't dˆjt|ƒ|f ¡‚q$ˆ�r4t d|jj ¡ q$dS )z`Emit UPDATE statements corresponding to value lists collected
    by _collect_update_commands().Nc                     sl   t  ¡ } ˆ jˆ D ]}| j |t j|j|jd�k¡ q	ˆr/| j ˆ jt jˆ jjˆ jjd�k¡ ˆ 	| ¡}|S ©N)Útype_)
r   Úand_rc   Úclausesrb   Ú	bindparamr¬   Útyperl   rk   ©Úclauser¨   Ústmt©r(   Úneeds_version_idrH   r   r   Úupdate_stmtN  s    ÿþÿÿ
z,_emit_update_statements.<locals>.update_stmtrk   c                 S   ó*   | d t | d ƒt| d ƒ| d | d fS ©Né   r   é   é   é   ©r—   rƒ   ©Úrecr   r   r   Ú<lambda>i  ó   

ûz)_emit_update_statements.<locals>.<lambda>r   FTc                 S   ó   g | ]}|d  ‘qS ©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#   ÚcÚcheck_rowcountÚmultiparamsr   rÐ   r   rY   =  s  
þþýÿþý

ÿýþ

÷
ø

é
÷ÿ
ø
æÿÿ
÷
ø€þÿÿþÿ€ �ïrY   c                 C   sÄ  |   d|f|j¡}t|dd„ ƒD ]Ï\\}}	}
}}}|}|r*|s&| jr&|jjsh|rh|
sht|ƒ}dd„ |D ƒ}||  ||¡}|rgt||j	j
ƒD ] \\}}}}}}}}}|r`t||||||||ƒ qFt|||ƒ qFq|sr| jrr| ¡ }n|jdur}| |j¡}|D ]_\}}}}}}}}|r•| | |¡|¡}n||  ||¡}|j	j}|durÄt||j| ƒD ]\}}|j| }| |j¡du rÃ|||j< q­|rÞ|rØt|||||||j	j
d |ƒ qt|||ƒ qqdS )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>Nr   )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 duoˆ j ˆ jˆ v ‰‡ ‡‡fdd„}|  dˆf|¡}t|dd„ ƒD ]§\}}	d}
t|	ƒ}	|d }ˆ j du r;|jjn|jj}|oD|jj}ˆ pI|}|st|}|	D ]"\}}}}}||  	||¡}t
||ˆ||||jjd ƒ |
|j7 }
qPn6dd	„ |	D ƒ}|p„|o„t|ƒd
k}||  	||¡}|
|j7 }
|	D ]\}}}}}t
||ˆ||||jjd ƒ q”|rÀ|
t|	ƒkr¿t dˆjt|	ƒ|
f ¡‚q$ˆrËt d|jj ¡ q$dS )zeEmit UPDATE statements corresponding to value lists collected
    by _collect_post_update_commands().Nc                     s‚   t  ¡ } ˆ jˆ D ]}| j |t j|j|jd�k¡ q	ˆr/| j ˆ jt jˆ jjˆ jjd�k¡ ˆ 	| ¡}ˆ jd ur?| 
ˆ 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 ƒfS )Né   rÕ   )r—   rÚ   r   r   r   rÜ   ¤  r   z._emit_post_update_statements.<locals>.<lambda>r   c                 S   s   g | ]	\}}}}}|‘qS 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   y  sŠ   
þþ
ÿýþ

ù	õþÿ

ù
þÿÿþÿ€½rr   c                    sX  ˆ j duoˆ j ˆ jˆ v ‰‡ ‡‡fdd„}|  dˆf|¡}t|dd„ ƒD ]…\}}	dd„ |	D ƒ}
|| }t|
ƒ}d	}d
}|jjrM| ||
¡}ˆsId}|j}n8ˆrx|jj	rfd}|
D ]}| ||¡}||j7 }qWnt
jd|jj dd� | ||
¡ n| ||
¡}ˆs‚d}|j}| jr©|d	kr©||kr©|ržt
 dˆj||f ¡ q$t dˆj||f ¡‚q$dS )z`Emit DELETE statements corresponding to value lists collected
    by _collect_delete_commands().Nc                     sh   t  ¡ } ˆ jˆ D ]}| j |t j|j|jd�k¡ q	ˆr/| j ˆ jt jˆ jjˆ jjd�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 | ]\}}|‘qS r   r   )r   r    r'   r   r   r   r   
  r   z+_emit_delete_statements.<locals>.<listcomp>éÿÿÿÿFTr   zMDialect %s does not support deleted rowcount - versioning cannot be verified.é   )Ú
stacklevelz­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"ˆ  ‡ fdd„|j D ƒ¡}|r"ˆ  ˆ j|¡ g }| jr1| ˆ j |j¡¡ |j	durJ|j
du rJ|jjˆ jv rJ| |jjg¡ |r`|  ˆ ¡ˆ _tj|j |¡ˆ jˆ |d� |sk|j ||ˆ ¡ n|j ||ˆ ¡ |j
du rŠ|j	durŠ||jj du rŠt d¡‚qdS )zzfinalize state on states that have been inserted or updated,
    including calling after_insert/after_update events.

    c                    sB   g | ]}|j r|jr|jˆ jv s|j s|js|jˆ jvr|j‘qS r   )Úexpire_on_flushÚdeferredr2   r   r3   rS   r   r   r   Q  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   H  sN   
þÿÿÿ


ü

ÿ€Ärd   c                    sî   |  |¡rd S |jjj}|jjj}ˆ jd ur&ˆ jˆ j| v r&t|ƒˆ jg }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‡ fdd„|D ƒ¡ d S d S )Nc                    ó"   g | ]}|ˆ j v rˆ j | j‘qS r   ©r´   r2   ©r   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_colsr0  Úload_evt_attrsrú   r   r*  r   r  Ž  s8   



€ÿ
þþÿr  c              
      sn  |j jj}|j jj}	|j jj}
ˆ jdur$ˆ jˆ j| v r$t|ƒˆ jg }tˆ j	j
jƒ}|r/g }|
r]|j j}|dur]|
D ]!}|jrE|j jrEq;ˆ j |¡}|r\|| ||j< |r\| |j¡ q;|D ]"}|j|v r�|ˆ jv r�||j |ˆ j| j< |r�| ˆ j| j¡ q_|r�|r�ˆ j	j
 |||¡ |	rŸ| |j‡ fdd„|	D ƒ¡ ˆ j| D ]\}}t ||||||ˆ j¡ q¤dS )z†Expire attributes in need of newly persisted database state,
    after an INSERT or UPDATE statement has proceeded for that
    state.Nc                    r'  r   r(  r)  r*  r   r   r   í  r+  z_postfetch.<locals>.<listcomp>)rê   r,  r-  r.  Ú	returningrl   r£   r:   rƒ   r/  r|   r0  Úreturned_defaultsr  Úisinsertr´   r­   r2   rb   r  r   Ú_table_to_equatedr	   ÚpopulateÚpassive_updates)r(   re   rH   r   r   r  r    r#   r1  r2  Úreturning_colsr0  r3  Úrowr¨   r¾   rú   ÚmÚequated_pairsr   r*  r   ré   ·  sb   



€€ÿ
þþùÿré   c                 C   s&   | j | D ]\}}t |||¡ qd S r   )r7  r	   Úbulk_populate_inherit_keys)r(   r   rH   r<  r=  r   r   r   rÿ     s   ÿrÿ   c                 c   s`   � |j jr
|j j}n|j | ¡}d}t|ƒD ]}|r!|| | ¡ ƒ}|jj}||j||fV  qdS )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‚     s   €
úr‚   c                    s   t  ‡ fdd„¡S )Nc                    s   | j ˆ jd�S )N)Úcompiled_cache)Úexecution_optionsÚ_compiled_cache)r"   ©r5   r   r   rÜ   %  s    ÿz)_cached_connection_dict.<locals>.<lambda>)r   ÚPopulateDictrD  r   rD  r   r6   "  s   
ÿr6   c                 C   sF   t | ƒ}t dd„ |D ƒƒ}| |¡ t|t d¡d�t|dd„ d� S )Nc                 s   s   � | ]
}|j d ur|V  qd S r   r1   )r   Úsr   r   r   r)   -  s   € z_sort_states.<locals>.<genexpr>Úinsert_orderr1   c                 S   s
   | j d S r  r1   )Úqr   r   r   rÜ   0  s   
 z_sort_states.<locals>.<lambda>)r—   Údifference_updateÚsortedÚoperatorÚ
attrgetter)rG   ÚpendingÚ
persistentr   r   r   r_   +  s   
ÿr_   c                   @   sr   e Zd ZdZdd„ Zdd„ Zedd„ ƒZedd	„ ƒZ	d
d„ Z
dd„ Zdd„ Ze d¡dd„ ƒZdd„ Zdd„ ZdS )Ú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__7  s   zBulkUD.__init__c              	   C   s€   ddd t jfddd t jfdddt jfdd	dt jfd
ddt jfdddt jffD ]\}}}}|t| j|ƒ|ƒs=t d|f ¡‚q&d S )NÚ_limitzlimit()Ú_offsetzoffset()Ú	_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)rK  Úis_ÚeqÚgetattrr"  Úsa_excÚInvalidRequestError)rT  ÚattrÚmethnameÚnotsetÚopr   r   r   rR  <  s&   üúÿÿÿózBulkUD._validate_query_statec                 C   s   | j jS r   )r"  r7   ©rT  r   r   r   r7   P  s   zBulkUD.sessionc                 G   sD   z	|| }W ||Ž S  t y!   t dd tdd„ |D ƒƒ¡ ¡‚w )Nz3Valid strategies for session synchronization are %sz, c                 s   s   � | ]}t |ƒV  qd S r   )Úrepr)r   Úxr   r   r   r)   [  ó   € z"BulkUD._factory.<locals>.<genexpr>)ÚKeyErrorr_  ÚArgumentErrorÚjoinrJ  )ÚclsÚlookupÚsynchronize_sessionÚargÚklassr   r   r   Ú_factoryT  s   
úÿÿÿzBulkUD._factoryc                 C   s4   |   ¡  |  ¡  |  ¡  |  ¡  |  ¡  |  ¡  d S r   )Ú_do_before_compileÚ_do_preÚ_do_pre_synchronizeÚ_do_execÚ_do_post_synchronizeÚ_do_postre  r   r   r   Úexec_`  s   zBulkUD.exec_c                 C   s*   | j jj|| j j| jd�| _| jj| _d S )N)r    r(   )r"  r7   rè   Ú_paramsr(   r  rì   )rT  rÏ   r   r   r   Ú_execute_stmth  s   ÿzBulkUD._execute_stmtc                 C   s   t ƒ ‚r   )r9   re  r   r   r   rr  n  s   zBulkUD._do_before_compilezsqlalchemy.orm.queryc                 C   s¬   | j }| |¡| _t|jd |jƒr@tƒ }|jD ]}t||jƒs&| ¡   n	| t	|j
ƒ¡ qt|ƒdkr:t d¡‚| ¡ | _n| d¡jj| _|j}|jrT| ¡  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í   r_  r`  r»   Úprimary_tableÚ_only_entity_zeror(   Úlocal_tabler7   Ú
_autoflush)rT  Úquerylibr"  ÚtablesÚentr7   r   r   r   rs  q  s.   
ÿÿýÿzBulkUD._do_prec                 C   ó   d S r   r   re  r   r   r   rt  ”  ó   zBulkUD._do_pre_synchronizec                 C   r‡  r   r   re  r   r   r   rv  —  rˆ  zBulkUD._do_post_synchronizeN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__rU  rR  Úpropertyr7   Úclassmethodrq  rx  rz  rr  r   Údependenciesrs  rt  rv  r   r   r   r   rO  4  s    


"rO  c                   @   ó    e Zd ZdZdd„ Zdd„ ZdS )ÚBulkEvaluatezDBulkUD which does the 'evaluate' method of session state resolution.c                 C   r‡  r   r   )rT  Úevaluator_compilerr   r   r   Ú_additional_evaluatorsž  rˆ  z#BulkEvaluate._additional_evaluatorsc              
      s”   | j }| ¡ j‰zt ˆ¡}|jd ur| |j¡‰ ndd„ ‰ |  |¡ W n tjy8 } zt	 
d| ¡‚d }~ww ‡ ‡fdd„|jj ¡ D ƒ| _d S )Nc                 S   s   dS )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|‘qS r   )Ú
issubclass)r   rl  r  Úidentity_tokenr@  ©r”  Ú
target_clsr   r   r   ¸  s    ùùÿz4BulkEvaluate._do_pre_synchronize.<locals>.<listcomp>)r"  Ú_mapper_zeroÚclass_r   ÚEvaluatorCompilerÚwhereclauseÚprocessr“  ÚUnevaluatableErrorr_  r`  r7   rˆ   r<   Úmatched_objects)rT  r"  r’  Úerrr   r—  r   rt  ¡  s&   


þÿ€ÿ
úz BulkEvaluate._do_pre_synchronizeN)r‰  rŠ  r‹  rŒ  r“  rt  r   r   r   r   r‘  ›  s    r‘  c                   @   ó   e Zd ZdZdd„ ZdS )Ú	BulkFetchzABulkUD which does the 'fetch' method of session state resolution.c                 C   sB   | j }|j}| ¡ }|j | jj¡}|j|| j|j	d� 
¡ | _d S )N)r(   r    )r"  r7   Ú_compile_contextrö   Úwith_only_columnsr€  r  rè   r(   ry  ÚfetchallÚmatched_rows)rT  r"  r7   rê   Úselect_stmtr   r   r   rt  Æ  s   ÿ
ÿþzBulkFetch._do_pre_synchronizeN)r‰  rŠ  r‹  rŒ  rt  r   r   r   r   r¢  Ã  s    r¢  c                       s\   e Zd ZdZ‡ fdd„Zedd„ ƒZdd„ Zedd	„ ƒZ	ed
d„ ƒZ
dd„ Zdd„ Z‡  ZS )Ú
BulkUpdatezBulkUD which handles UPDATEs.c                    s    t t| ƒ |¡ || _|| _d S r   )Úsuperr¨  rU  rµ   Úupdate_kwargs)rT  r"  rµ   rª  ©Ú	__class__r   r   rU  Õ  s   
zBulkUpdate.__init__c                 C   s   t  tttdœ||||¡S ©N)ÚevaluateÚfetchF)rO  rq  ÚBulkUpdateEvaluateÚBulkUpdateFetchr¨  )rl  r"  rn  rµ   rª  r   r   r   ÚfactoryÚ  s   ý÷zBulkUpdate.factoryc                 C   ó<   | j jjr| j jjD ]}|| j | ƒ}|d ur|| _ q
d S d S r   )r"  r|   Úbefore_compile_update©rT  ÚfnÚ	new_queryr   r   r   rr  è  ó   
€üzBulkUpdate._do_before_compilec                 C   sš   g }t | jdƒr| j ¡ n| jD ]:\}}| jrCt|tjƒr,t| j|ƒ}| | 	|¡¡ qt|t
jƒr;| | 	|¡¡ q| ||f¡ q| ||f¡ q|S )Nr<   )r™   rµ   r<   r(   rš   r   Ústring_typesr
   r  Ú_bulk_update_tuplesr   ÚQueryableAttributerb   )rT  rµ   rP   rQ   Údescr   r   r   Ú_resolved_valuesï  s   
ÿ
ýzBulkUpdate._resolved_valuesc              	   C   sž   g }| j D ]G\}}t|tjƒr| |j|f¡ qt|dƒr!| ¡ }| jrFt|t	j
ƒrFz| jj| }W n
 tjy<   Y qw | |j|f¡ qt d| ¡‚|S )Nr”   zInvalid expression type: %r)r½  rš   r   r»  rb   r2   r™   r”   r(   r   ÚColumnElementr´   rŒ   ÚUnmappedColumnErrorr_  r`  )rT  rµ   rP   rQ   ra  r   r   r   Ú"_resolved_values_keys_as_propnames  s$   
ÿÿz-BulkUpdate._resolved_values_keys_as_propnamesc                 C   sH   | j }| j dd¡st|ƒ}tj| j| jj|fi | j¤Ž}|  	|¡ d S )NÚpreserve_parameter_orderF)
r½  rª  r­   r   r   rk   r€  rê   rœ  rz  )rT  rµ   rÒ   r   r   r   ru    s   ýüzBulkUpdate._do_execc                 C   ó   | j j}|j | ¡ d S r   )r"  r7   r|   Úafter_bulk_update©rT  r7   r   r   r   rw  )  ó   zBulkUpdate._do_post)r‰  rŠ  r‹  rŒ  rU  rŽ  r²  rr  r�  r½  rÀ  ru  rw  Ú__classcell__r   r   r«  r   r¨  Ò  s    


r¨  c                       sD   e Zd ZdZ‡ fdd„Zedd„ ƒZdd„ Zdd	„ Zd
d„ Z	‡  Z
S )Ú
BulkDeletezBulkUD which handles DELETEs.c                    s   t t| ƒ |¡ d S r   )r©  rÇ  rU  rS  r«  r   r   rU  1  s   zBulkDelete.__init__c                 C   s   t  tttdœ||¡S r­  )rO  rq  ÚBulkDeleteEvaluateÚBulkDeleteFetchrÇ  )rl  r"  rn  r   r   r   r²  4  s   ýùzBulkDelete.factoryc                 C   r³  r   )r"  r|   Úbefore_compile_deleterµ  r   r   r   rr  @  r¸  zBulkDelete._do_before_compilec                 C   s    t  | j| jj¡}|  |¡ d S r   )r   r€   r€  rê   rœ  rz  )rT  r  r   r   r   ru  G  s   zBulkDelete._do_execc                 C   rÂ  r   )r"  r7   r|   Úafter_bulk_deleterÄ  r   r   r   rw  L  rÅ  zBulkDelete._do_post)r‰  rŠ  r‹  rŒ  rU  rŽ  r²  rr  ru  rw  rÆ  r   r   r«  r   rÇ  .  s    
rÇ  c                   @   r�  )r°  zSBulkUD which handles UPDATEs using the "evaluate"
    method of session resolution.c                 C   s4   i | _ | j}|D ]\}}| t |¡¡| j |< qd S r   )Úvalue_evaluatorsrÀ  r�  r   Ú_literal_as_binds)rT  r’  rµ   r2   r§   r   r   r   r“  U  s   ÿÿz)BulkUpdateEvaluate._additional_evaluatorsc           	      C   s¶   | j j}tƒ }t| j ¡ ƒ}| jD ]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)	rT  r7   rG   Úevaluated_keysr@  r   r   Úto_evaluater2   r   r   r   rv  ]  s"   
þÿz'BulkUpdateEvaluate._do_post_synchronizeN)r‰  rŠ  r‹  rŒ  r“  rv  r   r   r   r   r°  Q  s    r°  c                   @   r¡  )rÈ  zSBulkUD which handles DELETEs using the "evaluate"
    method of session resolution.c                 C   s   | j j dd„ | jD ƒ¡ d S )Nc                 S   s   g | ]}t  |¡‘qS r   )r   r‰   )r   r@  r   r   r   r     r   z;BulkDeleteEvaluate._do_post_synchronize.<locals>.<listcomp>)r"  r7   Ú_remove_newly_deletedrŸ  re  r   r   r   rv  }  s   ÿz'BulkDeleteEvaluate._do_post_synchronizeN©r‰  rŠ  r‹  rŒ  rv  r   r   r   r   rÈ  y  ó    rÈ  c                   @   r¡  )r±  zPBulkUD which handles UPDATEs using the "fetch"
    method of session resolution.c                    s„   | j j‰ | j  ¡ ‰t‡ fdd„‡fdd„| jD ƒD ƒƒ}| j}tdd„ |D ƒƒ}|D ]}| |j¡}|r:ˆ  ||¡ q*ˆ  	|¡ d S )Nc                    s&   g | ]}|ˆ j v rt ˆ j | ¡‘qS r   )rˆ   r   r‰   )r   Úidentity_key)r7   r   r   r   Œ  s
    
ùz8BulkUpdateFetch._do_post_synchronize.<locals>.<listcomp>c                    s   g | ]	}ˆ   t|ƒ¡‘qS r   )Úidentity_key_from_primary_keyr:   )r   r  )Útarget_mapperr   r   r   Ž  s    ýÿÿc                 s   s   � | ]\}}|V  qd S r   r   rO   r   r   r   r)   ™  rh  z7BulkUpdateFetch._do_post_synchronize.<locals>.<genexpr>)
r"  r7   r™  r—   r¦  rÀ  r˜   r   Ú_expire_staterÓ  )rT  rG   rµ   Úattribr   Ú	to_expirer   )r7   rÛ  r   rv  ‡  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  ¡ }| jD ]}| t|ƒ¡}||jv r&| t 	|j| ¡g¡ qd S r   )
r"  r7   r™  r¦  rÚ  r:   rˆ   rÖ  r   r‰   )rT  r7   rÛ  r  rÙ  r   r   r   rv  ¥  s   

ÿ
ÿÿÿ€ùz$BulkDeleteFetch._do_post_synchronizeNr×  r   r   r   r   rÉ  ¡  rØ  rÉ  )F)FFF)T)6rŒ  Ú	itertoolsr   r   rK  Ú r   r   r   rŒ   r   r	   Úbaser
   r   r_  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_   ÚobjectrO  r‘  r¢  r¨  rÇ  r°  rÈ  r±  rÉ  r   r   r   r   Ú<module>   sp   	P
EV-.Q
ûZ
ÿ (8.
ù F
ù~s\F)L		g(\#(
