e}f N 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 dl
mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl m Z d dl m Z d dl m Z d dl!m"Z" d dl#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- dd$l%m.Z. dd%l%m/Z/ dd&l0m1Z1 d'd(l2m3Z3 d'd)l4m5Z5 d'd*l6m7Z7 d'd+l6m8Z8 d'd,l9m:Z: d'd-l9m;Z; d'd.l3mm?Z? d d1l@mAZA d d2lBmCZC d d3lDmEZE d d4lDmFZF d d5l mGZG d d6l mHZH d d7l mIZI d d8lJmKZK d d9lJmLZL d d:lMmNZN dd;l%mOZO d'dlTmUZU ejV eW ZX G d? d@e1 ZY G dA dBe( ZZ e*e/dC dedJ Z[ e*eZdC dfdK Z\ e*e)dC dgdM Z] e*e.dC dhdO Z^ e;j_ dP e:j_ dPdQ e7j` a dR G dS dTe7j` Zbe5jc d eb didX Zee5jf d e djd] Zgdkd^ZhdldaZidmddZjdS )n ) annotationsN)Any)cast)List)Optional)Sequence)Tuple)
TYPE_CHECKING)Union)Column)Index)literal_column)Numeric)text)types)BIGINT)ExcludeConstraint)INTEGER) CreateIndex) operators) ColumnClause)
TextClause)UnaryExpression)FunctionElement)NULLTYPE ) alter_column) alter_table) AlterColumn)
ColumnComment)compiles)format_column_name)format_table_name) format_type)IdentityColumnDefault) RenameTable) DefaultImpl )util)render)ops) schemaobj)BatchOperations)
Operations) sqla_compat)Literal)ARRAY)
PGDDLCompiler)HSTORE)JSON)JSONB)BinaryExpression)
ClauseElement) quoted_name)MetaData)Table)
TypeEngine)_ServerDefault)AutogenContext)_f_name)MigrationContextc e Zd ZdZdZej ddhfz ZdZd Zd Z d Z
d=d> fd!
Z d" Z d# Z
d?d(Zd@d+ZdAd-ZdBd/ZdCd4ZdDd6ZdEd8ZdFd:ZdGd<Z xZS )HPostgresqlImpl
postgresqlTFLOATzDOUBLE PRECISION)on_nullorderc |j dd pd}|D ]B}||j j vr2|j t |t j C| t | d S )Npostgresql_include )
kwargsgettablec
append_columnr sqltypesNullType_execr )selfindexrG cols h/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/ddl/postgresql.py create_indexzPostgresqlImpl.create_indexR s #\ - -.BD I I OR% J JC%+-'' ))&h6G*H*HIII
;u%%&&&&& c n |j D ],}|j #|j |j v r| | -d S N) constraintsnamenamed_constraintsdrop_constraint)rQ
batch_implrK
constraints rT prep_table_for_batchz#PostgresqlImpl.prep_table_for_batch\ sK + 1 1J+ Oz'CCC$$Z000 1 1rV c |j r||j j u rdS |}||k }|rdS d |||j fv r| S |j j }t |t rEt |j t s t j
dd| }d| d}t | }| j
t j t | |k S )NFz^'|'$ ') primary_keyrK _autoincrement_columnserver_defaultarg
isinstancestrtyper resubr
connectionscalarr/ _select)rQ inspector_columnmetadata_columnrendered_metadata_defaultrendered_inspector_defaultconn_col_defaultdefaults_equalmetadata_defaults rT compare_server_defaultz%PostgresqlImpl.compare_server_defaulte s
' ?#8#NNN55)-FF 5 % *
&% %*9= & , , @ . 3W==
;#%6(B8H#I#I #:'7#:#:#: -.> ? ? ?)) /004DD
rV NF
table_namerh column_namenullableOptional[bool]re %Union[_ServerDefault, Literal[False]]rZ
Optional[str]type_Optional[TypeEngine]schema
autoincrement
existing_typeexisting_server_defaultOptional[_ServerDefault]existing_nullableexisting_autoincrementkwr returnNonec
|
dd }||t j d |*| t |||||| |
| t j ||f|||||| |
| | d |
d S )Npostgresql_usingz6postgresql_using must be used with the type_ parameter)r usingr r r ) ry re rZ r r r r r r )popr) CommandErrorrP PostgresqlColumnTypesuperr )rQ rw rx ry re rZ r} r r r r r r r r __class__s rT r zPostgresqlImpl.alter_column s " )400 #H
JJ$ ! "/,C&7
)''$; /#9
rV c | d rt |d t t f rt j d|d }|rt
j |t d | d
}|r>|\ }}||d k r/t d||j
| |d= d S d S d S d S d S d S ) Ndefaultri z nextval\('(.+?)'::regclass\)a+ select c.relname, a.attname from pg_class as c join pg_depend d on d.objid=c.oid and d.classid='pg_class'::regclass and d.refclassid='pg_class'::regclass join pg_class t on t.oid=d.refobjid join pg_attribute a on a.attrelid=t.oid and a.attnum=d.refobjsubid where c.relkind='S' and c.relname=:seqnamer )seqnamerZ z^Detected sequence named '%s' as owned by integer column '%s(%s)', assuming SERIAL and omitting)rJ rg r r rj matchr/ _exec_on_inspectorr groupfirstloginforZ )rQ inspectorrK column_info seq_matchr r colnames rT autogen_column_reflectz%PostgresqlImpl.autogen_column_reflect s4 ??9 % % # 3* '6!2+
+
# 3 /Y1G I
3"5 E
&OOA . . %''
3'+$GW +f"555 ; $ !J #
( 2 2 2G# 3 # 3 # 3 # 3
3
3
3
355rV c d |D }|D ]}| | t j s| || d S d S )Nc F h | ] }|j d | S )duplicates_constraint)r rJ ).0rR s rT zAPostgresqlImpl.correct_for_autogen_constraints.. s=
z~~ 566
rV )remover/ sqla_2_skip_functional_indexes)rQ conn_unique_constraints conn_indexesmetadata_unique_constraintsmetadata_indexesdoubled_constraintsixs rT correct_for_autogen_constraintsz.PostgresqlImpl.correct_for_autogen_constraints s~
%
& $ $B # # # #! J ) )*:L I I I I I J JrV rR r
expr
remove_suffixc | }| dd }|j 5| |j j dd }|r<|d dk r0|d dk r$|dd }|r|d dk r |d dk $d |v rt j d
d| }|r-| | r|d t | }|S ) N"ra .r ()r z::z (::[\w ]+\w))lowerreplacerK rZ rj rk endswithlen)rQ rR r r s rT _cleanup_index_exprz"PostgresqlImpl._cleanup_index_expr s zz||||C$$ ; "<<5;#3#9#9#;#; > > >CCD tAw#~~$r(c//":D tAw#~~$r(c// 4<<6/2t44D /T]]= 9 9 /-3}----.D rV expr7 c d}t |t r|j |j }n|j }t |j t r
|j j }nd }|&|t j k rd}nX|t j k rd}nE|t j k r|t j k rd}n"|t j k r|t j k rd}nt |t |S )Nra z ascz nulls lastz asc nulls lastz nulls first) rg r modifierelementr asc_op nullslast_opdesc_op
nullsfirst_op)rQ r to_removeopinner_ops rT _default_modifiersz!PostgresqlImpl._default_modifiers s o.. |#k\ ck?;; $ "{ 3HH #H#Y ---$* y 555$1 !I$444 )"8 8 8 %6 I$555 )"9 9 9 %3 ; o.. < rV Tuple[Any, ...]c H t fdj D S )Nc 3 K | ]S} j gt |t r|d fn) | | fR V TdS )ra N)r rg rh _compile_elementr )r erR rQ s rT z2PostgresqlImpl.create_index_sig..8 s
%D $
"!S))PQGG //22D4K4KA4N4NO
rV )tuple expressions)rQ rR s ``rT create_index_sigz PostgresqlImpl.create_index_sig7 sF
&
rV r c J | | j ddd j S )NTF)
literal_binds
include_table)dialectcompile_kwargs)compiler string)rQ r s rT r z PostgresqlImpl._compile_elementD s2 L-1EJJ
rV r; autogen_contextr= Union[str, Literal[False]]c t | j }| d sdS t | d|j z r$t | d|j z } ||| S dS )Nz sqlalchemy.dialects.postgresqlFz_render_%s_type)ri
__module__
startswithhasattr__visit_name__getattr)rQ r} r modmeths rT render_typezPostgresqlImpl.render_typeJ sv 5kk$~~ >?? 5 4*U-AA B B 04!2U5I!IJJD4// /urV r3 c V t t t j ||dd S )N text_typez(.+?\(.*text_type=)r rh r* _render_type_w_subtyperQ r} r s rT _render_HSTORE_typez"PostgresqlImpl._render_HSTORE_typeW s3 ) 5K
rV r1 c V t t t j ||dd S )N item_typez(.+?\()r r s rT _render_ARRAY_typez!PostgresqlImpl._render_ARRAY_typea s2 ) Z
rV r4 c V t t t j ||dd S N astext_typez(.+?\(.*astext_type=)r r s rT _render_JSON_typez PostgresqlImpl._render_JSON_typek 3 )
7O
rV r5 c V t t t j ||dd S r r r s rT _render_JSONB_typez!PostgresqlImpl._render_JSONB_typeu r rV )
NFNNNNNNNN) rw rh rx rh ry rz re r{ rZ r| r} r~ r r| r rz r r~ r r r rz r rz r r r r )rR r
r rh r rh r rh )r r7 r rh )rR r
r r )r r7 r rh )r} r; r r= r r )r} r3 r r= r rh )r} r1 r r= r rh )r} r4 r r= r rh )r} r5 r r= r rh )__name__r __qualname__ __dialect__transactional_ddlr'
type_synonymsidentity_attrs_ignorerU r_ rv r r r r r r r r r r r r
__classcell__r s @rT rA rA J s K - $%1 M 1' ' '1 1 1)
)
)
^ $(@E "&* $(,.2<@,0153
3
3
3
3
3
3
j$3 $3 $3LJ J J( , D
rV rA c e Zd Zd fd Z xZS )
r rZ rh rx r} r; r r c | dd } t j ||fi | t j | | _ || _ d S )Nr )r r __init__rN to_instancer} r )rQ rZ rx r} r r r s rT r z PostgresqlColumnType.__init__ sV w%%{11b111 )%00
rV )rZ rh rx rh r} r; r r )r r r r r r s @rT r r s= rV r rB r r& compilerr2 r rh c j t || j | j dt || j d S )Nz RENAME TO )r rw r r# new_table_namer r r s rT visit_rename_tabler s=
Hg 0'.AAAA(G$:DAAA rV c t || j | j dt || j ddt || j z d| j r
d| j z ndS )N zTYPE %szUSING %sra )r rw r r rx r$ r} r r s rT visit_column_typer sq
Hg 0'.AAAAXw 23333K'-88888&-m;
W]""; rV r c d}| j 1|j | j t j nd}| t
|| j | j t || j
| S )Nz9COMMENT ON COLUMN {table_name}.{column_name} IS {comment}NULL)rw rx comment) r sql_compilerrender_literal_valuerN Stringformatr# rw r r" rx )r r r ddlr s rT visit_column_commentr s FC
? & 22 OX_ . .
::$ g('.
'x1DEE rV r% c t || j | j dt || j d}| j |dz
}|S | j $|dz
}|| | j z
}|S | j | j | j \ }}}| j }t | D ]T}|dk r|d|j rdndz z
}|d| t j d i |t || i z z
}U|S )
Nr z
DROP IDENTITYzADD alwayszSET GENERATED %s ALWAYSz
BY DEFAULTzSET %s rH )r rw r r rx r r visit_identity_columnimpl_compare_identity_defaultsortedr get_identity_optionsr/ Identityr )r r r r diff_identityattrs rT r r sJ
Hg 0'.AAAAXw 23333 D ( 0 ..w??? \;; OW <
a ?4LL Dx + (AHH\ H$A$A (KKD'(D2I2I+JKK% % rV create_exclude_constraint batch_create_exclude_constraintexclude_constraintc z e Zd ZdZdZ d"d#dZed$d Z d%d&dZed'd Z ed(d! Z
dS ))CreateExcludeConstraintOpz0Represent a create exclude constraint operation.excludeNconstraint_namesqla_compat._ConstraintNamerw Union[str, quoted_name]elementsIUnion[Sequence[Tuple[str, str]], Sequence[Tuple[ColumnClause[Any], str]]]where&Optional[Union[BinaryExpression, str]]r r| _orig_constraintOptional[ExcludeConstraint]r r c h || _ || _ || _ || _ || _ || _ || _ d S rX )r" rw r% r' r r) r )rQ r" rw r% r' r r) r s rT r z"CreateExcludeConstraintOp.__init__ s; / $
0 rV r^ r c t j | } | |j |j d |j D t d|j |j ||j |j |j S )Nc g | ]
\ }}}||f S rH rH )r r rZ r s rT
z=CreateExcludeConstraintOp.from_constraint.. s2
"D$ r
rV r( )r' r r)
deferrable initiallyr )
r/ _table_for_constraintrZ
_render_exprsr r' r r/ r0 r )clsr^ constraint_tables rT from_constraintz)CreateExcludeConstraintOp.from_constraint s '
8*:J $* '!, * "
rV migration_contextOptional[MigrationContext]c h | j | j S t j | }| | j | j }t
| j | j | j d| j
}|j D ].\ }}}| t |t /| | |S )N)r )rZ r' )r) r,
SchemaObjectsrK rw r r r% r" r' r r2 rM r r append_constraint)rQ r6
schema_objtexclr rZ opers rT
to_constraintz'CreateExcludeConstraintOp.to_constraint s ,( ( ,->??
T_T[ A A
]%*
g
4 4
OOF4 2 2 3 3 3 3 D!!! rV
operationsr. rh r r Optional[Table]c B | |||fi |}| | S )a Issue an alter to create an EXCLUDE constraint using the
current migration context.
.. note:: This method is Postgresql specific, and additionally
requires at least SQLAlchemy 1.0.
e.g.::
from alembic import op
op.create_exclude_constraint(
"user_excl",
"user",
("period", "&&"),
("group", "="),
where=("group != 'some group'"),
)
Note that the expressions work the same way as that of
the ``ExcludeConstraint`` object itself; if plain strings are
passed, quoting rules must be applied manually.
:param name: Name of the constraint.
:param table_name: String name of the source table.
:param elements: exclude conditions.
:param where: SQL expression or SQL string with optional WHERE
clause.
:param deferrable: optional bool. If set, emit DEFERRABLE or
NOT DEFERRABLE when issuing DDL for this constraint.
:param initially: optional string. If set, emit INITIALLY
when issuing DDL for this constraint.
:param schema: Optional schema name to operate within.
)invoke)r3 r@ r" rw r% r r s rT r z3CreateExcludeConstraintOp.create_exclude_constraint$ s4 V S *h
=
="
=
= $$$rV r- c t |j j |d< | ||j j |fi |}| | S )a3 Issue a "create exclude constraint" instruction using the
current batch migration context.
.. note:: This method is Postgresql specific, and additionally
requires at least SQLAlchemy 1.0.
.. seealso::
:meth:`.Operations.create_exclude_constraint`
r )r r rw rC )r3 r@ r" r% r r s rT r z9CreateExcludeConstraintOp.batch_create_exclude_constraintR sF & "-8
S */".do_expr_where_opts.. sO
(xx )/BBBB
rV zwhere=%sc " g | ] \ }}|d |
S )=rH )r kvs rT r. zC_exclude_constraint..do_expr_where_opts.. s' 777$!Q11%777rV )r2 r' appendr* _render_potential_exprextend)argsr r^ optss rT do_expr_where_optsz/_exclude_constraint..do_expr_where_opts s
,6+C
' KK/ $o
77$777888 rV z-%(prefix)screate_exclude_constraint(%(args)s)rY )prefixrd z%%(prefix)sExcludeConstraint(%(args)s))
_has_batchr/ ra rh r0 r rK r r* _identrZ _render_gen_namereprrc _alembic_autogenerate_prefixjoinrT )r^ r rI has_batchrf rd re s `` @rT rJ rJ s/
FHD *I @ \3z'<#=#=>??? > [#j&:";";<=== 6 Wc*"2 3 34555 H H:#3#: H Xv}Z-=-D E EFGGG
Z_
V,_joNN O
&
( */JJ K K
A T&-
(8(=>>??@ @D &&(()))> 9/JJIIdOOB
B
"!##65oFFIIdOO:
:
rV valueGUnion[ColumnClause[Any], Column[Any], TextClause, FunctionElement[Any]]c t | t r+| j rd}nd}|t j | | j dz S t j | |t | t t f S )Nz"%(prefix)sliteral_column(%(name)r)z%(prefix)scolumn(%(name)r))rg rZ ) wrap_in_text) rg r
is_literalr* _sqlalchemy_autogenerate_prefixrZ rb r r )ro r templates rT rZ rZ s % &&
4;HH3H <_MMJ
, #EJ+HII
rV )r r& r r2 r rh )r r r r2 r rh )r r r r2 r rh )r r% r r2 )r r= r r r rh )r^ r r r= rL r9 r rh )r r= r rh )r^ r r r= rI rU r rh )ro rp r r= r rh )k
__future__r loggingrj typingr r r r r r r
r
sqlalchemyr r
r r r r rN sqlalchemy.dialects.postgresqlr r r sqlalchemy.schemar sqlalchemy.sqlr sqlalchemy.sql.elementsr r r sqlalchemy.sql.functionsr sqlalchemy.typesr baser r r r r! r" r# r$ r% r& r r' ra r) autogenerater* r@ r+ r, operations.baser- r. r/ r0 $sqlalchemy.dialects.postgresql.arrayr1 #sqlalchemy.dialects.postgresql.baser2 %sqlalchemy.dialects.postgresql.hstorer3 #sqlalchemy.dialects.postgresql.jsonr4 r5 r6 r7 r8 sqlalchemy.sql.schemar9 r: sqlalchemy.sql.type_apir; r< autogenerate.apir= autogenerate.renderr> runtime.migrationr? getLoggerr r rA r r r r r register_operationAddConstraintOpregister_add_constraintr renderers dispatch_forrK _constraint_renderersrP rT rJ rZ rH rV rT r s " " " " " " % % % % % % ( ( ( ( ( ( 1 1 1 1 1 1 < < < < < < 2 2 2 2 2 2 ) ) ) ) ) ) $ $ $ $ $ $ 0 0 0 0 0 0 . . . . . . 3 3 3 3 3 3 4 4 4 4 4 4 % % % % % % $ $ $ $ $ $ # # # # # # ' ' ' ' ' ' ! ! ! ! ! ! " " " " " " - - - - - - ( ( ( ( ( ( 5 ::::::AAAAAA<<<<<<888888999999888888555555333333......++++++222222$$$$$$111111------444444 g!!s
s
s
s
s
[ s
s
s
l ;
+|$$ %$
-- .-
- && '&*
.. /. D :;;# # !B ,,-ABBI% I% I% I% I% 3 I% I% CB <;
I%X 899P P P :9P **+<== C C C >= C 9
9
9
9
x
rV