o znhd@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)/usr/local/lib/python3.10/dist-packages/sqlalchemy/sql/crud.py_setup_crud_params'sF    rc s~g_g_g_g_jdur!|jdur!fdd|jjDS|jr*|jdn|jt |\}}jdur=i}n t fddjD}g}durYt |||i}j rn|j rnrnt||||||jr|jrt||||||| n t||||||| |rrt|fddD|}|rtdd d d|D|jrt|||}|S) aPcreate 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 ResultProxy's prefetch_cols() and postfetch_cols() collections. Nc s g|] }|t|dddfqS)NT)required)_create_bind_param.0c)rrr Ysz$_get_crud_params..rc3s(|]}r |vr|tfVqdSN)rr key)_column_as_keystmt_parametersrr qs z#_get_crud_params..c3s|]}|VqdSr#r)r kr&rrr(zUnconsumed column names: %sz, css|]}d|VqdS)z%sNrrrrrr(r+) postfetchinsert_prefetchupdate_prefetch returning 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_namer1values check_columnscheckr)r&rr'rrFs         rTFcKsB|dur|j}tj|||j|d}d|_|r|j|fi|}|S)N)type_rT)r%r BindParametertype_is_crud_compiler_dispatch)rcolvalueprocessrnamer bindparamrrrrs rcsZ|jr|jrt|jfdd}fdd}fdd}n tj}td}}|||fS)Ncs.t|}t|dr|jvr|jj|fS|S)Nr2)rr&hasattrr2rP)r%str_key_etrrr&s  z4_key_getters_for_crud_column.._column_as_keycs|jvr |jj|jfS|jSr#r2rPr%rMrTrrrCs z6_key_getters_for_crud_column.._getattr_col_keycs"|jvrd|jj|jfS|jS)Nz%s_%srVrWrTrrrDs z4_key_getters_for_crud_column.._col_bind_namer%)r r8r=rr&operator attrgetter)rrr&rCrDrrTrr5s     r5c st|\} } } } fddjD} j|_g}jr4t| }jjD]}||vr3|jr3| |q%| D]#}||}||vrQ||vrQ| || |dfq6t ||||q6|rz| ||j |_t|jjtdd|D|j_dSdS)Ncsg|] }jj|qSrr2r!)r rPr&rrrr"sz1_scan_insert_from_select_cols..css|]\}}|VqdSr#r)r rMexprrrrr(r+z0_scan_insert_from_select_cols..)_get_returning_modifiersr:select_insert_from_select#include_insert_from_select_defaultsr=r2r3defaultappendpop&_append_param_insert_select_hasdefaultextend _generatetuple _raw_columns)rrr1rCr&rDrFrErneed_pksimplicit_returningimplicit_return_defaultspostfetch_lastrowidcolsadd_select_colscol_setrMr!col_keyrr[rr;s@       r;c  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|]}|qSrrr$r*rrr"0sz_scan_cols..csg|]}jj|qSrrZr$)rrrr"4scsg|] }|jvr|qSr)r%r) ordered_keysrrr"4s)r]_parameter_orderingr=r2r!r3_append_param_parameterr primary_key_autoincrement_column!_append_param_insert_pk_returning_append_param_insert_pkra_append_param_insert_hasdefaultserver_defaultr/rbr,nullable"_warn_pk_with_no_anticipated_valuer _append_param_update)rrr1rCr&rDrFrErrirjrkrlparameter_orderingrmr!rpr)r&rqrrr<s             r<c 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_m0rrP)rcr _is_literalrrr4 isinstancerIrJ_isnull_clonertr/rbrO self_groupr,) rrr!rpr1rDrjrkrErrNrrrrsxs6        rscCs|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)ra is_sequencedialectsupports_sequencesoptionalsequences_optionalrOrbr/is_clause_elementargr"_create_insert_prefetch_bind_paramr2ruryrzr{rrr!rErprocrrrrvs*  rvcC"t||d||d}|j||SN)rOrP)rr-rbrr!rOrPparamrrrr rcCrr)rr.rbrrrr"_create_update_prefetch_bind_paramrrc@s eZdZdZddZddZdS)_multiparam_columnTcCs4||_d|j|df|_||_|j|_|j|_dS)N%s_m%dr)indexr%originalrarJ)selfrrrrr__init__s  z_multiparam_column.__init__cCs"t|to|j|jko|j|jkSr#)rrr%r)rotherrrr__eq__s   z_multiparam_column.__eq__N)__name__ __module__ __qualname___is_multiparam_columnrrrrrrrs rcCs`|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) rarr@rrOrrrrrInsertrr)rrr!rrrMrrr _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) rarrrr2ru"preexecute_autoincrement_sequencesrbrryrzr{)rrr!rErrrrrws    rwcCs|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#)rarrrrrrOrbr/rtr,rrrr)rrr!rkrErrrrrrx$s.  rxcCs|jjr"|jjr|jjr|jjs |j}|||fdSdSdS|jjr5|jj }|||fdS||t ||ddfdS)NF)rO) rarrrrrrb next_valuerrrrrrrrrd?s rdcCs|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) onupdaterrrbrOrrr/r,rserver_onupdate_return_defaults)rrr!rkrErrrrr|Ps*    r|c 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#)r_clause_element_as_expr)r r!rrrrr(ws  z)_get_multitable_params..r~)rP)r6itemsr=r8r!addrrrrr,rbrOrrrrrrr) rrr'rFrDrCrErnormalized_paramsaffected_tablestr!rNrrrr9lsf            r9c 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)NrrPr) enumerater1r%rrrrOrrrb) rrrErvalues_0irow extensionrMrr% new_paramrrrrBs0   rBc Cs|D];\}}||}|dur|||qt|r-|jtjd||jdfi|}n |j|fi|}|||fqdS)N)rH) r setdefaultrrrOrIrJrrb) rr1r'r&rErr)vcolkeyrrrr7s r7cCs|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 inline _returningr4rrjr2rr r=r!rl)rrrirjrkrlrrrr]s>    r]cCs>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.)r2fullnamerPrrtrwarn)r!msgrrrr{ s r{)TFN)TN)"__doc__rXrrrrsymbolrr r r rrrr5r;r<rsrvrr ColumnElementrrrwrxrdr|r9rBr7r]r{rrrrsF       u $0\. ) %@  +