>S t d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
Z
d d
l m Z
d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ ddlmZ ddlmZ e rBd dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d dlm Z d dlm!Z! d d l"m#Z# dd l$m%Z% G d d Z&d S )! ) annotations)Any)Dict)List)Optional)Sequence)Tuple)
TYPE_CHECKING)Unionschema)Column)
Constraint)Index)Integer)NULLTYPE )util) sqla_compat)
ColumnElement)
TextClause)CheckConstraint)
ForeignKey)ForeignKeyConstraint)MetaData)PrimaryKeyConstraint)Table)UniqueConstraint)
TypeEngine)MigrationContextc e Zd Z ddZ d?d@d Z dz8SchemaObjects.primary_key_constraint..4 s% ???Q9#Ax00???r, r
c * g | ]}j | S r6 cr: r; ts r* r< z8SchemaObjects.primary_key_constraint..9 s
#
#
#ac!f
#
#
#r, r- )metadatar8 r r )
r) r- r/ r1 r
dialect_kwmcolumnsprA s
@r* primary_key_constraintz$SchemaObjects.primary_key_constraint+ s
MMOO??$??? OJ CG C C CF C C
*
#
#
#
#d
#
#
#
*.
2<
r, sourcereferent
local_cols List[str] remote_colsonupdateondelete
deferrableOptional[bool]
source_schemareferent_schema initiallymatchr c
| }||k r | |
k r||z }n"|}t j ||gd |D R d|
i t j ||gd t j | D R d| i}|
r|
d|n|| |
d< t j |fd|D f||||| d|
}| | |S )Nc B g | ] }t j |t S r6 r7 r9 s r* r< z8SchemaObjects.foreign_key_constraint..U s% EEEA)"1h//EEEr, r
c B g | ] }t j |t S r6 r7 r9 s r* r< z8SchemaObjects.foreign_key_constraint..\ s5 H-- r, .rT c g | ] }d |
S )rX r6 )r: r; tnames r* r< z8SchemaObjects.foreign_key_constraint..m s$ 7 7 7aqq
! 7 7 7r, )r- rM rN rO rS )rB r8 r r unique_listr append_constraint)r) r- rH rI rJ rL rM rN rO rQ rR rS rT rC rD t1_colst1frZ s @r* foreign_key_constraintz$SchemaObjects.foreign_key_constraint= sr
MMOO X -?"B"B ;.GG G O
FEEEE
'
_
)'22
!
1 1 $
7 * 7 7 7 7; 7 7 7
!
Q r, r c t j || gd |D R d|i||d< t j fd|D i |} | |S )Nc B g | ] }t j |t S r6 r7 r9 s r* r< z3SchemaObjects.unique_constraint.. s%
@
@
@i q(++
@
@
@r, r
r- c * g | ]}j | S r6 r> r@ s r* r< z3SchemaObjects.unique_constraint.. s )E)E)EQ!#a&)E)E)Er, )r8 r rB r r\ )r) r- rH rJ r
kwuqrA s @r* unique_constraintz SchemaObjects.unique_constrainty s
O MMOO
A
@Z
@
@
@
6
')E)E)E)E*)E)E)E
L
L
L
B r, condition*Union[str, TextClause, ColumnElement[Any]]Union[CheckConstraint]c t j || t j dt | }t j |fd|i|}| | |S )Nxr r- )r8 r rB r r r r\ )r) r- rH rg r
rd rA cks r* check_constraintz SchemaObjects.check_constraint ss
O MMOO S' * *
&y
B
Bt
Br
B
B B r, type_r c
v | || }d t j t j d t j d} || } || }| | |S # t $ rB} t dd t d |D z | d } ~ ww xY w) Nr c 0 t j g g | S )Nr- )r8 r rq s r* z2SchemaObjects.generic_constraint.. s y'EBT( ( ( r, c . t j d| S )N rq )r8 r rq s r* rr z2SchemaObjects.generic_constraint.. s )";BT"J"J"J r, )
foreignkeyprimaryuniquecheckNrq z'type' can be one of %sz, c 3 4 K | ]}t | V d S r( )repr)r: rk s r* z3SchemaObjects.generic_constraint.. s( ":":q477":":":":":":r, )
tabler8 r r r r\ KeyError TypeErrorjoinsorted)
r) r- r/ rn r
rd rA typesconstkes
r* generic_constraintz SchemaObjects.generic_constraint s
JJz&J 1 1 !5 0JJ&+
+
%LE Et$$$E
& & &L )))F":":E":":": : :;;<
s A, ,
B86=B33B8r c i }| j :d| j j v r,| j j d }t |d r
|j |d< t j di |S )Ntarget_metadatanaming_conventionr6 )r# optshasattrr r8 r )r) rd mts r* rB zSchemaObjects.metadata sj
" .!T%;%@@@',->?Br.//
?*,*>&'!''B'''r, r c Z | }d |D }| dd }|r|D ]}dx|_ |_ t j ||g|R i | fd|D }|D ]} | j D ]}
| ||
S )Nc r g | ]4}t |t |j t j | n|5S r( )
isinstancer r| r _copy)r: r? s r* r< z'SchemaObjects.table.. sO
!V$$
$%G$7K a Q
r, _constraints_includedFc g | ]Z}t |t t f t |d d ur't |d d t j | n|[S )parentN) target_table)r r r getattrr r )r: elemrA s r* r< z'SchemaObjects.table.. s
$U 344
tXt,,A55h--9
d 3 3 3 3
r, ) rB poprw indexr8 r r\ foreign_keys_ensure_table_for_fk) r) r- rE rd rD r1 r r? constraintsr r_ rA s @r* r| zSchemaObjects.table s MMOO
!#'> F F +
+
+%**177 OD! 1d 1 1 1b 1 1
! ' 'E
& & & & , ,A % %a + + + +r, r r c ( t j ||fi |S r( )r8 r )r) r- rn rd s r* columnzSchemaObjects.column s e22r222r, tablenamerE 4Sequence[Union[str, TextClause, ColumnElement[Any]]]r c t j |pd| | |d< t j |gfd|D R i |}|S )Nno_tabler _tablec P g | ]"}t j | #S r6 )r r _textual_index_columnr@ s r* r< z'SchemaObjects.index.. s,
L
L
Lqd 44Q::
L
L
Lr, )r8 r rB r )r) r- r rE r
rd idxrA s @r* r zSchemaObjects.index s
O # MMOO
8 o
L
L
L
LG
L
L
L
r, table_keyTuple[Optional[str], str]c d|v r;| d }d |dd }|d }n|}d }||fS )NrX r )splitr )r) r tokenssnamerZ s r* _parse_table_keyz SchemaObjects._parse_table_key sV ) __S))F#&88F1R4L#9#9E2JEE EEu~ r, rB fkr c v t |j t r|j dd \ }}| | \ }}||j vrt
j ||| }n
|j | }||j vr1| t
j
|t dS dS dS )zUcreate a placeholder Table object for the referent of a
ForeignKey.
rX r N) r _colspecr0 rsplitr tablesr8 r r?
append_columnr r )r) rB r r cnamer rZ rel_ts r* r z"SchemaObjects._ensure_table_for_fk s
bk3 ' '
G !{ 1 1Q Iu 00;;LE5//!xFFF 2EG####I$4UH$E$EFFFFF
G
G $#r, r( )r# r$ r% r& )
r- r. r/ r0 r1 r2 r
r3 r% r )NNNNNNN)r- r. rH r0 rI r0 rJ rK rL rK rM r3 rN r3 rO rP rQ r3 rR r3 rS r3 rT r3 r% r )
r- r. rH r0 rJ r2 r
r3 r% r )
r- r. rH r0 rg rh r
r3 r% ri )
r- r. r/ r0 rn r3 r
r3 r% r )r% r )r- r0 r% r )r- r0 rn r r% r )
r- r3 r r3 rE r r
r3 r% r )r r0 r% r )rB r r r r% r& )__name__
__module__ __qualname__r+ rG r` rf rm r rB r| r r r r r6 r, r* r" r" % sb >B3 3 3 3 3 !% 2 #'"&%)'+)-#' #: : : : :B !% 4 !% . !% < ( ( ( (# # # #J3 3 3 3 !% * G G G G G Gr, r" N)'
__future__r typingr r r r r r r
r
sqlalchemyr
r8 sqlalchemy.sql.schemar r r sqlalchemy.typesr r rt r r sqlalchemy.sql.elementsr r r r r r r r r sqlalchemy.sql.type_apir runtime.migrationr r" r6 r, r* r sK " " " " " " * * * * * * ( ( ( ( ( ( , , , , , , ' ' ' ' ' ' $ $ $ $ $ $ % % % % % % 5555555222222555555000000::::::......::::::++++++666666222222444444zG zG zG zG zG zG zG zG zG zGr,