U
    ÌZjŒ  ã                   @   s0  d Z ddl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 e dd¡Zdd„ Zd:dd„Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd;d(d)„Zd<d*d+„ZG d,d-„ d-ejƒZd.d/„ Zd0d1„ Zd2d3„ Z d4d5„ Z!d6d7„ Z"d8d9„ Z#dS )=ziFunctions used by compiler.py to determine the parameters rendered
within INSERT and UPDATE statements.

é    Né   )Ú	coercions)Údml)Úelements)Úroles)ÚSelecté   )Úexc)ÚutilÚREQUIREDa8  
Placeholder for the value within a :class:`.BindParameter`
which is required to be present when the statement is passed
to :meth:`_engine.Connection.execute`.

This symbol is typically used when a :func:`_expression.insert`
or :func:`_expression.update` statement is compiled without parameter
values present.

c                    s8  g ˆ_ g ˆ_g ˆ_g ˆ_tˆ||ƒ \‰ }}}|ˆ_ˆjdkrZ|jrZ‡fdd„|jjD ƒS |j	rx|j
d ‰tˆ ¡ ƒ}n6|jrŒ|j‰|j}n"|jr¦|j‰tˆ ¡ ƒ}nd }‰ˆjdkr¾i }n8|rÞt‡ ‡fdd„ˆjD ƒƒ}nt‡ fdd„ˆjD ƒƒ}g }	|dk	�rtˆ|||ˆ |	|ƒ i }
|j�rD|j�rDtˆ||||
|||	|ƒ	 |j�rp|j�rptˆ||||ˆ ||
|	|ƒ
 ntˆ||||ˆ ||
|	|ƒ
 |�rÜ|�rÜt|ƒ ‡ fdd„|D ƒ¡ |
¡}|�rÜt d	d
 dd„ |D ƒ¡ ¡‚|j	�røtˆ|||	ˆ |ƒ}	n<|	�s4ˆj�r4ˆj j!�r4|jjd ˆj" #|jjd ¡dfg}	|	S )aQ  create a set of tuples representing column/string pairs for use
    in an INSERT or UPDATE statement.

    Also generates the Compiled object's postfetch, prefetch, and
    returning column collections, used for default handling and ultimately
    populating the CursorResult's prefetch_cols() and postfetch_cols()
    collections.

    Nc              
      s*   g | ]"}|ˆ j  |¡tˆ |d dd�f‘qS )NT)Úrequired)ÚpreparerÚformat_columnÚ_create_bind_param©Ú.0Úc)Úcompiler© úV/var/www/html/TRUCKING_PROJECT/venv/lib/python3.8/site-packages/sqlalchemy/sql/crud.pyÚ
<listcomp>D   s
   ü
ýz$_get_crud_params.<locals>.<listcomp>r   c                 3   s"   | ]}|ˆkrˆ |ƒt fV  qd S ©N©r   ©r   Úkey)Ú_column_as_keyÚspdr   r   Ú	<genexpr>^   s   þz#_get_crud_params.<locals>.<genexpr>c                 3   s   | ]}ˆ |ƒt fV  qd S r   r   r   ©r   r   r   r   d   s    c                 3   s   | ]\}}ˆ |ƒV  qd S r   r   )r   ÚkÚvr   r   r   r   ¥   s     zUnconsumed column names: %sú, c                 s   s   | ]}d |f V  qdS )z%sNr   r   r   r   r   r   «   s     ÚDEFAULT)$Ú	postfetchÚinsert_prefetchÚupdate_prefetchÚ	returningÚ_key_getters_for_crud_columnZcolumn_keysZ_no_parametersÚtableÚcolumnsÚ_has_multi_parametersÚ_multi_parametersÚlistÚitemsZ_ordered_valuesZ_dict_parametersÚdictÚ!_get_stmt_parameter_tuples_paramsÚisupdateZis_multitableÚ_get_update_multitable_paramsÚisinsertÚ_select_namesÚ_scan_insert_from_select_colsÚ
_scan_colsÚsetÚintersectionÚ
differencer	   ÚCompileErrorÚjoinÚ_extend_values_for_multiparamsÚfor_executemanyÚdialectZsupports_default_metavaluer   r   )r   ÚstmtÚcompile_stateÚkwÚ_getattr_col_keyÚ_col_bind_nameÚgettersÚstmt_parameter_tuplesÚ
parametersÚvaluesÚcheck_columnsÚcheckr   )r   r   r   r   Ú_get_crud_params&   sê    	
ü
ú	

