o [h!e@s$dZddlZddlmZddlmZddlmZddlmZed d Zed Z ed Z ed Z ddZ ddZ d:ddZddZddZddZddZddZd;d d!Zd;d"d#ZGd$d%d%ejZd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Z d8d9Z!dS)[sz$_get_crud_params..rc3s(|]}r |vr|tfVqdSN)rrkey)_column_as_keystmt_parametersrr ss z#_get_crud_params..c3s|]}|VqdSrr)rkr rrr"zUnconsumed column names: %sz, css|]}d|VqdS)z%sNrrrrrr"r%) postfetchinsert_prefetchupdate_prefetch returningZ column_keys parameterstablecolumns_has_multi_parameters_key_getters_for_crud_columndict_get_stmt_parameters_paramsr _extra_froms_get_multitable_paramsr select_names_scan_insert_from_select_cols _scan_colsset intersection differencer CompileErrorjoin_extend_values_for_multiparams) rrr_getattr_col_key_col_bind_namer*values check_columnscheckr)r rr!rrHs         rTFcKsB|dur|j}tj|||j|d}d|_|r|j|fi|}|S)N)type_rT)rr BindParametertypeZ_is_crudZ_compiler_dispatch)rcolvalueprocessrnamerZ bindparamrrrrs rcsZ|jr|jrt|jfdd}fdd}fdd}n tj}td}}|||fS)Ncs.t|}t|dr|jvr|jj|fS|S)Nr+)rr hasattrr+rG)rZstr_keyZ_etrrr s  z4_key_getters_for_crud_column.._column_as_keycs|jvr |jj|jfS|jSrr+rGrrDrIrrr<s z6_key_getters_for_crud_column.._getattr_col_keycs"|jvrd|jj|jfS|jS)Nz%s_%srJrKrIrrr=s z4_key_getters_for_crud_column.._col_bind_namer)r r1r6rr operator attrgetter)rrr r<r=rrIrr.s     r.c st|\} } } } fddjD} j|_g}jr4t| }jjD]}||vr3|jr3| |q%| D]#}||}||vrQ||vrQ| || |dfq6t ||||q6|rz| ||j |_t|jjtdd|D|j_dSdS)Ncsg|] }jj|qSrr+r)rrGr rrrrsz1_scan_insert_from_select_cols..css|]\}}|VqdSrr)rrDexprrrrr"r%z0_scan_insert_from_select_cols..)_get_returning_modifiersr3selectZ_insert_from_selectZ#include_insert_from_select_defaultsr6r+r,defaultappendpop&_append_param_insert_select_hasdefaultextendZ _generatetupleZ _raw_columns)rrr*r<r r=r?r>rneed_pksimplicit_returningimplicit_return_defaultspostfetch_lastrowidcolsZadd_select_colsZcol_setrDrcol_keyrrOrr4s@       r4c  st|\} } } } jr/fddjD} t| fdd| DfddjjD}njj}|D]}||}||vrQ||vrQt|||||| | || q5|jr|jrw| rw| sc| rc|jj urw| rnt ||||q5t ||||q5|j durt ||| ||q5|jdur| r|| vr|j|q5|js|j|q5| r|| vr|j|q5|jr|jj ur|jst|q5|jrt||| ||q5dS)Ncsg|]}|qSrrrr$rrr2sz_scan_cols..csg|]}jj|qSrrNr)rrrr6scsg|] }|jvr|qSr)rr) ordered_keysrrr6s)rQZ_parameter_orderingr6r+rr,_append_param_parameterr primary_key_autoincrement_column!_append_param_insert_pk_returning_append_param_insert_pkrS_append_param_insert_hasdefaultserver_defaultr)rTr&nullable"_warn_pk_with_no_anticipated_valuer _append_param_update)rrr*r<r r=r?r>rrYrZr[r\Zparameter_orderingr]rr^r)r r_rrr5s             r5c Cs||} t| r%t||| f| tu|js||nd||d| } nUt| tjr7| jj r7| } |j| _|j rN|rN|j ||j| fi| } n,|rf||vrf|j ||j| fi| } n|j so|j ||j| fi| } | || fdS)Nz%s_m0rrG)rUr _is_literalrrr- isinstancerBrCZ_isnullZ_clonerar)rTrF self_groupr&) rrrr^r*r=rZr[r>rrErrrr`zs6        r`cCs|jdurX|jjr.|jjr&|jjr|jjs&|j|jfi|}|||f|j|dS|jj rL|||j|jj fi|f|j|dS||t ||fdS||j jusc|jdurk|j|dS|jstt|dSdS)aCreate a primary key expression in the INSERT statement and possibly a RETURNING clause for it. If the column has a Python-side default, we will create a bound parameter for it and "pre-execute" the Python function. If the column has a SQL expression default, or is a sequence, we will add it directly into the INSERT statement and add a RETURNING element to get the new value. If the column has a server side default or is marked as the "autoincrement" column, we will add a RETRUNING element to get at the value. If all the above tests fail, that indicates a primary key column with no noted default generation capabilities that has no parameter passed; raise an exception. N)rS is_sequencedialectsupports_sequencesoptionalsequences_optionalrFrTr)is_clause_elementargrm"_create_insert_prefetch_bind_paramr+rbrfrgrhrrrr>rprocrrrrcs*  rccC"t||d||d}|j||SN)rFrG)rr'rTrrrFrGparamrrrru rucCrxry)rr(rTrzrrr"_create_update_prefetch_bind_paramr|r}c@s eZdZdZddZddZdS)_multiparam_columnTcCs4||_d|j|df|_||_|j|_|j|_dS)N%s_m%dr)indexroriginalrSrC)selfrrrrr__init__s  z_multiparam_column.__init__cCs"t|to|j|jko|j|jkSr)rlr~rr)rotherrrr__eq__s   z_multiparam_column.__eq__N)__name__ __module__ __qualname__Z_is_multiparam_columnrrrrrrr~s r~cCs`|js td||jjr|j|jjfi|St||}t|t j r+t ||St ||S)NzINSERT value for column %s is explicitly rendered as a boundparameter in the VALUES clause; a Python-side value or SQL expression is required) rSrr9rsrFrtrmr~rlrZInsertrur})rrrrrrDrrr _process_multiparam_default_binds    rcCs|jdur |jjr|jjs||jjur'|jjs|jjr'||t||fdS|jdur:|j dur<|j s>t |dSdSdSdS)aCreate a bound parameter in the INSERT statement to receive a 'prefetched' default value. The 'prefetched' value indicates that we are to invoke a Python-side default function or expliclt SQL expression before the INSERT statement proceeds, so that we have a primary key value available. if the column has no noted default generation capabilities, it has no value passed in either; raise an exception. N) rSrnrorpr+rbZ"preexecute_autoincrement_sequencesrTrurfrgrh)rrrr>rrrrrds    rdcCs|jjr@|jjr:|jjr|jjs<|j|jfi|}|||f|r/||vr/|j|dS|j s>|j |dSdSdSdS|jj rs|j|jj fi|}|||f|rf||vrf|j|dS|j sq|j |dSdS||t||fdSr)rSrnrorprqrrrFrTr)rar&rsrtrmru)rrrr[r>rrwrrrre&s.  recCs|jjr"|jjr|jjr|jjs |j}|||fdSdSdS|jjr5|jj }|||fdS||t ||ddfdS)NF)rF) rSrnrorprqrrrTZ next_valuersrtrmrurvrrrrVAs rVcCs|jdurA|jjsA|jjr5|||j|jjfi|f|r-||vr-|j|dS|j|dS||t ||fdS|j dur\|rT||vrT|j|dS|j|dS|ro|j durq||vrs|j|dSdSdSdS)NT) onupdaternrsrTrFrtrmr)r&r}server_onupdate_return_defaults)rrrr[r>rrrrriRs*    ric CsTtdd|D}t} |jD]F} | jD]@} | |vrV| | | ||| <|| } t| r>t|| | | t u|| d} n|j | |j | fi|} | | | fqq| D]M} | jD]G} | |vrfq_| jdur| jjs| jjr| | |j | jj fi|f|j | q_| | t|| || dfq_| jdur|j | q_qZdS)Ncss"|] \}}t||fVqdSr)rZ_clause_element_as_expr)rrr{rrrr"ys  z)_get_multitable_params..rj)rG)r/itemsr6r1raddrrkrrr&rTrFrmrrnrsrtr}r) rrr!r?r=r<r>rZnormalized_paramsZaffected_tablestrrErrrr2nsf            r2c Cs|}|g}t|jddD]^\}}g}|D]P\}} ||vs#|j|vrW||vr)|n|j} t|| rIt|||| fdd|j|dfi|} n|j|| fi|} nt|||||} | || fq| |q|S)NrrGr) enumerater*rrrkrrFrmrrT) rrr>rZvalues_0irow extensionrDr{r new_paramrrrr;s0   r;c Cs|D];\}}||}|dur|||qt|r-|jtjd||jdfi|}n |j|fi|}|||fqdS)N)rA) r setdefaultrrkrFrBrCrmrT) rr*r!r r>rr#vZcolkeyrrrr0s r0cCs|jo|j o|j o|j }|o|jjo|jj}|jr"|o |j}n|jr1|jjo/|jjo/|j}nd}|rF|jdurAt |jj }nt |j}|oK|jj }||||fS)NFT) r inlineZ _returningr-rorZr+rr r6rr\)rrrYrZr[r\rrrrQs>    rQcCs>d|jj|j|jjf}t|jjdkr|d7}t|dS)NaColumn '%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.ra0 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+fullnamerGrrarwarn)rmsgrrrrhs rh)TFN)TN)"__doc__rLrrrrsymbolrr r r rrrr.r4r5r`rcrur}Z ColumnElementr~rrdrerVrir2r;r0rQrhrrrrsF       !u $0\. ) %@  +