, A 6 d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d l
m Z d d
l
m Z d d l
m
Z
d d l
mZ d d
l
mZ d dl
mZ d dl
mZ d dl
mZ d dl
mZ d dlmZ d dlmZ d dlmZ ddlm Z ddl m Z ddl m Z ddl m Z ddl m!Z! ddl m"Z" dd l m#Z# dd l m$Z$ dd l m%Z% dd l m&Z& dd l m'Z' dd!l m(Z( dd"l m)Z) dd#l m*Z* er0d d$lm+Z+ d d%l,m-Z- d d&l.m/Z/ d d'l.m0Z0 d d(l1m2Z2 d d)l3m4Z4 d d*l5m6Z6 dd+l7m8Z8 G d, d- Z9 G d. d/ Z:d0S )1 ) annotations)Any)Dict)List)Optional)Tuple)
TYPE_CHECKING)Union)CheckConstraint)Column)ForeignKeyConstraint)Index)MetaData)PrimaryKeyConstraintschema)Table)types)SchemaEventTarget) OrderedDict) topological )exc)_columns_for_constraint)_copy)_copy_expression)_ensure_scope_for_ddl)_fk_is_self_referential) _idx_table_bound_expressions)_insert_inline)_is_type_bound) _remove_column_from_collection)_resolve_for_variant)_select)constraint_name_defined)constraint_name_string)Literal)Dialect) ColumnClause) quoted_name)Function)
Constraint)
TypeEngine) DefaultImplc e Zd Zd Zed d Zed d Zd dZd d
Zd d Z d d Z
d d
Z d!dZ d!dZ
d Zd"dZd"dZd Zd Zd Zd Zd Zd S )#BatchOperationsImplc || _ || _ || _ |dvrt d || _ || _ || _ t | | _ || _ t | | _
| j
dt | j
dd | j
d d|j j f |
| _ | | _ g | _ d S )N)autoalwaysneverz4recreate may be one of 'auto', 'always', or 'never'. listeners column_reflect)
operations
table_namer
ValueErrorrecreate copy_from
table_argsdict table_kwargs reflect_argsreflect_kwargs
setdefaultlistgetappendimplautogen_column_reflectnaming_conventionpartial_reorderingbatch) selfr8 r9 r r; r< r= r? r@ rA rH rI s j/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/operations/batch.py__init__z BatchOperationsImpl.__init__5 s %$ 6 6 6 F
!
"$ . . ( "> 2 2 && d 1 5 5k2 F FGG
K(//
z E F
"3 "4
returnr( c $ | j j j S N)r8 rF dialectrK s rL rR zBatchOperationsImpl.dialectZ s #++rN r. c | j j S rQ )r8 rF rS s rL rF zBatchOperationsImpl.impl^ s
##rN boolc t | j dk r | j j | S | j dk rdS dS )Nr2 r3 TF)r; r8 rF requires_recreate_in_batchrS s rL _should_recreatez$BatchOperationsImpl._should_recreateb s? =F " "?'BB4HH H
]h
&
&45rN Nonec P | }t | j j 5 |s2| j D ](\ }}}t | j j | } ||i | )n+| j rt | j }nt }| j
| j }d}n| j j
j r4t j
d| j j
j j d| j d t# | j |g| j R | j | j d| j }d}t- | j || j | j || j } | j D ] \ }}}t | | } ||i | | | j d d d d S # 1 swxY w Y d S ) N)rH FzEThis operation cannot proceed in --sql mode; batch mode with dialect zF requires a live database connection with which to reflect the table "z". To generate a batch SQL migration script using table "move and copy", a complete Table object should be passed to the "copy_from" argument of the batch_alter_table() method so that table reflection can be skipped.)r
autoload_withT)rI )rX r rF
connectionrJ getattrr8 rH r r< migration_contextas_sqlr CommandErrorrR namer9 r r@ r get_bindrA ApplyBatchImplr= r? rI _create)
rK should_recreateopnameargkwfnm1existing_table reflected
batch_impls
rL flushzBatchOperationsImpl.flushj sQ //11
"49#7
8
8 4 . 4 ." 3
.'+z # #OFC !5v>>BBNrNNNN# ) $!D4JKKKBB!B>-%)^N %II8?
! .
9 # @ H M
9
9 6:_
9
9
9 &+ &
* & & ${&*o&>&>&@&@ & & -
& &N !%I +I"O% '+'>
(,z # #OFC V44BBNrNNNN""49---i4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 . 4 .s E FF "F c @ | j d||f d S )N alter_columnrJ rE rK rg rh s rL rp z BatchOperationsImpl.alter_column %
>3344444rN c d|v sd|v r(| st j d | j d||f d S )N
insert_before insert_afterz^Can't specify insert_before or insert_after when using ALTER; please specify recreate='always'
add_column)rX r r` rJ rE rr s rL rw z BatchOperationsImpl.add_column sj r ! !^r%9%9'')) &:":
<b122222rN c @ | j d||f d S )N drop_columnrq rr s rL ry z BatchOperationsImpl.drop_column s%
=#r233333rN constr, c B | j d|fi f d S )Nadd_constraintrq rK rz s rL r| z"BatchOperationsImpl.add_constraint s(
+eXr:;;;;;rN c B | j d|fi f d S )Ndrop_constraintrq r} s rL r z#BatchOperationsImpl.drop_constraint s(
,uh;<<<<C62677777rN c B | j d|fi f d S )N
drop_indexrq r s rL r z BatchOperationsImpl.drop_index s'
<#455555rN c B | j d|fi f d S )Ncreate_table_commentrq rK tables rL r z(BatchOperationsImpl.create_table_comment s(
1E8R@AAAAArN c B | j d|fi f d S )Ndrop_table_commentrq r s rL r z&BatchOperationsImpl.drop_table_comment s(
/%2>?????rN c t d )Nz Can't create table in batch modeNotImplementedErrorr s rL create_tablez BatchOperationsImpl.create_table s !"DEEErN c t d )Nz Can't drop table in batch moder r s rL
drop_tablez BatchOperationsImpl.drop_table s !"BCCCrN c B | j d|fi f d S )Ncreate_column_commentrq rK columns rL r z)BatchOperationsImpl.create_column_comment s(
2VIrBCCCCCrN N)rO r( )rO r. )rO rU rO rY rz r, rO rY r r rO rY )__name__
__module__ __qualname__rM propertyrR rF rX rn rp rw ry r| r r r r r r r r r r6 rN rL r0 r0 4 s # # #J , , , X, $ $ $ X$ 7. 7. 7. 7.r5 5 5 53 3 3 34 4 4 4< < < <= = = =5 5 58 8 8 86 6 6 6B B B@ @ @F F FD D DD D D D DrN r0 c e Zd Z dFdGdZedHd ZdIdZdIdZdIdZdJdZ dKd Z
dLd Z dMdNd0Z dOd4Z
dPdQd7ZdRd9Zd: Zd; Zd< ZdSd?ZdSd@ZdTdCZdTdDZdE Zd S )Urc r6 rF r. r r r= tupler? Dict[str, Any]rl rU rI rO rY c ^ || _ || _ || _ || _ | |j | _ d | _ || _ d| _ t d | j j D | _ t | j | _ || _ | d S )Nr6 c 3 , K | ]}|j d |ifV dS exprN)ra .0cs rL z*ApplyBatchImpl.__init__.. s> ,
,
&'QVfa[ !,
,
,
,
,
,
rN )rF r r= r? _calc_temp_namera temp_table_name new_tablerI add_col_orderingr r column_transfersrC existing_orderingrl _grab_table_elements)rK rF r r= r? rl rI s rL rM zApplyBatchImpl.__init__ s
$ ( # 3 3EJ ? ? *."4
!, ,
,
+/:<,
,
,
!
!
"&d&;!N :% ) )C%(DL " "" B BA ( (DJ,=a,@ A A A A B BrN c z t }t j } j rF j D ]=}t | D ]+\ }}|dk r | ||dz
|f ,>n@t j D ]+\ }}|dk r | ||dz
|f ,| j d |D }t t j
||d }t fd|D _ t fd|D _ d S )Nr c 6 g | ]}|d |d k |S )r r r6 )r ps rL
zNApplyBatchImpl._adjust_self_columns_for_partial_reordering..6 s% 777A!A$!A$,,a,,,rN T)deterministic_orderc 3 6 K | ]}|j | fV d S rQ )r r r rK s rL r zMApplyBatchImpl._adjust_self_columns_for_partial_reordering..; s. "I"IAAt|A#7"I"I"I"I"I"IrN c 3 6 K | ]}|j | fV d S rQ )r r s rL r zMApplyBatchImpl._adjust_self_columns_for_partial_reordering..< sA ,
,
./Q%a( ),
,
,
,
,
,
rN )
setrC r rI enumerateaddr updater r sortr r )rK pairs
col_by_idxtuple_r elem
pairs_listsorted_s ` rL +_adjust_self_columns_for_partial_reorderingz:ApplyBatchImpl._adjust_self_columns_for_partial_reordering" s $,''
" = 1
=
=#,V#4#4 = =KE4 qyy 6%!)#4d";<<<=
=
))? @ @
=
= t199IIz%!) 4d ;<<<
T*+++
87777
Z N N N
#"I"I"I"I"I"I"III + ,
,
,
,
3:,
,
,
!
!
rN c d | j
J d t }| j j }| j s| j r| t | j |gt | j
t | j z R d|i| j
x| _ }t | j | j z D ]}d t! | D }| | j s6t' |t( r9t+ | rt- ||| j }n$t- || }nt- ||| }t' |t( r| || | | d S )Nz Can only create new table oncer c h | ] }|j
S r6 )keyr s rL zAApplyBatchImpl._transfer_elements_to_new_table..U s K K KqQU K K KrN )r target_tabler )r r r r rI r r r r rC r valuesr= r? r r r issubsetr r r
r r _setup_referentappend_constraint)rK mr r rz
const_columns
const_copys rL _transfer_elements_to_new_tablez.ApplyBatchImpl._transfer_elements_to_new_table@ s ~%%'G%%% JJ" " ?d&; ? < < > > >%*
&
4<&&(())D ,A,AA&
&
&
&
&
&
'..00 1 1D4L L 4 4E L K,CE,J,J K K KM ))$*?@@
%!566
*511 = "' f4:" " "JJ "'uV!. s5 %T4>:: rN )r _tablec 4 g | ]}j j | S r6 )r r )r r rK s rL r zCApplyBatchImpl._gather_indexes_from_both_tables.. s# LLLdn&s+LLLrN r )
r r r _column_flagr ra r r r rE r r keys)rK r idx_existingidx_copyr s ` rL _gather_indexes_from_both_tablesz/ApplyBatchImpl._gather_indexes_from_both_tabless sj ~))) L / / 1 1 ! !L (
! <\ J J #*
~ % H
JJx %,,.. E JJJ MLLLu}7I7I7K7KLLL < l
rN metadatar
constraintr
c X |j d }| d }|d }t | dk r |d }nd }|| j k rt j || }d d ||j v rs|j | }|j D ]\} | d d }
|
|j vr-| t |
t j ]d S t ||gd fd
|j D D R d |i d S d S )
Nr . r r c * | S rQ ) _get_colspec)r s rL colspecz/ApplyBatchImpl._setup_referent..colspec s ((***rN c B g | ] }t |t j S r6 )r sqltypesNULLTYPE)r ns rL r z2ApplyBatchImpl._setup_referent.. s5 q("344 rN c X g | ]&} | d d 'S )r r )split)r r r s rL r z2ApplyBatchImpl._setup_referent.. sD " " " $ $GDMM / / 4 4R 8" " "rN r )r r )elementsr r lenr
sql_schema_get_table_keytablesr
append_columnr r r r ) rK r r specpartstnamereferent_schemar tr colnamer s @rL r z ApplyBatchImpl._setup_referent s "
,..
3b u::?? #AhOO "O D( ( ( +E?CCC
+
+
+
+ ho%% OC( & / L LD %gdmm 1 1# 6 6r :G ac))w8I(J(JKKKL L
" " " "(2(;" " " + ) (rN op_implc
z | | | | j | j J | | j | t
| j t d | j
D t d | j D |
| j | | j | j j | j j | j j | j _ | D ]}| | | j | j _ d S # | j | j _ w xY w# |
| j xY w)Nc 3 ( K | ]
\ }}d |v |V dS r r6 )r r transfers rL r z)ApplyBatchImpl._create.. s= 'Ax!X-- ---- rN c & g | ]}d |v |d S )r r6 )r r s rL r z*ApplyBatchImpl._create.. s2 ( % 1 1 %V , 1 1 1rN r )r prep_table_for_batchr r r _execr from_selectrC r itemsr$ r r r r ra r r r )rK r r s rL rd zApplyBatchImpl._create s ,,...$$T4:666~))) T^,,, ; MM t~..:: +/+@+F+F+H+H
,0,A,H,H,J,J
tz * * *
$djodj>O
!
#'*/DN
; @@BB . .C((----. '+&:###d&:#:::: t~ . . . s B+F ,F F F:NFr9 column_namenullableOptional[bool]server_default)Optional[Union[Function[Any], str, bool]]ra
Optional[str]type_Optional[TypeEngine]
autoincrement&Optional[Union[bool, Literal['auto']]]commentUnion[str, Literal[False]]c | j | }
| j | } |||k r||
_ || d< | dd } | r\t | d | j j }
t |
t r'|
j r | j
|
j d |it j | }t |
j
t rdx|
j
_ |
j
_ | j |
| | ||
_
|||
_ |dur1|d |
_ n't' j | |
|t- | |
_ |dur ||
_ d S d S )Nra
existing_typeF)r r ra rD r# rF rR r r r popr to_instancer _create_eventscreate_constraintcast_for_batch_migrater r r
DefaultClause _set_parentrU r r" )rK r9 r r r ra r r r" rh existingexisting_transferr% resolved_existing_types rL rp zApplyBatchImpl.alter_column s < ,,0,A+,N 3 3 !HM(, f % FF?D 9 9M
)='):* *& 57HII.3 *...3
(//E (-):;;
, M3
I , ,+U
"HM (H & & %*.''(" + $%)-%8%8H " % &H rN r ru rv c | j }d t | D }| j s|rG|sE||v r'|| dz }|t | k r|| }nt | j | }|r7|s5||v r|| dz
}|dk r|| }nd | j D | }|r| xj ||ffz
c_ |r| xj ||ffz
c_ | j s!|s!|s!|r!| xj |d |ffz
c_ d S d S d S d S d S )Nc i | ]\ }}|| S r6 r6 )r ira s rL
zEApplyBatchImpl._setup_dependencies_for_add_column..6 s DDD71dtQDDDrN r r c i | ]\ }}|| S r6 r6 )r abs rL r3 zEApplyBatchImpl._setup_dependencies_for_add_column..N s+ ( ( (%)QAq( ( (rN r )r r rI r r> r )rK r ru rv
index_cols col_indexesr s rL "_setup_dependencies_for_add_columnz1ApplyBatchImpl._setup_dependencies_for_add_column/ s +
DDi
.C.CDDD & )
$
#{22 ), 7! ; Z 00,6sOM )-T-B(C(C ()
)#
)$ 33 )- 81 < !88+5c?L( (-1-B( ( ('() A ! !w
&>%@ @ ! ! @ ! !|W&=%? ? ! ! ' B! B ! B B
! !z"~w&?%A A ! ! ! !
B B B B B B B BrN r Column[Any]c | |j || t || j j | j |j < i | j |j < d S )Nr )r9 ra r r r r r )rK r9 r ru rv rh s rL rw zApplyBatchImpl.add_column_ s\
// K
%*&9J$K$K$K V[!-/ fk***rN %Union[ColumnClause[Any], Column[Any]]c |j | j j j v r t | j j j | | j |j = | j |j = | j |j d|v rPt |d t r7|d j r,| j
|d j d d S d S d S d S )Nr% ) ra r primary_keyr r" r r remover r r r& )rK r9 r rh s rL ry zApplyBatchImpl.drop_columno s ;$*08 8 8 *
&.
L % !&+ . %%fk222
r ! !2o.0ABB
"?#(
"
" & &?#($
" ! ! ! ! !rN c dS )a^ the batch table creation function will issue create_column_comment
on the real "impl" as part of the create table process.
That is, the Column object will have the comment on it already,
so when it is received by add_column() it will be a normal part of
the CREATE TABLE and doesn't need an extra step here.
Nr6 r s rL r z$ApplyBatchImpl.create_column_comment rN c dS )zthe batch table creation function will issue create_table_comment
on the real "impl" as part of the create table process.
Nr6 r s rL r z#ApplyBatchImpl.create_table_comment rA rN c dS )zthe batch table creation function will issue drop_table_comment
on the real "impl" as part of the create table process.
Nr6 r s rL r z!ApplyBatchImpl.drop_table_comment rA rN rz r, c l t |j st d t |t j r7| j j | j v r$| j | j j t |j r|| j |j < d S | j | d S )NConstraint must have a name)
r% ra r: r r r r r> r r? r& r rE r} s rL r| z ApplyBatchImpl.add_constraint s &uz22 < :;; ; eZ < = = Hz%)AAA(//
0FGGG !%* - - 316D "5: . . . $ + +E 2 2 2 2 2rN c |j st d |j | j v r| j |j \ }}t | j |j j D ]<}|j |j k r*| j |j j | =nWt |j r | j |j }n#|| j
v r| j
| t |t r!|j D ]}d| j |j _
d S d S # t $ r* t | rY d S t d|j z w xY w)NrE FzNo such constraint: '%s')ra r: r r& rC r r r? r& r r r r r> KeyErrorr! )rK rz r col_consts rL r z ApplyBatchImpl.drop_constraint s z < :;; ; ?zT 777 ! 7 ; ;EJ G G
U!%dl38&<&H!I!I M MI ~33 SX.:AA)LLLM (
33
7 .225:>>$222(//666 %!566
? = ? ?C9>DL*66
?
?? ? F F F e$$
7%* DEE E Fs C)D; ;E/E/r r c $ || j |j < d S rQ )r ra r s rL r zApplyBatchImpl.create_index s %("""rN c l | j |j = d S # t $ r t d|j z w xY w)NzNo such index: '%s')r ra rG r: r s rL r zApplyBatchImpl.drop_index sJ ? SX&&& ? ? ? 2SX =>> > ?s
"3c t d )NTODOr rr s rL r zApplyBatchImpl.rename_table s !&)))rN )r6 )rF r. r r r= r r? r rl rU rI r rO rY )r r rO r r )rO r )r r r r
rO rY )r r. rO rY )NFNNNF)r9 r r r r r r r ra r r r r r! r" r# rO rY )r r ru r rv r rO rY )NN)
r9 r r r: ru r rv r rO rY )r9 r r r< rO rY r r )r r r rM classmethodr r r r r r rd rp r9 rw ry r r r r| r r r r r6 rN rL rc rc s %'$ $ $ $ $: 5 5 5 [5,B ,B ,B ,B\
<14 14 14 14f B% % % %N%; %; %; %;V $(DI "&*@D.3K' K' K' K' K'Z.B .B .B .Bh (,&* 0 0 0 0 0 0
3
3
3
3? ? ? ?8) ) ) )? ? ? ? * * * * *rN rc N);
__future__r typingr r r r r r r
sqlalchemyr r r
r r r r r r r r sqlalchemy.eventsr sqlalchemy.utilr r utilr util.sqla_compatr r r r r r r r! r" r# r$ r% r& r' sqlalchemy.enginer( sqlalchemy.sql.elementsr) r* sqlalchemy.sql.functionsr+ sqlalchemy.sql.schemar, sqlalchemy.sql.type_apir- ddl.implr. r0 rc r6 rN rL r[ sv " " " " " " & & & & & & + + + + + + + + + + + + + + + + + + ( ( ( ( ( ( / / / / / / ' ' ' ' ' ' ' ' ' ' ' ' 6 6 6 6 6 6 $ $ $ $ $ $ / / / / / / 4 4 4 4 4 4 6 6 6 6 6 6 ; ; ; ; ; ; - - - - - - - - - - - - = = = = = = 3 3 3 3 3 3 & & & & & & 6 6 6 6 6 6 5 5 5 5 5 5
' ))))))444444333333111111000000222222&&&&&&[D [D [D [D [D [D [D [D|}* }* }* }* }* }* }* }* }* }*rN