þ
ÿ
ù
÷ööÿþÿÿÿú	ÿþý
ýÿrI   TFc                 K   s>   |d kr|j }tj|||j|d�}d|_|r:|j| f|Ž}|S )N)Útype_r   T)r   r   ÚBindParameterÚtypeZ_is_crudÚ_compiler_dispatch)r   ÚcolÚvalueÚprocessr   Únamer@   Ú	bindparamr   r   r   r   Ë   s       ÿr   c                 K   sR   d|k}|s0|j r0t|jtjƒr0|| jd|jf< |jjrD| |j¡}|j	| f|ŽS )NZvisiting_cterR   )
ÚuniqueÚ
isinstancer   r   Z_truncated_labelZtruncated_namesrL   Ú_isnullÚ_with_binary_element_typerM   )r   rN   rO   rQ   r@   Zis_cter   r   r   Ú_handle_values_anonymous_paramÙ   s    ÿþýrW   c                    sv   |j rN|jrNt|jƒ‰ t tjtj¡‰‡ ‡fdd„}‡ fdd„}‡ fdd„}nt tjtj¡}t	 
d¡ }}|||fS )Nc                    s0   ˆ| ƒ}t | dƒr(| jˆ kr(| jj|fS |S d S )Nr(   )Úhasattrr(   rQ   )r   Zstr_key©Ú_etZ
c_key_roler   r   r     s    z4_key_getters_for_crud_column.<locals>._column_as_keyc                    s"   | j ˆ kr| j j| jfS | jS d S r   ©r(   rQ   r   ©rN   ©rZ   r   r   rA     s    
z6_key_getters_for_crud_column.<locals>._getattr_col_keyc                    s&   | j ˆ krd| j j| jf S | jS d S )Nz%s_%sr[   r\   r]   r   r   rB     s    
z4_key_getters_for_crud_column.<locals>._col_bind_namer   )r0   Ú_extra_fromsr6   Ú	functoolsÚpartialr   Zexpect_as_keyr   ÚDMLColumnRoleÚoperatorÚ
attrgetter)r   r>   r?   r   rA   rB   r   rY   r   r'   ÿ   s    
 ÿ ÿr'   c
                    s\  t | ˆ|ƒ\}
}}}‡ ‡fdd„ˆjD ƒ}| jd d ˆks@t‚ˆj| jd d< g }ˆjrˆt|ƒ}ˆjjD ]}||krj|j	rj| 
|¡ qj|D ]P}||ƒ}||krÌ||krÌ| |¡ | 
|| j |¡d f¡ qŒt| ˆ|||	ƒ qŒ|�rX| |¡ | jd d }t|tƒ�s(t dd dd	„ |D ƒ¡f ¡‚| ¡ }t|jƒd
d„ |D ƒ |_|| jd d< d S )Nc                    s   g | ]}ˆj jˆ |ƒ ‘qS r   )r(   r   )r   rQ   ©r   r>   r   r   r   =  s     z1_scan_insert_from_select_cols.<locals>.<listcomp>éÿÿÿÿÚ
selectableZinsert_from_selectzßCan't extend statement for INSERT..FROM SELECT to include additional default-holding column(s) %s.  Convert the selectable to a subquery() first, or pass include_defaults=False to Insert.from_select() to skip these columns.r!   c                 s   s   | ]\}}}t |ƒV  qd S r   )Úrepr)r   Ú_r   r   r   r   r   ^  s     z0_scan_insert_from_select_cols.<locals>.<genexpr>c                 S   s   g | ]\}}}|‘qS r   r   )r   rN   Úcol_exprÚexprr   r   r   r   b  s    )Ú_get_returning_modifiersr3   ÚstackÚAssertionErrorÚselectZ#include_insert_from_select_defaultsr6   r(   r)   ÚdefaultÚappendÚpopr   r   Ú&_append_param_insert_select_hasdefaultÚextendrT   r   r	   r9   r:   Z	_generater,   Z_raw_columns)r   r>   r?   rE   rA   r   rB   rG   rF   r@   Úneed_pksÚimplicit_returningÚimplicit_return_defaultsÚpostfetch_lastrowidÚcolsZadd_select_colsZcol_setrN   r   Úcol_keyZins_from_selectr   rd   r   r4   )  sR    
û
    ÿ
