9x>` j d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ej d d
Z ej d Z ej d Z
ej d
Z d Z d Z
d&dZd Zd Zd Zd Zd Zd'dZd'dZ G d dej Zd Zd Zd Zd Zd Z d! Z d" Z d# Z d$ 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 | j dk pd|v }|t
u rd| _ d| _ n5|t u rd| _ d| _ n |t u r
|sd| _ n
J d |t
t fv r&t | |fi ||r|| _ || _ || _ S S |r|| _ || _ || _ d S d S # |r|| _ || _ || _ w xY w)Nr visiting_cteFTz(ISINSERT, ISUPDATE, or ISDELETE expected) isinsertisupdateisdeletelenstackr
r r _get_crud_params)compilerstmtlocal_stmt_typekwrestore_isinsertrestore_isupdaterestore_isdeleteshould_restores g/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/sql/crud.py_setup_crud_paramsr ' st ( ( (
A - A1A x~ " R ("" ! H $ $ ! H $ $ % $H @@@@u1 x 2 2 2#Hd99b99 1 0H 0H 0H 1 3 1 0H 0H 0H 1 1> 1 0H 0H 0H 0 0 0 0s C C1c h
g _ g _ g _ g _ j |j fd|j j D S |j r|j d
n|j
t | \ }} j i }n!t
fd j D }g }
t |
|| i } j r |j
r
rt |
||||| j r |j rt# ||| |||| nt% ||| |||| |ry
rwt' | fd
D | }|r4t- j dd d |D z |j rt3 ||| }|S ) aP 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 ResultProxy's prefetch_cols() and postfetch_cols()
collections.
Nc : g | ]}|t |d d fS )NT)required)_create_bind_param).0cr s r
z$_get_crud_params..[ s@
"8QtDDD E
r c 3 H K | ] }r|v | t fV d S N)r )r# key_column_as_keystmt_parameterss r z#_get_crud_params..s sS
"
'* &@&@ ^C
( +&@&@&@&@
r& c 3 . K | ]} | V d S r( )r# kr* s r r, z#_get_crud_params.. s- EE..++EEEEEEr& zUnconsumed column names: %sz, c 3 K | ] }d |z V
dS )z%sNr. )r# r$ s r r, z#_get_crud_params.. s& 5 5!TAX 5 5 5 5 5 5r& ) 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) r r r _getattr_col_key_col_bind_namer6 values
check_columnscheckr* r+ s ` @@r r r H s H !H !H H #(?
Z'
! * /!, / %Xt44
#
+
F"# j/>62
M
T .
?
T .
%
o
OO
\EEEE_EEE
E
E
Z
&
&
"-99 5 5u 5 5 5557
! L/$KK Mr& TFc ||j }t j |||j | }d|_ |r |j | fi |}|S )N)type_r! T)r) r
BindParametertype_is_crud_compiler_dispatch)r colvalueprocessr! namer bindparams r r" r" se |w& e38h I I A0I0@@R@@ r& c | j r+|j r$t |j fd}fd}fd}n"t j }t j d x}}|||fS )Nc ~ t j | }t | d r| j v r| j j |fS |S )Nr7 )r r* hasattrr7 rV )r) str_key_ets r r* z4_key_getters_for_crud_column.._column_as_key sE -c22GsG$$
c)9)9 00 r& c J | j v r| j j | j fS | j S r( r7 rV r) rS r\ s r rH z6_key_getters_for_crud_column.._getattr_col_key s( yC 00wr& c P | j v r| j j d| j S | j S )N_r^ r_ s r rI z4_key_getters_for_crud_column.._col_bind_name s. yC "%)...#''::wr& r) )r r= rB r r* operator
attrgetter)r r r* rH rI r\ s @r r: r: s GT . G $#$$ "0,4,?,F,FF> +^ ;;r& c t | \ } }
} } fdj D }
j | _ g }j r>t |
}j j D ]"}||vr |j r|
| #|
D ]U} || }||v r1||vr-|
| | |d f Bt | ||| V|rr| | | j
| _ t | j j t d |D z | j _ d S d S )Nc F g | ] }j j | S r. r7 r$ )r# rV r* r s r r% z1_scan_insert_from_select_cols.. s, M M M4DJL -- . M M Mr& c 3 K | ] \ }}|V
d S r( r. )r# rS exprs r r, z0_scan_insert_from_select_cols.. s& 8893$888888r& )_get_returning_modifiersr? select_insert_from_select#include_insert_from_select_defaultsrB r7 r8 defaultappendpop&_append_param_insert_select_hasdefaultextend _generatetuple _raw_columns)r r r6 rH r* rI rK rJ r need_pksimplicit_returningimplicit_return_defaultspostfetch_lastrowidcolsadd_select_colscol_setrS r$ col_keys ` ` r r@ 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$1119 9r& c
t | \ } }
} } j rJfdj D }
t |
fd|
D fdj j D z }n j j }|D ]K} || }||v r ||vrt
| |||||
| ||
/| j r|j r>| r<|
s| r|j j ur*|
rt | ||| gt | ||| {|j t | || || |j C| r || v r| j | |j s| j | | r || v r | j | |j r$|j j ur|j st' | 0| j rt+ | || || Md S )Nc & g | ]
} | S r. r. )r# r) r* s r r% z _scan_cols..2 s0
$'NN3
r& c 4 g | ]}j j | S r. rf )r# r) r s r r% z _scan_cols..6 s" @@@c
S!@@@r& c & g | ]
}|j v |S r. )r) )r# r$ ordered_keyss r r% z _scan_cols..6 s/ D
D
D
quL'@'@A'@'@'@r& )ri _parameter_orderingrB r7 r$ r8 _append_param_parameterr primary_key_autoincrement_column!_append_param_insert_pk_returning_append_param_insert_pkrm _append_param_insert_hasdefaultserver_defaultr4 rn r1 nullable"_warn_pk_with_no_anticipated_valuer _append_param_update)r r r6 rH r* rI rK rJ r ru rv rw rx parameter_orderingry r$ r| r s ` ` @r rA rA s " !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 &--a0000 1 &--a000)
6a3K.K.K"))!,,,,
6TZ ===
> 31555
$#;VR
s; ;r& c
| | }
t j |
r9t | ||
f|
t u |j s || n
d || z d| }
n t
|
t j r,|
j j r |
}
|j |
_ |j r=|r;| j
| | j |
fi | }
n|r?||v r;| j
| | j |
fi | }
nA|j s| j
| | j |
fi | }
|
||
f d S )Nz%s_m0r! rV )ro r _is_literalr" r r9
isinstancerO rP _isnull_cloner r4 rn rU
self_groupr1 ) r r r$ r| r6 rI rv rw rJ r rT s r r r z s
NN7 # #EE"" ?"
h&-- """>>!,,,
eX 3 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 |j j rj| j j rB|j j r | j j s* | j |j fi |}| ||f | j | dS |j j r[| | | j |j j
fi |f | j | dS | |t | | f dS ||j
j u s|j | j | dS |j st# | dS dS )a Create 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_optionalrU rn r4 is_clause_elementargr "_create_insert_prefetch_bind_paramr7 r r r r r r r$ rJ r procs r r r s " y 9 P 2
)I&
) ':
) (x' 88R88
q$i((( % %a ( ( ( ( (
Y
( P MM$H$QY]%=%=%?%?FF2FFG
% %a ( ( ( ( ( MM1 @1 M MN O O O O O
dj. . .!2B2N!!!$$$$$
Z . +1-----. .r& c b t | |d || }| j | |S N)rU rV )r" r2 rn r r$ rU rV params r r r 6 xD' M M ME ##A&&& Lr& c b t | |d || }| j | |S r )r" r3 rn r s r "_create_update_prefetch_bind_paramr r r& c e Zd ZdZd Zd ZdS )_multiparam_columnTc z || _ d|j |dz fz | _ || _ |j | _ |j | _ d S )N%s_m%dr )indexr) originalrm rP )selfr r s r __init__z_multiparam_column.__init__ s>
x|UQY 77
' M r& c l t |t o |j | j k o|j | j k S r( )r r r) r )r others r __eq__z_multiparam_column.__eq__ s7 u 0 1 1
0 TX%
0$-/
r& N)__name__
__module__ __qualname___is_multiparam_columnr r r. r& r r r s8 " " "
r& r c > |j st j d|z |j j r* | j |j j fi |S t || }t |t j
rt | | S t | | 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)
rm r rE r rU r r r r r Insertr r )r r r$ r r rS s r _process_multiparam_default_bindr s 9
E
@BC
D
$ E x
8 8 : :AAbAAA E** dCJ ' ' E5hDD D5hDD Dr& c 2 |j |j j r2| j j s&||j j u r?| j j s | j j r'| |t | | f dS |j |j |j
st | dS dS dS dS )a Create 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) rm r r r r7 r "preexecute_autoincrement_sequencesrn r r r r )r r r$ rJ r s r r r s
% 2
TZ
---
/ .
B .
q|r ||v r | j | d S | j | d S |r)|j dur"||v r | j | d S d S d S d S )NT) onupdater r rn rU r r r4 r1 r server_onupdate_return_defaults)r r r$ rw rJ r s r r r R s z aj&< : ' P MM$H$QZ^%>%>%@%@GGBGGH
(
-A1I,I,I"))!,,,,,"))!,,,,, MM1 @1 M MN O O O O O
& # )-E(E(E % %a ( ( ( ( ( % %a ( ( ( ( ( % ! - -
) ) )!!!$$$$$ % % - - ) )r& c t d | D }t } |j D ]}
|
j D ]} | |v r| |
| | || < || } t
j | r%t | | | | t u || } n:| j
| | j |
fi |} | | | f | D ]}
|
j D ]} | |v r| j | j j s| j j rZ| | | j | j j
fi |f | j
| | | t% | | || f | j | j
| ӌd S )Nc 3 H K | ] \ }}t j | |fV d S r( )r _clause_element_as_expr)r# r$ r s r r, z)_get_multitable_params..y sJ Au
)! , ,e4 r& r )rV )r; itemsrB r= r$ addr r r" r r1 rn rU r r r r r r r )
r r r+ rK rI rH rJ r normalized_paramsaffected_tablestr$ rT s
r r> r> n s '--// eeO
* * * *A%%% ##A&&&56
. .q 1 12)!,'..
G . !&(!2 +^A . . EE &--a000 ,H ,U-=-=-?-? F F2 F FE
q%j)))! *& - - - -A%%%'
0F':/ MM ,H , !
9 9 ; ; ?A &--a0000MM > (!..2C2C "."))!,,,1 -- -r& c
|}|g}t |j dd D ]\ }}g }|D ]\ }} ||v s |j |v rs||v r|n|j }
t j ||
r%t | |||
fdd|j |dz fz i|} n: | j ||
fi |} nt | |||| } | || f | | |S )Nr rV r )
enumerater6 r) r r r" rU r r rn ) r r rJ r values_0irow extensionrS r r) new_params r rG rG sQ HXFDOABB /00 ! !3 $ / /LS%czzSW^^ CZZccSW'C11 N 2 C! ! &!a%(8 8 !
! !II !1 0S1D1D1F1F M M" M MII < dCB
c9 - . . . .
i Mr& c | D ]\ }} || }|| || )t j | r) | j t j d ||j fi |}nM|j r&|j j r| |j } | j |
fi |}| ||f d S )N)rN ) r
setdefaultr r rU rO rP _is_bind_parameterr _with_binary_element_typer rn ) r r6 r+ r* rJ r r/ vcolkeys r r< r< s %%'' " "1 "" ! !&! , , , ,
#A&&
;$H$ *4!&AAA EG ' rG r< ri r r. r& r r sC 4; 4;z " " 4;z " " 4;z " " 1 1 1Bq q qj >B !< !< !