F:&\ H d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d d lm Z d d
lm Z d d l
mZ d d l
mZ d d
l
mZ d dl
mZ d dl
mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d dl m!Z! d dl m"Z" d dl m#Z# d dl$m%Z% d d l&m'Z' e
rTd d l
m(Z( d d l
m)Z) d d lm*Z* d d lm+Z+ d d!lm,Z, d d"l-m.Z. d d#l/m0Z0 d d$l1m2Z2 d d%l3m4Z4 d d&l m5Z5 d d'l6m7Z7 d d(l6m8Z8 d d)l9m:Z: d d*l9m;Z; e d+e d, - Zd3 ej? d4e D Z@e@d5k ZAe@d6k ZBe@d7k ZCe@d8k ZDe@d9k ZEeZF d d:lGmHZH n# eI$ r d d:l mHZH Y nw xY w G d; d< ZJ d d=l
mKZK d>ZLe@d?k ZMn # eI$ r G d@ dAeJ ZKdBZLdBZMY nw xY w d dCl
mNZN dDZOeOdEz ZPd>ZQn# eI$ r G dF dGeJ ZNdBZQY nw xY weErd dHl/mRZR nd dIlSmTZR e deUeRf ZVe eUeRf ZWddMZXddOZYddQZZdRZ[ej\ ddV Z]ddWZ^ddZZ_dd\Z`dd]Zadd_ZbddcZcddeZdddgZeddiZfddnZgdo Zhdp ZiddqZjddrZkddvZldw Zmdd{Znd| Zo epejq d} rd~ Zrd Zsnd Zrd Zsd ZtddZuddZvd ZwddZxddZyddZz G d dej5 Z{ ee{ dd Z| G d de Z} ee} dd Z~d Zd ZddZddZddZddZd ZddZeBr
d dl
mZ d dl
mZ n
d dl
mZ d ZddZddZdS ) ) annotationsN)Any)Iterable)Iterator)Mapping)Optional)
TYPE_CHECKING)TypeVar)Union) __version__)inspectschema)sql)types)url)compiles)CheckConstraint)Column)ForeignKeyConstraint)visitors)
BindParameter) ColumnClause) quoted_name)
TextClause)UnaryExpression)traverse) TypeGuard)Index)Table)
Connection)Dialect) Transaction) Inspector)ColumnCollection) SQLCompiler)Insert)
ColumnElement)
Constraint)
SchemaItem)Select) TableClause_CE)zColumnElement[Any]r* )boundvaluestrreturnUnion[int, str]c 4 t | S # | cY S xY wN)int)r/ s j/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/util/sqla_compat.py _safe_intr7 4 s% 5zz s c , g | ]}t | S )r7 ).0xs r6
r< < s EEEaYq\\EEE z
(\d+|[abc]\d)) )r> )r> r@ )r> r@ ) )
_NONE_NAMEc e Zd ZdZdS ) _Unsupportedz.Placeholder for unsupported SQLAlchemy classesN)__name__
__module__ __qualname____doc__r9 r= r6 rF rF L s 4444r= rF )ComputedT)r> r? c e Zd ZdS )rK NrG rH rI r9 r= r6 rK rK T r= rK F)Identity) start incrementminvaluemaxvalue
nominvalue
nomaxvaluecyclecacheorder)on_nullc e Zd ZdS )rP NrN r9 r= r6 rP rP a rO r= rP ) _NoneName)symbolname_ConstraintName!TypeGuard[_ConstraintNameDefined]c L | t u pt | t t f S r4 )rD
isinstancer0 r\ r^ s r6 constraint_name_definedrd s# : CD3 2B!C!CCr= TypeGuard[str]c , t | t S r4 )rb r0 rc s r6 constraint_name_stringrg s dC r=
Optional[str]c ( t | r| nd S r4 )rg rc s r6 constraint_name_or_nonerj s *$// 944T9r= auto
connectionOptional[Connection]Iterator[None]c # K | j }n# t $ r d }Y nw xY w|d V d S | s7| J | 5 d V d d d d S # 1 swxY w Y d S d V d S r4 )in_transactionAttributeErrorbeginrl rp s r6 _ensure_scope_for_ddlrt s
#2
~ )))!!##
EEEEEs AA !A c T t r | j | S | j | S )N)
hide_password)sqla_14render_as_string
__to_string__)r rv s r6 url_render_as_stringrz s6 >#s#-@@@@ s }====r= r! r# c P t | }|r|S | S r4 )_get_connection_transactionrr rl transactions r6 "_safe_begin_connection_transactionr s1 .j99K " !!!r= Nonec T t | }|r| d S d S r4 )r| commitr} s r6 #_safe_commit_connection_transactionr s; .j99K r= c T t | }|r| d S d S r4 )r| rollbackr} s r6 %_safe_rollback_connection_transactionr s; .j99K r= boolc H | j } | S # t $ r Y dS w xY w)NF)rp rq rs s r6 _get_connection_in_transactionr sC #2
~ uus
!!idxr Iterable[ColumnElement[Any]]c | j S r4 ) expressionsr s r6 _idx_table_bound_expressionsr s
?r= schema_itemc V t | d r
| j di |S | j di |S )N_copyr9 )hasattrr copy)r kws r6 r r sF {G$$ & { &&2&&& { %%"%%%r= Optional[Transaction]c T t r| S | j }|j S r4 )rw get_transaction_root_Connection__transaction)rl rs r6 r| r| s/ *))+++ ))r= url.URLc t t j d rt j j | i |S t j | i |S )Ncreate)r r URLr )argr s r6 _create_urlr sB sw!! #w~s)b)))w"r"""r= connectable tablename
schemanameUnion[str, None]c t r#t | || S | j | || S r4 )rw r
has_tabledialect)r r r s r6 _connectable_has_tabler sL
{##--iDDD",, J
r= c t rD| 5 }| || cd d d S # 1 swxY w Y d S | j || S r4 )rw _operation_contextexecutebind) inspector statementparamsconns r6 _exec_on_inspectorr s 9
)
)
+
+ 3t<< 622 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 ~%%i888s ?AAc F t s| j S ddlm} | j |j u S )Nr r )rw nullablesqlalchemy.sqlr _user_defined_nullableNULL_UNSPECIFIED)metadata_columnr s r6 _nullability_might_be_unsetr s:
''))))))
2f6M M
r= c F t sdS t d | D S )NFc 3 @ K | ]}t |t V d S r4 )rb rK r: sds r6 z._server_default_is_computed.. , EE:b(++EEEEEEr= ) has_computedanyserver_defaults r6 _server_default_is_computedr s- FuEEnEEEEEEr= c F t sdS t d | D S )NFc 3 @ K | ]}t |t V d S r4 )rb rP r s r6 r z._server_default_is_identity.. r r= )rw r r s r6 _server_default_is_identityr s- FuEEnEEEEEEr=
constraintr) r c T t | t r
| j }|J |S | j S r4 )rb r parenttable)r r s r6 _table_for_constraintr s7 * 233 ! r= c t | t rd | j D S t | t rt | j S t
| j S )Nc g | ] }|j
S r9 )r )r: fks r6 r< z+_columns_for_constraint..+ s 888b 888r= )rb r elementsr
_find_columnssqltextlistcolumnsr s r6 _columns_for_constraintr ) s` * 233 (88J$78888 J 0 0 ( Z /000J&'''r= r r$ r c h t r| |d S | |d S r4 )rw
reflect_table reflecttable)r r s r6 _reflect_tabler 2 s=
&&ud333%% 4
r= c ~ t | r-t | \ }}| |j | S | S r4 )_type_has_variants_get_variant_mappinggetr^ )type_r base_typemappings r6 _resolve_for_variantr ; s? % 1% 8 8 7{{7<333 r= _variant_mappingc * t | j S r4 )r r r s r6 r r E s E*+++r= c | | j fS r4 )r r s r6 r r H s e,,,r= c 8 t | t j u S r4 )typesqltypesVariantr s r6 r r M s E{{h ...r= c | j | j fS r4 )implr r s r6 r r P s z5=((r= c
< fd j D } j j } j j } j d j j j } j d j j j }d j D } j } j } j } j
}
||||||||| |
f
S )Nc 4 g | ]}j | j S r9 )r r^ )r: keyr s r6 r< z _fk_spec..U s1 ),
3 $ r= r c & g | ]}|j j S r9 )columnr^ )r: elements r6 r< z _fk_spec..] s MMMggn)MMMr= ) column_keysr r^ r r r r ondeleteonupdate
deferrable initially) r source_columns source_table
source_schema
target_schema target_tabletarget_columnsr r r r s ` r6 _fk_specr T s 0:0F N $)L %,M '*17>M &q)06;LMM9LMMMN"H"H&J$I r= r c | j d }| d }| d d | }| j J || j j k S )Nr .)r _get_colspecsplitpopjoinr r )r spectokenstablekeys r6 _fk_is_self_referentialr p sj q ! . . 0 0D
ZZ__F
JJrNNNxx H ( ( ( z(, ,,r= c | j S r4 ) _type_boundr s r6 _is_type_boundr y s
!!r= c R t }t | i d|j i |S )z2locate Column objects within the given expression.r )setr add)clausecolss r6 r r s+ 55D VR(DH-... Kr=
collectionr% r %Union[Column[Any], ColumnClause[Any]]c |j J | |j }t | d st | d r | j | dS | | dS )z(remove a column from a ColumnCollection.N
_immutable _readonly)r r _parentremove)r r to_removes r6 _remove_column_from_collectionr sz : ! ! !6:&I z<(( %GJ ,L,L %!!),,,,,)$$$$$r= text_*Union[str, TextClause, ColumnElement[Any]]&Union[ColumnElement[Any], Column[Any]]c t |t r1t |t j }| | |S t |t rt | | S t |t rt | |j S t |t j rt || S t d )zAa workaround for the Index construct's severe lack of flexibilityz#String or text() construct expected)rb r0 r r NULLTYPE
append_columnr _textual_index_element_textual_index_columntextr r( _copy_expression
ValueError)r r cs r6 r" r" s % @ 5(+ , ,
A E: & & @%eU333 E1 2 2 @$UEJ777 E3, - - @ u--->???r=
expressionr c : fd}t j | i | S )Nc t | t rV| j O| j urF| j j v rj | j S t | } | |S d S r4 )rb r r r^ r& r r )colr& r s r6 replacez!_copy_expression..replace sq sF # # % --x<>))#~ch//#JJ **1---4r= )r replacement_traverse)r' r r+ s ` r6 r$ r$ s:
(B r= c $ e Zd ZdZdZd d Zd
Zd S )
r! aQ Wrap around a sqlalchemy text() construct in such a way that
we appear like a column-oriented SQL expression to an Index
construct.
The issue here is that currently the Postgresql dialect, the biggest
recipient of functional indexes, keys all the index expressions to
the corresponding column expressions when rendering CREATE INDEX,
so the Index we create here needs to have a .columns collection that
is the same length as the .expressions collection. Ultimately
SQLAlchemy should support text() expressions in indexes.
See SQLAlchemy issue 3174.
_textual_idx_elementr r r# r r1 r c || _ || _ |j | _ t j | j j t
j | _ | | j d S r4 ) r r# r r r r r fake_columnr )selfr r# s r6 __init__z _textual_index_element.__init__ sQ
9!= 9JKK
D ,-----r= c | j gS r4 )r0 )r1 s r6 get_childrenz#_textual_index_element.get_children s !!r= N)r r r# r r1 r )rG rH rI rJ __visit_name__r2 r4 r9 r= r6 r! r! sH
,N. . . ." " " " "r= r! r compilerr& c ( |j | j fi |S r4 )processr# r r6 r s r6 _render_textual_index_columnr: s! 8 GL / /B / //r= c e Zd ZdS )_literal_bindparamNrN r9 r= r6 r< r< s Dr= r< c |j | fi |S r4 )render_literal_bindparamr9 s r6 _render_literal_bindparamr? s -8 ,W ; ; ; ;;r= c * t | j S r4 )r r r s r6 _get_index_expressionsrA s r= c 4 d t | D S )Nc 0 g | ]}t |d d S )r^ N)getattr)r: exps r6 r< z+_get_index_column_names.. s$ N N N3GC & & N N Nr= )rA r s r6 _get_index_column_namesrF s N N2H2M2M N N NNr= r* r r c " t r| j S i S r4 )sqla_13kwargs)r* s r6 _column_kwargsrJ s z r= Union[Index, Constraint]r Optional[Dialect]c | j d S |J t r |j | d S t | j d rt | j }nt } |t | j d }| | } t | t j r+| |d }|
| S |j | S )NF_alembic_quotequote)rP rc )r^ rw identifier_preparerformat_constraintr r r r0 __class__rb r r ddl_compiler_prepared_index_name)r r quoted_name_clsnew_nameds r6 _get_constraint_final_namerY s
t M *<< u =
:?G , , * %) $9$9OO )O"?3z#7#7uEEE ))x)88
j&, / / M$$Wd33A))
.@@LL Lr= Union[Constraint, Index]c | t r-| j dS |J |j | d }|d uS | j d uS )NFrN )rw r^ rQ rR )r r r^ s r6 _constraint_is_namedr\ ' sa + ? "5"""*<< u =
4 d**r=
mysql_dialectr" c T t r| j S t | j o| j S r4 )rw
is_mariadbr server_version_info _is_mariadbr] s r6 ra ra 6 s6
'' -
* )
r= c | j S r4 ) _mariadb_normalized_version_inforb s r6 rd rd @ s 99r= Union[TableClause, Table]r' c t r&| S | d S )NT)inline)rw insertrg )r s r6 _insert_inlineri D s8 )||~~$$&&&||4|(((r= )create_mock_engine)select
create_enginec & t dd| S )Nz
postgresql://mock)strategyexecutorrl )r rq r s r6 rj rj Q s fx
r= r+ c @ t j t | fi |S r4 )r rk r )r r s r6 _selectrs V s z$w--..2...r= indexc | j D ]R}t |t r |j }t |t t |t r|j r dS SdS )NTF)r rb r r r
is_literal)rt exprs r6 is_expression_indexrx Z sm ! // r s " " " " " " " " " " " " ( ( ( ( ( ( ! ! ! ! ! ! , , , , , , - - - - - - $ $ $ $ $ $ 2 2 2 2 2 2 # # # # # # 1 1 1 1 1 1 0 0 0 0 0 0 / / / / / / . . . . . . 3 3 3 3 3 3 , , , , , , ' ' ' ' ' ' 6 ,,,,,,))))))------666666444444333333))))))555555000000000000000000555555
ge5!CDEEE EE:2:&6 DDEEE 6/
6/
j
j
$ A>>>>>>> A A A@@@@@@@@A5 5 5 5 5 5 5 5 2###### L #z 1 $ $ $
<
L #
$######
. > > >" " " " & & & &* * * *# # # #
9 9 9
F F F FF F F F ( ( (
78 233 ), , ,- - - -
/ / /) ) ) 8- - - -" " " " % % % %$@ @ @ @$ *" " " " "S . " " ":
!!0 0 0 "!0
< < < < ! ! !O O O %M %M %M %MP + + + +
: : :) ) ) ) /------,,,,,,,((((((
/ / / / s6 E$ $ E21E2F F,+F,0G GG