ûÿÿ
r4   c
                    s¨  t | ˆ|ƒ\}
}}}|jr`‡ fdd„|jD ƒ}t|ƒ‰‡fdd„|D ƒ‡fdd„ˆjjD ƒ }nˆjj}|D �]4}||ƒ}||kr¨||kr¨t| ˆ|||||||||	ƒ ql|j�rˆ|jrâ|
râ|rÐt	| ˆ|||	ƒ nt
| ˆ|||	ƒ n¤|jd k	�rt| ˆ||||	ƒ n„|jd k	�rB|�r,||k�r,| j |¡ n|j�s†| j |¡ nD|�r`||k�r`| j |¡ n&|j�r¢|ˆjjk	�r¢|j�s¢t|ƒ ql|jrlt| |ˆ||||	ƒ qld S )Nc                    s   g | ]}ˆ |ƒ‘qS r   r   r   r   r   r   r   |  s    z_scan_cols.<locals>.<listcomp>c                    s0   g | ](}t |tjƒr|ˆ jjkrˆ jj| ‘qS r   )rT   r
   Ústring_typesr(   r   r   )r>   r   r   r   €  s    þc                    s   g | ]}|j ˆ kr|‘qS r   )r   r   )Úordered_keysr   r   r   „  s     
 )rk   Z_parameter_orderingr6   r(   r   r)   Ú_append_param_parameterr2   Úprimary_keyÚ!_append_param_insert_pk_returningÚ$_append_param_insert_pk_no_returningro   Ú_append_param_insert_hasdefaultÚserver_defaultr&   rp   r#   Ú_autoincrement_columnÚnullableÚ"_warn_pk_with_no_anticipated_valuer0   Ú_append_param_update)r   r>   r?   rE   rA   r   rB   rG   rF   r@   rt   ru   rv   rw   Zparameter_orderingrx   r   ry   r   )r   r{   r>   r   r5   h  s¦    
û
ÿ
þü
õ
    ÿ    ÿ     ÿÿ
þý
ùr5   c                 C   s8  |  |¡}| jj||jd�}t |¡r\t| ||f|tk|jsD||ƒn
d||ƒ dœ|
—Ž}nÈ|j	r’t
| ||fd|js|||ƒn
d||ƒ i|
—Ž}n’| j| ¡ f|
Ž}|jrÒ|rÄ||krÄ| j |¡ n| j |¡ nR|jrú|rê| j |¡ n| jjrød| _n*|�r||k�r| j |¡ n| j |¡ |	 |||f¡ d S )N©Z	use_tablez%s_m0©r   rQ   rQ   T)rq   r   r   Úinclude_table_with_column_exprsr   Ú_is_literalr   r   r*   Ú_is_bind_parameterrW   rP   Ú
self_groupr0   r&   rp   r#   r}   r=   rw   )r   r>   r?   r   ry   rE   rB   ru   rv   rF   r@   rO   Ú	col_valuer   r   r   r|   à  sX    
 ÿ
ýÿ

ùø
ýÿ

úù	r|   c                 C   sö   |j dk	rÀ|j jr\| jjrN|j jr*| jjsN| || j |¡| j	|j f|Žf¡ | j
 |¡ qò|j jrœ| || j |¡| j	|j j ¡ f|Žf¡ | j
 |¡ qò| || j |¡t| |f|Žf¡ n2||jjksÖ|jdk	rä| j
 |¡ n|jsòt|ƒ dS )z—Create a primary key expression in the INSERT statement where
    we want to populate result.inserted_primary_key and RETURNING
    is available.

    N)ro   Úis_sequencer=   Úsupports_sequencesÚoptionalÚsequences_optionalrp   r   r   rP   r&   Úis_clause_elementÚargr‹   Ú"_create_insert_prefetch_bind_paramr(   r‚   r�   rƒ   r„   ©r   r>   r   rF   r@   r   r   r   r~   )  s@    
ÿþ
ýÿ
ýÿ
ýÿr~   c                 C   sÐ   |j dk	r*|j jrj| jjr*|j jrj| jjrj||jjkrŽ| jjsŽ|j dk	rX|j jrX| jjsj|j dkrŽ| jj	rŽ| 
|| j |¡t| |f|Žf¡ n>|j dkr¾|jdkr¾|js¾||jjk	r¾t|ƒ n| jjrÌd| _dS )ao  Create a primary key expression in the INSERT statement where
    we want to populate result.inserted_primary_key and we cannot use
    RETURNING.

    Depending on the kind of default here we may create a bound parameter
    in the INSERT statement and pre-execute a default generation function,
    or we may use cursor.lastrowid if supported by the dialect.


    NT)ro   r�   r=   rŽ   r�   r�   r(   r‚   rw   Z"preexecute_autoincrement_sequencesrp   r   r   r“   r�   rƒ   r„   r”   r   r   r   r   Y  sR    þúø
öõ
ïìèçæ!ß"Þ(
ýÿÿþý
ü
r   c                 C   sø   |j jrr| jjrô|j jr | jjsô| || j |¡| j	|j f|Žf¡ |r^||kr^| j
 |¡ qô|jsô| j |¡ n‚|j jrÒ| || j |¡| j	|j j ¡ f|Žf¡ |r¾||kr¾| j
 |¡ qô|jsô| j |¡ n"| || j |¡t| |f|Žf¡ d S r   )ro   r�   r=   rŽ   r�   r�   rp   r   r   rP   r&   r}   r#   r‘   r’   r‹   r“   )r   r>   r   rv   rF   r@   r   r   r   r€   ¥  sB    ÿÿ
ýÿ
ýÿ
ýÿr€   c                 C   s˜   |j jr@| jjr”|j jr | jjs”| || j |¡|j  	¡ f¡ nT|j j
rj| || j |¡|j j ¡ f¡ n*| || j |¡t| |fddi|—Žf¡ d S )NrP   F)ro   r�   r=   rŽ   r�   r�   rp   r   r   Z
next_valuer‘   r’   r‹   r“   r”   r   r   r   rr   Ï  s4    ÿÿÿÿ
 ÿÿÿýÿrr   c                 C   sø   |j }|jd k	rž|jjsž|jjrv| || jj||d�| j|jj 	¡ f|Žf¡ |rh||krh| j
 |¡ qœ| j |¡ qô| || jj||d�t| |f|Žf¡ nV|jd k	rÐ|rÂ||krÂ| j
 |¡ qô| j |¡ n$|rô|jsà|jsô||krô| j
 |¡ d S )Nr†   )rˆ   Úonupdater�   r‘   rp   r   r   rP   r’   r‹   r&   r#   Ú"_create_update_prefetch_bind_paramÚserver_onupdateÚ_return_defaults_columnsÚ_return_defaults)r   r?   r>   r   rv   rF   r@   Úinclude_tabler   r   r   r…   è  sJ    þúÿ
þúÿ

ÿþþýr…   c                 K   s*   t | |d f||dœ|—Ž}| j |¡ |S ©N)rP   rQ   )r   r$   rp   ©r   r   rP   rQ   r@   Úparamr   r   r   r“     s      ÿ ÿÿr“   c                 K   s*   t | |d f||dœ|—Ž}| j |¡ |S r›   )r   r%   rp   rœ   r   r   r   r–      s      ÿ ÿÿr–   c                   @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )Ú_multiparam_columnTc                 C   s4   || _ d|j|d f | _|| _|j| _|j| _d S )Nú%s_m%dr   )Úindexr   Úoriginalro   rL   )Úselfr¡   r    r   r   r   Ú__init__-  s
    z_multiparam_column.__init__c                 K   s
   t ƒ ‚d S r   ©ÚNotImplementedError©r¢   Úotherr@   r   r   r   Úcompare4  s    z_multiparam_column.comparec                 K   s
   t ƒ ‚d S r   r¤   r¦   r   r   r   Ú_copy_internals7  s    z"_multiparam_column._copy_internalsc                 C   s"   t |tƒo |j| jko |j| jkS r   )rT   rž   r   r¡   )r¢   r§   r   r   r   Ú__eq__:  s
    

ÿ
ýz_multiparam_column.__eq__N)Ú__name__Ú
__module__Ú__qualname__Z_is_multiparam_columnr£   r¨   r©   rª   r   r   r   r   rž   *  s
   rž   c                 C   s‚   |j st d| ¡‚nh|j jr4| j|j j ¡ f|ŽS |j jrL| j|j f|ŽS t||ƒ}t	|t
jƒrpt| |f|ŽS t| |f|ŽS d S )Nz�INSERT value for column %s is explicitly rendered as a boundparameter in the VALUES clause; a Python-side value or SQL expression is required)ro   r	   r9   r‘   rP   r’   r‹   r�   rž   rT   r   ZInsertr“   r–   )r   r>   r   r    r@   rN   r   r   r   Ú _process_multiparam_default_bindB  s    þÿ	
r®   c	                 C   s´  t dd„ |D ƒƒ}	|j}
tƒ }|jD ]Ä}|jD ]¸}||	kr.| |¡ ||||ƒ< |	| }| j||
d�}t |¡r”t	| ||f|t
k||ƒdœ|—Ž}nB|jr¸t| ||fd||ƒi|—Ž}n| j |¡ | j| ¡ f|Ž}| |||f¡ q.q$|D ]À}|jD ]´}||	k�r
qøqø|jd k	�r–|jj�s–|jj�rd| || j||
d�| j|jj ¡ f|Žf¡ | j |¡ n0| || j||
d�t| |fd||ƒi|—Žf¡ qø|jd k	rø| j |¡ qøqîd S )Nc                 s   s$   | ]\}}t  tj|¡|fV  qd S r   )r   Úexpectr   ra   )r   r   r�   r   r   r   r   h  s   ÿz0_get_update_multitable_params.<locals>.<genexpr>©rš   r‡   rQ   )r.   rˆ   r6   r^   r   ÚaddrP   r   r‰   r   r   rŠ   rW   r#   rp   r‹   r•   r�   r‘   r’   r–   r—   )r   r>   r?   rD   rG   rB   rA   rF   r@   Znormalized_paramsrš   Zaffected_tablesÚtr   rO   rŒ   r   r   r   r1   ]  sˆ    þ



ýûú  ÿÿÿ



ÿÿýÿ	 ÿÿÿýÿ	
r1   c              	      sà   |}|g}t |jdd … ƒD ]¾\}}g }	‡ fdd„| ¡ D ƒ}|D ]Œ\}
}}|
j|kr®|
j}t || ¡r–t| |
|| fdd|
j|d f i|—Ž}q¾| j||  ¡ f|Ž}nt	| ||
||ƒ}|	 
|
||f¡ qB| 
|	¡ q|S )Nr   c                    s   i | ]\}}ˆ |ƒ|“qS r   r   )r   r   r    r   r   r   Ú
<dictcomp>·  s      z2_extend_values_for_multiparams.<locals>.<dictcomp>rQ   rŸ   )Ú	enumerater+   r-   r   r   r‰   r   rP   r‹   r®   rp   )r   r>   r?   rF   r   r@   Zvalues_0ÚiÚrowÚ	extensionrN   ri   r�   r   Ú	new_paramr   r   r   r;   ©  s:    
ýüû    ÿr;   c                 C   s¤   |D ]š\}}||ƒ}	|	d k	r*|  |	|¡ q| j||jd�}
t |¡rb| jtjd ||jd�f|Ž}n,|jr||jj	r|| 
|j¡}| j| ¡ f|Ž}| ||
|f¡ qd S )Nr°   )rJ   )Ú
setdefaultrP   rˆ   r   r‰   r   rK   rL   rŠ   rU   rV   r‹   rp   )r   r?   rE   rD   r   rF   r@   r   r    Zcolkeyri   r   r   r   r/   Ó  s$    
	 ÿ
ÿÿr/   c                 C   s¶   |j o2|j o2| j s$| jjo2|jo2|j o2|j }|oF| jjoF|j	j}|j rZ|oV|j}n"|j
rx| jjot|j	jot|j}nd}|rž|js”t|j	jƒ}n
t|jƒ}|o¨| jj}||||fS )NF)r2   Z_inliner<   r=   Zinsert_executemany_returningr™   Z
_returningr*   ru   r(   r0   r˜   r6   r   rw   )r   r>   r?   rt   ru   rv   rw   r   r   r   rk   ý  sF    ÿú	÷
õÿýÿý
ürk   c                 C   s>   d| j j| j| j jf }t| j jƒdkr0|d7 }t |¡ d S )Na  Column '%s.%s' is marked as a member of the primary key for table '%s', but has no Python-side or server-side default generator indicated, nor does it indicate 'autoincrement=True' or 'nullable=True', and no explicit value is passed.  Primary key columns typically may not store NULL.r   a0   Note that as of SQLAlchemy 1.1, 'autoincrement=True' must be indicated explicitly for composite (e.g. multicolumn) primary keys if AUTO_INCREMENT/SERIAL/IDENTITY behavior is expected for one of the columns in the primary key. CREATE TABLE statements are impacted by this change as well on most backends.)r(   ÚfullnamerQ   Úlenr}   r
   Úwarn)r   Úmsgr   r   r   r„   0  s    úÿ	ÿr„   )TFN)TN)TN)$Ú__doc__r_   rb   Ú r   r   r   r   rf   r   r	   r
   Úsymbolr   rI   r   rW   r'   r4   r5   r|   r~   r   r€   rr   r…   r“   r–   ZColumnElementrž   r®   r1   r;   r/   rk   r„   r   r   r   r   Ú<module>   sT   þ '     ÿ
&*?xI0L*.   ÿ
   ÿ

L**3