tk= d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m
Z d d l m
Z
d d
lmZ d d lmZ d d lmZ d d
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddl"m#Z# d d l$m%Z% d d l%m&Z& er6d d lm'Z' d d l(m)Z) d d!l(m*Z* d d"l+m,Z, d d#l-m.Z. d d$l-m/Z/ d d%l0m1Z1 d d&l2m3Z3 dd'lm4Z4 G d( d)e# Z5 G d* d+ee Z6 G d, d-ee Z7 e
e6d. dFd4 Z8 e
e7d. dGd5 Z9 e
ed. dHd8 Z:dId;Z; e
ed. dJd= Z< e
ed. dKd? Z= e
ed. dLdA Z> e
e d. dMdC Z? e
e!d. dNdE Z@dS )O ) annotationsN)Any)List)Optional)
TYPE_CHECKING)Union)types)compiles)Column) CreateIndex)
Executable)
ClauseElement ) AddColumn) alter_column) alter_table)
ColumnDefault)
ColumnName)ColumnNullable)
ColumnType)format_column_name)format_server_default)format_table_name) format_type) RenameTable) DefaultImpl )util) sqla_compat)Literal)
MSDDLCompiler)
MSSQLCompiler) CursorResult)Index)Table) TableClause)
TypeEngine)_ServerDefaultc e Zd ZdZdZdZej ddhfz ZdZd1 fd Z d2 fd
Z
d1dZ d1 fd Z d3d4 fd#
Z
d5d&Zd6 fd+ Z d7d8 fd.
Zd/ Z fd0Z xZS )9 MSSQLImplmssqlTGOVARCHARNVARCHAR) minvaluemaxvalue
nominvalue
nomaxvaluecyclecacheorderon_nullr5 returnNonec t j |i | | j d| j | _ d S )Nmssql_batch_separator)super__init__ context_optsgetbatch_separator)selfargkw __class__s c/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/ddl/mssql.pyr< zMSSQLImpl.__init__A sH #$$$$ # 0 4 4 #T%9
constructr Optional[CursorResult]c t j |g|R i |}| j r!| j r| | j |S N)r; _execas_sqlr?
static_output)r@ rF argsrB resultrC s rD rJ zMSSQLImpl._execG sX y64666266 ; 54/ 5 t 3 4 4 4
rE c @ | d| j z d S )NzBEGIN TRANSACTION)rL command_terminator)r@ s rD
emit_beginzMSSQLImpl.emit_beginM s% .1HHIIIIIrE c t | j r#| j r | | j d S d S d S rI )r; emit_commitrK r? rL )r@ rC s rD rS zMSSQLImpl.emit_commitP s^
; 54/ 5 t 3 4 4 4 4 4 5 5 5 5rE NF
table_namestr column_namenullableOptional[bool]server_default/Optional[Union[_ServerDefault, Literal[False]]]name
Optional[str]type_Optional[TypeEngine]schema
existing_typeexisting_server_defaultOptional[_ServerDefault]existing_nullablerB c B | ||}d }n8|t j d n!|
||
}|}d }n|t j d d} t j || st j || r d} || d< | | d< t
j ||f|||||
d| |durV| du rR| dus|%| t ||d| |%t
|||| |'t
||||
d S d S ) NzhMS-SQL ALTER COLUMN operations with NULL or NOT NULL require the existing_type or a new type_ be passed.zMS-SQL ALTER COLUMN operations that specify type_= should also specify a nullable= or existing_nullable= argument to avoid implicit conversion of NOT NULL columns to NULL.FTrY ra )rW r] r_ r` rc sys.default_constraints)r_ rY )r_ r[ )
r CommandErrorwarnr _server_default_is_identity_server_default_is_computedr; r rJ _ExecDropConstraint)r@ rT rV rW rY r[ r] r_ r` ra rc rB used_defaultrC s rD r zMSSQLImpl.alter_columnU s " !&
&'> '
*u/@(H "MEE
I/
2 3
D
4 3
D
L#1B ,CB( )
' /
& &<5+@+@&e33~7M
'"#1 )$$ !#1 % GG KT
!
rE indexr$ c |j dd pd}|j J |D ]B}||j j vr2|j t |t j C| t | d S )N
mssql_include )
kwargsr> tablec
append_columnr sqltypesNullTyperJ r )r@ rl rn cols rD create_indexzMSSQLImpl.create_index s (( $??E2
{&&& J JC%+-'' ))&h6G*H*HIII
;u%%&&&&&rE rq Union[TableClause, Table]rows
List[dict]c Z | j r| d| j j | z t j ||fi | | d| j j | z d S t j ||fi | d S )NzSET IDENTITY_INSERT %s ONzSET IDENTITY_INSERT %s OFF)rK rJ dialectidentifier_preparer format_tabler; bulk_insert)r@ rq ry rB rC s rD r zMSSQLImpl.bulk_insert s ; 3 JJ+,2??FFG
EGG t 2 2r 2 2 2 JJ,,2??FFG
EGG t 2 2r 2 2 2 2 2rE column Column[Any]c | dd }|r%| t ||d| | dd }|r%| t ||d| | dd }|r$| t ||| t j ||fd|i| d S )Nmssql_drop_defaultFre mssql_drop_checkzsys.check_constraintsmssql_drop_foreign_keyr_ )poprJ rj _ExecDropFKConstraintr; drop_column) r@ rT r r_ rB drop_default
drop_checkdrop_fksrC s rD r zMSSQLImpl.drop_column s vv 2E:: JJ# (A6
VV .66
JJ# (?
662E:: J JJ,ZHH I I IJDDvDDDDDDrE c n |t j dd| }|t j dd| }||k S )Nz [\(\) \"\'] )resub)r@ inspector_columnmetadata_columnrendered_metadata_defaultrendered_inspector_defaults rD compare_server_defaultz MSSQLImpl.compare_server_default sV % 0(* $=) ) % & 1 *, $>* * & *-FFFrE c t || \ }}}|||s|j |j j rd}|||fS )NF)r; _compare_identity_defaultr primary_key)r@ metadata_identityinspector_identitydiffignoredis_alterrC s rD r z#MSSQLImpl._compare_identity_default sj "'''"C"C 1#
#
gx
%". /")5")5 6
HWh&&rE )r7 r8 )rF r r7 rG )NFNNNNNN)rT rU rV rU rW rX rY rZ r[ r\ r] r^ r_ r\ r` r^ ra rb rc rX rB r r7 r8 )rl r$ r7 r8 )rq rx ry rz rB r r7 r8 rI )rT rU r r r_ r\ r7 r8 )__name__
__module__ __qualname__ __dialect__transactional_ddlr? r
type_synonymsidentity_attrs_ignorer< rJ rQ rS r rw r r r r
__classcell__)rC s @rD r* r* / s K O -)Z1H0JJM
J J J J5 5 5 5 5 5 $(
"&* $.2<@,0W W W W W W Wr ' ' ' '3 3 3 3 3 3( !% E E E E E E E4G G G0' ' ' ' ' ' ' ' 'rE r* c e Zd ZdZd
d Zd S )rj FtnamerU colnameUnion[Column[Any], str]r] r_ r\ r7 r8 c > || _ || _ || _ || _ d S rI )r r r] r_ )r@ r r r] r_ s rD r< z _ExecDropConstraint.__init__ s$
rE N)
r rU r r r] rU r_ r\ r7 r8 r r r
inherit_cacher< ro rE rD rj rj s- M
rE rj c e Zd ZdZd d
Zd S )
r Fr rU r r r_ r\ r7 r8 c 0 || _ || _ || _ d S rI )r r r_ )r@ r r r_ s rD r< z _ExecDropFKConstraint.__init__! s
rE N)r rU r r r_ r\ r7 r8 r ro rE rD r r s- M rE r r+ elementcompilerr" r7 rU c | j | j | j | j f\ }}}}d|||t ||| |r|dz nddz S )Na declare @const_name varchar(256)
select @const_name = QUOTENAME([name]) from %(type)s
where parent_object_id = object_id('%(schema_dot)s%(tname)s')
and col_name(parent_object_id, parent_column_id) = '%(colname)s'
exec('alter table %(tname_quoted)s drop constraint ' + @const_name).r )typer r tname_quoted
schema_dot)r_ r r r] r )r r rB r_ r r r] s rD _exec_drop_col_constraintr ) sn
%!FE7E G
)(E6BB&,4fsll" J J
rE c l | j | j | j }}}d||t ||| |r|dz nddz S )Nay declare @const_name varchar(256)
select @const_name = QUOTENAME([name]) from
sys.foreign_keys fk join sys.foreign_key_columns fkc
on fk.object_id=fkc.constraint_object_id
where fkc.parent_object_id = object_id('%(schema_dot)s%(tname)s')
and col_name(fkc.parent_object_id, fkc.parent_column_id) = '%(colname)s'
exec('alter table %(tname_quoted)s drop constraint ' + @const_name)r r )r r r r )r_ r r r )r r rB r_ r r s rD _exec_drop_col_fk_constraintr B sZ %^W]GO7EF G )(E6BB&,4fsll" J J
rE r r! c d t || j | j dt || j fi |S N )r rT r_ mssql_add_columnr r r rB s rD visit_add_columnr V sB Hg 0'.AAAA7>88R888 rE r r c $ d | j |fi |z S )NzADD %s)get_column_specification)r r rB s rD r r ^ s% 7h7EE"EE EErE r c t || j | j dt || j dt || j d| j rdndS )Nr NULLzNOT NULL)r rT r_ r rV r r` rW r s rD visit_column_nullabler d sf
Hg 0'.AAAAXw 23333Hg 34444"2
2 rE r c t || j | j dt || j dt || j S )Nz
ADD DEFAULT z FOR )r rT r_ r defaultr rV r s rD visit_column_defaultr p sP Hg 0'.AAAA h88888W%8999 rE r c dt || j | j dt || j dt || j dS )NEXEC sp_rename 'r ', z
, 'COLUMN')r rT r_ r rV newnamer s rD visit_rename_columnr } sU
(G$6GGGG8W%899998W_5555 rE r c t || j | j dt || j dt || j S r )r rT r_ r rV r r] r s rD visit_column_typer sP
Hg 0'.AAAAXw 23333Hgm,,, rE r c l dt || j | j dt || j d S )Nr r )r rT r_ new_table_namer s rD visit_rename_tabler s?
(G$6GGGG(G$:DAAA rE )r rj r r" r7 rU )r r r r" r7 rU )r r r r! r7 rU )r r! r r r7 rU )r r r r! r7 rU )r r r r! r7 rU )r r r r! r7 rU )r r r r! r7 rU )r r r r! r7 rU )A
__future__r r typingr r r r r
sqlalchemyr rt sqlalchemy.ext.compilerr
sqlalchemy.schemar r sqlalchemy.sql.baser
sqlalchemy.sql.elementsr baser r r r r r r r r r r r implr r r r r sqlalchemy.dialects.mssql.baser! r" sqlalchemy.engine.cursorr# sqlalchemy.sql.schemar$ r% sqlalchemy.sql.selectabler& sqlalchemy.sql.type_apir' r( r* rj r r r r r r r r r r ro rE rD r s " " " " " " ( ( ( ( ( ( , , , , , , $ $ $ $ $ $ ) ) ) ) ) ) * * * * * * 1 1 1 1 1 1 $ $ $ $ $ $ ' ' ' ' ' ' # # # # # # % <<<<<<<<<<<<555555++++++++++++555555222222$$$$$$\' \' \' \' \' \' \' \'~
*m
J
w'' ('0
)) *)&
)W F F F F
.'"" #"
-!! "!
*g
*g
+w rE