9x>`jdZddlZddlmZddlmZddlmZddlmZejd d Zejd Z ejd Z ejd Z dZ dZ d&dZdZdZdZdZdZd'dZd'dZGddejZdZdZdZdZd Zd!Zd"Zd#Zd$Z d%Z!dS)(ziFunctions used by compiler.py to determine the parameters rendered within INSERT and UPDATE statements. N)dml)elements)exc)utilREQUIREDa8 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. ISINSERTISUPDATEISDELETEc |j}|j}|j}|p|p|pt|jdkpd|v}|t urd|_d|_n5|t urd|_d|_n|tur |sd|_n Jd |t t fvr&t||fi||r||_||_||_SS |r||_||_||_dSdS#|r||_||_||_wxYw)Nr visiting_cteFTz(ISINSERT, ISUPDATE, or ISDELETE expected) isinsertisupdateisdeletelenstackr r r _get_crud_params)compilerstmtlocal_stmt_typekwrestore_isinsertrestore_isupdaterestore_isdeleteshould_restores /srv/buildsys-work-dir/castor/build_node/builder-2/WGSG1/unpkd_srcs/cloudlinux-venv-1.0.6/venv/lib/python3.11/site-packages/sqlalchemy/sql/crud.py_setup_crud_paramsr'st(((  A- A1A x~   " R  (""!  H $ $ ! H $ $ % $H @@@@u1 x2 2 2#Hd99b99  1 0H  0H  0H   1 3  1 0H  0H  0H    1 1> 1 0H  0H  0H  0 0 0 0s CC1c h g_g_g_g_j|jfd|jjDS|jr|jd n|j t|\ }}ji}n!t fdjD}g} t| ||i}j r|j r rt| |||||jr|jrt#||| |||| nt%||| |||| |ry rwt'| fd D|}|r4t-jddd|Dz|jrt3|||}|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 :g|]}|t|ddfS)NT)required)_create_bind_param).0crs r z$_get_crud_params..[s@   "8QtDDD E   rc3HK|]}r|v|tfVdSN)r )r#key_column_as_keystmt_parameterss r z#_get_crud_params..ssS  " '*&@&@^C ( +&@&@&@&@  r&c3.K|]}|VdSr()r#kr*s rr,z#_get_crud_params..s-EE..++EEEEEEr&zUnconsumed column names: %sz, c3 K|] }d|zV dS)z%sNr.)r#r$s rr,z#_get_crud_params..s&55!TAX555555r&) 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_namer6values check_columnscheckr*r+s ` @@rrrHsH!H!HH#(?    Z'     !*/!,/ %Xt44  #      +     F"# j/>62   M  T.  ?            T. %                      o   OO \EEEE_EEE E E Z & &   "-9955u555557   !L/$KK Mr&TFc ||j}tj|||j|}d|_|r|j|fi|}|S)N)type_r!T)r)r BindParametertype_is_crud_compiler_dispatch)rcolvalueprocessr!namer bindparams rr"r"se |w& e38hIIA0I0@@R@@ r&c|jr+|jr$t|jfd}fd}fd}n"tj}t jdx}}|||fS)Nc~tj|}t|dr|jvr|jj|fS|S)Nr7)rr*hasattrr7rV)r)str_key_ets rr*z4_key_getters_for_crud_column.._column_as_keysE-c22GsG$$ c)9)9 00r&cJ|jvr|jj|jfS|jSr(r7rVr)rSr\s rrHz6_key_getters_for_crud_column.._getattr_col_keys(yC 00wr&cP|jvr|jjd|jS|jS)N_r^r_s rrIz4_key_getters_for_crud_column.._col_bind_names.yC"%)...#''::wr&r))rr=rBrr*operator attrgetter)rrr*rHrIr\s @rr:r:sGT.G$#$$                 "0,4,?,F,FF> +^ ;;r&c t|\} } } } fdjD} j|_g}jr>t | }jjD]"}||vr|jr| |#| D]U}||}||vr1||vr-| || |dfBt||||V|rr| ||j |_t|jjtd|Dz|j_dSdS)NcFg|]}jj|Sr.r7r$)r#rVr*rs rr%z1_scan_insert_from_select_cols..s, M M M4DJL-- . M M Mr&c3 K|] \}}|V dSr(r.)r#rSexprs rr,z0_scan_insert_from_select_cols..s&8893$888888r&)_get_returning_modifiersr?select_insert_from_select#include_insert_from_select_defaultsrBr7r8defaultappendpop&_append_param_insert_select_hasdefaultextend _generatetuple _raw_columns)rrr6rHr*rIrKrJrneed_pksimplicit_returningimplicit_return_defaultspostfetch_lastrowidcolsadd_select_colscol_setrSr$col_keys ` ` rr@r@s" !400   N M M M M4;L M M MD#';H O /!d)):% ! !C'!!ck! C   ""1%% j WM%A%A NN7 # # # MM1d) $ $ $ $ 2$?B    9 o&&&'/'C'M'M'O'O$49  ( 55 5 88888 8 859$11199r&c t|\} } } } jrJfdjD} t| fd| DfdjjDz}n jj}|D]K}||}||vr||vrt |||||| | || /|jr|jr>| r<| s| r|jj ur*| rt||||gt||||{|j t||| |||jC| r|| vr|j||js|j|| r || vr|j||jr$|jj ur|jst'|0|jrt+||| ||MdS)Nc&g|] }|Sr.r.)r#r)r*s rr%z_scan_cols..2s0   $'NN3     r&c4g|]}jj|Sr.rf)r#r)rs rr%z_scan_cols..6s"@@@c S!@@@r&c&g|] }|jv |Sr.)r))r#r$ ordered_keyss rr%z_scan_cols..6s/D D D quL'@'@A'@'@'@r&)ri_parameter_orderingrBr7r$r8_append_param_parameterr primary_key_autoincrement_column!_append_param_insert_pk_returning_append_param_insert_pkrm_append_param_insert_hasdefaultserver_defaultr4rnr1nullable"_warn_pk_with_no_anticipated_valuer_append_param_update)rrr6rHr*rIrKrJrrurvrwrxparameter_orderingryr$r|rs ` ` @rrArAs" !400    "    +/+C   -.. @@@@-?@@@D D D D z|D D D  z! ;;""1%% j WM%A%A #"(     )  # 6# 6' # 6 / # 6  @@@&K5 $62,HdAvrJJJJ&/dA'?!-+15M0M0M&--a00001&--a000) 6a3K.K.K"))!,,,,  6TZ=== >31555    $#;VR   s;;r&c ||} tj| r9t||| f| tu|js ||n d||zd| } n t | tjr,| jj r | } |j| _|j r=|r;|j ||j| fi| } n|r?||vr;|j ||j| fi| } nA|j s|j ||j| fi| } | || fdS)Nz%s_m0r!rV)ror _is_literalr"r r9 isinstancerOrP_isnull_clonerr4rnrU self_groupr1) rrr$r|r6rIrvrwrJrrTs rrrzs NN7 # #EE""?"    h&--""">>!,,,       eX3 4 4 9K LLNNEEJ = ?/ ?   % %a ( ( ($H$U%5%5%7%7>>2>>EE % ?!/G*G*G   % %a ( ( ($H$U%5%5%7%7>>2>>EE = -"))!,,,$H$U%5%5%7%7>>2>>E MM1e*r&c|j|jjrj|jjrB|jjr |jjs*|j|jfi|}|||f|j|dS|jj r[|||j|jj fi|f|j|dS||t||fdS||j jus|j|j|dS|jst#|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)rm is_sequencedialectsupports_sequencesoptionalsequences_optionalrUrnr4is_clause_elementargr"_create_insert_prefetch_bind_paramr7rrrrrrr$rJrprocs rrrs" y 9  P2 )I& )': )(x' 88R88 q$i(((   % %a ( ( ( ( ( Y ( P MM$H$QY]%=%=%?%?FF2FFG      % %a ( ( ( ( ( MM1@1MMN O O O O O dj. . .!2B2N!!!$$$$$ Z. +1-----..r&cbt||d||}|j||SN)rUrV)r"r2rnrr$rUrVparams rrr6 xD' M M ME ##A&&& Lr&cbt||d||}|j||Sr)r"r3rnrs r"_create_update_prefetch_bind_paramrrr&ceZdZdZdZdZdS)_multiparam_columnTcz||_d|j|dzfz|_||_|j|_|j|_dS)N%s_m%dr)indexr)originalrmrP)selfrrs r__init__z_multiparam_column.__init__s> x|UQY77  ' M r&clt|to|j|jko|j|jkSr()rrr)r)rothers r__eq__z_multiparam_column.__eq__s7 u0 1 1 0 TX% 0$-/ r&N)__name__ __module__ __qualname___is_multiparam_columnrrr.r&rrrs8 """     r&rc>|jstjd|z|jjr*|j|jjfi|St||}t|tj rt||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) rmrrErrUrrrrrInsertrr)rrr$rrrSs r _process_multiparam_default_bindrs 9 E @BC D    $Ex 8 8 : :AAbAAA E** dCJ ' ' E5hDD D5hDD Dr&c2|j|jjr2|jjs&||jjur?|jjs |jjr'||t||fdS|j|j |j st|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) rmrrrr7r"preexecute_autoincrement_sequencesrnrrrr)rrr$rJrs rrrs  % 2  TZ ---   / . B .  q|r ||vr|j|dS|j|dS|r)|j dur"||vr |j|dSdSdSdS)NT) onupdaterrrnrUrrr4r1rserver_onupdate_return_defaults)rrr$rwrJrs rrrRs zaj&< : ' P MM$H$QZ^%>%>%@%@GGBGGH   ( -A1I,I,I"))!,,,,,"))!,,,,, MM1@1MMN O O O O O  & # )-E(E(E   % %a ( ( ( ( (   % %a ( ( ( ( ( %  ! - - ) ) )!!!$$$$$ %% - - ) )r&c td|D}t} |jD]} | jD]} | |vr| | | ||| <|| } t j| r%t|| | | tu|| } n:|j | |j | fi|} | | | f| D]} | jD]} | |vr| j| jjs| jjrZ| | |j | jj fi|f|j | | | t%|| || f| j|j | ӌdS)Nc3HK|]\}}tj||fVdSr()r_clause_element_as_expr)r#r$rs rr,z)_get_multitable_params..ysJ Au  )! , ,e4r&r)rV)r;itemsrBr=r$addrrr"r r1rnrUrrrrrrr) rrr+rKrIrHrJrnormalized_paramsaffected_tablestr$rTs rr>r>ns'--//eeO  ** * *A%%%##A&&&56 ..q112)!,'.. G. !&(!2+^A.. EE&--a000,H,U-=-=-?-?FF2FFE q%j)))! *&-- - -A%%%' 0F':/MM,H, !  9 9 ; ;?A&--a0000MM> (!..2C2C"."))!,,,1 ---r&c |}|g}t|jddD]\}}g}|D]\}} ||vs |j|vrs||vr|n|j} tj|| r%t |||| fdd|j|dzfzi|} n:|j|| fi|} nt|||||} | || f| ||S)NrrVr) enumerater6r)rrr"rUrrrn) rrrJrvalues_0irow extensionrSrr) new_params rrGrGsQHXFDOABB/00!!3 $ / /LS%czzSW^^ CZZccSW'C11 N 2 C!!&!a%(88 !  !!II!1 0S1D1D1F1F M M" M MII<dCB    c9- . . . . i    Mr&c|D]\}}||}||||)tj|r)|jtjd||jfi|}nM|jr&|jjr| |j}|j| fi|}| ||fdS)N)rN) r setdefaultrrrUrOrP_is_bind_parameterr_with_binary_element_typerrn) rr6r+r*rJrr/vcolkeys rr<r<s %%''""1""    ! !&! , , , , #A&& ;$H$*4!&AAAEG'rGr<rirr.r&rrsC 4;    4;z " " 4;z " " 4;z " "111Bqqqj>B    !<!<!