]j d Z ddlmZ ddlZddlZddlZddlT ddlmZ ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd lm
Z
dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ n
# e$ r dZY nw xY w G d de Zd Zd Z G d de Z G d de Z G d de Z G d d ed d Z G d d e Z G d! d"e Z d# Z dS )$ah
Lightweight schema migrations.
Example Usage
-------------
Instantiate a migrator:
# Postgres example:
my_db = PostgresqlDatabase(...)
migrator = PostgresqlMigrator(my_db)
# SQLite example:
my_db = SqliteDatabase('my_database.db')
migrator = SqliteMigrator(my_db)
Then you will use the `migrate` function to run various `Operation`s which
are generated by the migrator:
migrate(
migrator.add_column('some_table', 'column_name', CharField(default=''))
)
Migrations are not run inside a transaction, so if you wish the migration to
run in a transaction you will need to wrap the call to `migrate` in a
transaction block, e.g.:
with my_db.transaction():
migrate(...)
Supported Operations
--------------------
Add new field(s) to an existing model:
# Create your field instances. For non-null fields you must specify a
# default value.
pubdate_field = DateTimeField(null=True)
comment_field = TextField(default='')
# Run the migration, specifying the database table, field name and field.
migrate(
migrator.add_column('comment_tbl', 'pub_date', pubdate_field),
migrator.add_column('comment_tbl', 'comment', comment_field),
)
Renaming a field:
# Specify the table, original name of the column, and its new name.
migrate(
migrator.rename_column('story', 'pub_date', 'publish_date'),
migrator.rename_column('story', 'mod_date', 'modified_date'),
)
Dropping a field:
migrate(
migrator.drop_column('story', 'some_old_field'),
)
Making a field nullable or not nullable:
# Note that when making a field not null that field must not have any
# NULL values present.
migrate(
# Make `pub_date` allow NULL values.
migrator.drop_not_null('story', 'pub_date'),
# Prevent `modified_date` from containing NULL values.
migrator.add_not_null('story', 'modified_date'),
)
Renaming a table:
migrate(
migrator.rename_table('story', 'stories_tbl'),
)
Adding an index:
# Specify the table, column names, and whether the index should be
# UNIQUE or not.
migrate(
# Create an index on the `pub_date` column.
migrator.add_index('story', ('pub_date',), False),
# Create a multi-column index on the `pub_date` and `status` fields.
migrator.add_index('story', ('pub_date', 'status'), False),
# Create a unique index on the category and title fields.
migrator.add_index('story', ('category_id', 'title'), True),
)
Dropping an index:
# Specify the index name.
migrate(migrator.drop_index('story', 'story_pub_date_status'))
Adding or dropping table constraints:
.. code-block:: python
# Add a CHECK() constraint to enforce the price cannot be negative.
migrate(migrator.add_constraint(
'products',
'price_check',
Check('price >= 0')))
# Remove the price check constraint.
migrate(migrator.drop_constraint('products', 'price_check'))
# Add a UNIQUE constraint on the first and last names.
migrate(migrator.add_unique('person', 'first_name', 'last_name'))
)
namedtupleN)*)
CommaNodeList)EnclosedNodeListEntity)
Expression)Node)NodeList)OP) callable_) sort_models)sqlite3)_truncate_constraint_name)CockroachDatabasec * e Zd ZdZd Zd Zd Zd ZdS ) Operationz/Encapsulate a single schema altering operation.c > || _ || _ || _ || _ d S N)migratormethodargskwargs)selfr r r r s H/opt/imunify360/venv/lib64/python3.11/site-packages/playhouse/migrate.py__init__zOperation.__init__ s"
c D | j j | d S r )r databaseexecute)r nodes r r zOperation.execute s!
&&t,,,,,r c 2 t |t t f r| | d S t |t r| d S t |t t f r|D ]}| | d S d S r )
isinstancer
Contextr r runlisttuple_handle_result)r resultitems r r( zOperation._handle_result s ftWo . . * LL
*
* * JJLLLLL
u
.
. *
*
*##D)))) * *
*
*r c | j }d|d< t | j | j }| || j i | d S )NT with_context)r copygetattrr r r( r )r r r s r r% z
Operation.run s] !!##!%~ 44 FFDI 8 8 899999r N)__name__
__module__ __qualname____doc__r r r( r% r r r r sV 99 - - -* * *: : : : :r r c F t j fd }|S )Nc x | dd }|r | g|R i |S t | j g|R i |S )Nr, F)popr r/ )r r r r, fns r innerzoperation.<locals>.inner s` zz.%88 -2d,T,,,V,, ,r{<T<<<V<<<r ) functoolswraps)r7 r8 s ` r operationr; s8 _R= = = = =
Lr c d | ft | z }t | dk rPt j | d }|d d d|d d }|S )N_@ zutf-88 )joinr' lenhashlibmd5encode hexdigest)
table_namecolumns
index_name
index_hashs r make_index_namerK s :-%..899J
: [!2!27!;!;<<FFHH
*3B3BQB @
r c e Zd ZdZdZd Zd Zed Ze d Z
d Z d Z e d Z
e d Ze d
Ze d Zd Ze d d Ze d Ze d Ze d d Ze d Ze d Ze d Ze d Ze d Ze d d Ze d Z e d d Z e d Z d
S ) SchemaMigratorFc || _ d S r )r )r r s r r zSchemaMigrator.__init__ s
r c 4 | j S r )r get_sql_contextr s r make_contextzSchemaMigrator.make_context s },,...r c T t r$t |t rt | S t |t rt | S t |t
rt
| S t |t rt | S t d|z )NzUnsupported database: %s)
r r# CockroachDBMigratorPostgresqlDatabasePostgresqlMigrator
MySQLDatabase
MySQLMigratorSqliteDatabaseSqliteMigrator
ValueError)clsr s r
from_databasez SchemaMigrator.from_database s ,H6G!H!H ,&x00 0
"4
5
5 ,%h// /
-
0
0 , ** *
.
1
1 ,!(++ +3h>???r c |j }t | r
| }| d t | d t
t | t j | | d S )NzUPDATE z SET T)flat)
defaultr
rR literalsqlr r r EQdb_value)r table column_namefieldr` s r
apply_defaultz SchemaMigrator.apply_default s - W giiG!!####VE]]##!!Z;''ENN7++ !r c l | d t | S )N ALTER TABLE )ra rb r )r ctxre s r _alter_tablezSchemaMigrator._alter_table s( {{>**..ve}}===r c | || d t | S )N ALTER COLUMN rl ra rb r r rk re columns r
_alter_columnz SchemaMigrator._alter_column s: c5)))**VF^^$$ &r c | }|j dc}|_ |j |k r|x|_ |_ | || d | | ||_ t |t r|
|| |S )NTz ADD COLUMN ) rR nullrf namerl ra rb ddlr# ForeignKeyFieldadd_inline_fk_sql)r re rf rg rk
field_nulls r alter_add_columnz SchemaMigrator.alter_add_column s !!!&T
EJ + +-8 8EJ*
,sE
"
"
'.
!
!
#eiinn
e_ - - / " "3 . . .
r c | | | d t | d | S )N ADD CONSTRAINT rl rR ra rb r r re ru
constraints r add_constraintz SchemaMigrator.add_constraint s\ d / / 1 1599+,,VD\\"" Z "r c dd | z }t t d t d |D f }| ||| S )Nzuniq_%sr= UNIQUEc , g | ]}t | S r3 r .0rq s r
<listcomp>z-SchemaMigrator.add_unique.<locals>.<listcomp> s H H HfVnn H H Hr )rA r SQLr r )r re column_namesconstraint_namer s r
add_uniquezSchemaMigrator.add_unique sl #chh|&<&<< MM H H< H H H I I K L L
""5/:FFFr c | | | d t | S )Nz DROP CONSTRAINT r~ r re ru s r drop_constraintz SchemaMigrator.drop_constraint sF d / / 1 1599,--VD\\"" $r c | d t |j j j d t
t |j j f }|j | d|j z }|j
| d|j
z }|S )N REFERENCES r}
ON DELETE %s
ON UPDATE %s) ra rb r rel_model_metarG r rel_fieldrf on_delete on_updater rk rg s r rx z SchemaMigrator.add_inline_fk_sql s ''F5?0;<<== $fU_-H&I&I%KLLMM
? &++o ?@@C ? &++o ?@@C
r Nc
|p
d|d|d|}| d t | d t t | d t t | f d t | d t | d } || d
|z } || d |z } | S ) Nfk_r= _refs_rj r| z
FOREIGN KEY r ()r r )rR ra rb r r r )
r re rf rel
rel_columnr r r r rk s
r add_foreign_key_constraintz)SchemaMigrator.add_foreign_key_constraint sD % C Ceee>Ikk>Ac)C
''F5MM""*++F4Z@@AABB(($f[&9&9%;<<==''F3KK
F:&&'' ++o 9::C ++o 9::C
r c
|j s|j t d|z t |t }|r|j st d | ||| g}|j s@| | ||| | || g |rW| j
rP| | |||j
j j |j j |j |j |j s|j rBt+ |dd }| | ||f|j | |S )Nz!%s is not null but has no defaultz$Foreign keys must specify a `field`.
index_type)rt r` r[ r# rw r rz extendrh add_not_nullexplicit_create_foreign_keyappendr r r rG rf r r indexuniquer. add_index)r re rf rg is_foreign_key
operationsusings r
add_columnzSchemaMigrator.add_column9 s z Pem 3 @; NOO O#E?;; E%/ E CDD D++E;FFG
z 8 ""5+u==!!%55 7
8
8
8 &d > & // O)4O/OO
% %
&
&
& ; C%, CE<66E dnnU[N-2\5 B B
C
C
C r c t r NotImplementedError)r re rf s r drop_foreign_key_constraintz*SchemaMigrator.drop_foreign_key_constraint` !!r Tc | | }| || d t | |r| d d | j | D }||v r | j r| || |gS |S )N
DROP COLUMN CASCADEc g | ] }|j
S r3 rq )r foreign_keys r r z.SchemaMigrator.drop_column.<locals>.<listcomp>n s1 F F F
F F Fr ) rR rl ra rb r r get_foreign_keysexplicit_delete_foreign_keyr )r re rf cascaderk
fk_columnss r drop_columnzSchemaMigrator.drop_columnd s !!
3 & &
'/
"
"
#f[!!
"
"
" $ KK
# # #F F #} = =e D DF F F
* $ $)I $44UKHH#N N
r c | | | d t | d t | S )N RENAME COLUMN TO r~ )r re old_namenew_names r
rename_columnz SchemaMigrator.rename_columnv sh d / / 1 1599*++VH%%&& VH%%&& (r c z | | || d S )Nz
SET NOT NULLrr rR ra r re rq s r r zSchemaMigrator.add_not_null s4 t 0 0 2 2E6BB )) +r c z | | || d S )Nz DROP NOT NULLr r s r
drop_not_nullz SchemaMigrator.drop_not_null s5 t 0 0 2 2E6BB)** ,r c |t d t | r
| }t |t r$| d rt | }| | | d t | d | S )N `default` must be not None/NULL.r 'rn z
SET DEFAULT ) r[ r
r# strendswithr rl rR ra rb r )r re rq r` s r add_column_defaultz!SchemaMigrator.add_column_default s ? ?@@ @ W giiG gs # # #(8(8(D(D #'llG d / / 1 1599)**VF^^$$ ))W r c | | | d t | d S )Nrn z
DROP DEFAULTr~ r s r drop_column_defaultz"SchemaMigrator.drop_column_default sS d / / 1 1599)**VF^^$$ )) +r c j | }| ||| d | | }|Lt |t st | }| d | }|S )Nz TYPE z USING )rR rr ra rb ddl_datatyper# r
r r re rq rg castrk s r alter_column_typez SchemaMigrator.alter_column_type s !!
c5&11!!E&&s++,, dD))
!4yy++i((,,T22C
r c | | | d t | S )Nz RENAME TO r~ r r r s r rename_tablezSchemaMigrator.rename_table sG d / / 1 18<< ''VH%%&& (r c | }t || }t | fd|D }t | ||| }| | S )Nc : g | ]}t j | S r3 )r. c)r rq table_objs r r z,SchemaMigrator.add_index.<locals>.<listcomp> s% CCC V,,CCCr )r r )rR rK TableIndexrb )
r re rH r r rk rI colsr r s
@r r zSchemaMigrator.add_index so !!$UG44
%LL CCCC7CCCj)T&NNNwwu~~ r c | d t | S )N DROP INDEX rR ra rb r r re rI s r
drop_indexzSchemaMigrator.drop_index s5 ''VJ''(( *r )NNN)Tr )FN) r/ r0 r1 r r r rR classmethodr] r; rh rl rr rz r r r rx r r r r r r r r r r r r r r3 r r rM rM s "' "' ! ! !/ / / @ @ [ @
!
! Y
! > > >& & & Y( " " Y" G G YG $ $ Y$
=A37 Y2 $ $ Y$L " " Y" Y" ( ( Y( + + Y+
, , Y,
Y " + + Y+ Y ( ( Y( Y * * Y* * *r rM c J e Zd Zd Zed Ze fd Z xZS )rV c | d}| j ||z }d | D S )Nai
SELECT pg_attribute.attname
FROM pg_index, pg_class, pg_attribute
WHERE
pg_class.oid = '%s'::regclass AND
indrelid = pg_class.oid AND
pg_attribute.attrelid = pg_class.oid AND
pg_attribute.attnum = any(pg_index.indkey) AND
indisprimary;
c g | ]
}|d S r r3 )r rows r r z;PostgresqlMigrator._primary_key_columns.<locals>.<listcomp> s 4443A444r )r execute_sqlfetchall)r tblquerycursors r _primary_key_columnsz'PostgresqlMigrator._primary_key_columns sB **53;7744&//"3"34444r c V | d|z S )NzSET search_path TO %s)rR ra )r schema_names r set_search_pathz"PostgresqlMigrator.set_search_path s' 0;>?? Ar c | | }t t | }| ||d g}t | dk r|d|d d}d}| j ||f }t | r7|d|d d} | | || |S )NT)r, r= r _seqz
SELECT 1
FROM information_schema.sequences
WHERE LOWER(sequence_name) = LOWER(%s)
)
r superrV r rB r r boolfetchoner ) r r r pk_names ParentClassr seq_namer r new_seq_name __class__s r r z PostgresqlMigrator.rename_table s ,,X66 .55
$ $Xxd $ K KM
x==A &.hh <HE
]..uxkBBFFOO%%&&
-.6hh D !!+":": l#, #, - - - r )r/ r0 r1 r r; r r
__classcell__)r s @r rV rV sq 5 5 5 A A YA
Y r rV c . e Zd ZdZd Zed ZdS )rT Tc d S r r3 r s r rx z%CockroachDBMigrator.add_inline_fk_sql r c | d t | d S )Nr r r r s r r z CockroachDBMigrator.drop_index sB ''VJ''(($$ &r N)r/ r0 r1 r rx r; r r3 r r rT rT sD "&
& & Y& & &r rT c X e Zd Zed Zed Zed ZddZdS ) MySQLColumnc | j dk S )NPRIpkrQ s r is_pkzMySQLColumn.is_pk
w% r c | j dk S )NUNIr rQ s r is_uniquezMySQLColumn.is_unique r r c | j dk S )NYES)rt rQ s r is_nullzMySQLColumn.is_null s yE!!r Nc || j }|| j }t | t | j g}| j r"| t d |r#| t d n"| t d | j r"| t d | j r'| t | j t | S )Nr NULLNOT NULLz PRIMARY KEY)
r ru r r
definitionr r r extrar )r rf r partss r rb zMySQLColumn.sql s ?lG )K ; " > ( LLX ' ' ' * LLV % % % % LLZ ) ) ) : - LL]++ , , , : * LLTZ ) ) ) r NN)r/ r0 r1 propertyr r r rb r3 r r r r sw
X X " " X" r r _Column)ru r rt r
r` r c e Zd ZdZdZd Zed Zd Zd Z ed Z
d Z ed Z ed Z
ed
Zedd Zed
Zd S )rX Tc | || d t | S )N MODIFY ro rp s r rr zMySQLMigrator._alter_column/ s9 c5))$$VF^^$$ &r c | d t | d t | S )Nz
RENAME TABLE r r r s r r zMySQLMigrator.rename_table5 sU ))VH%%&& VH%%&& (r c | j d|z }| }|D ]}t | }|j |k r|c S dS )NzDESCRIBE `%s`;F)r r r r ru )r re rf r rowsr rq s r _get_column_definitionz$MySQLMigrator._get_column_definition> se **+;e+CDD C #&F{k))
*ur c | j d||f }| }|st d|d|d |d S )NzSELECT constraint_name FROM information_schema.key_column_usage WHERE table_schema = DATABASE() AND table_name = %s AND column_name = %s AND referenced_table_name IS NOT NULL AND referenced_column_name IS NOT NULL;z+Unable to find foreign key constraint for "z " on table "z".r )r r r AttributeError)r re rf r r) s r get_foreign_key_constraintz(MySQLMigrator.get_foreign_key_constraintG ss **3 K " " "" > .). => >
> ayr c | || }| | | d t | S )Nz DROP FOREIGN KEY )r% rl rR ra rb r )r re rf
fk_constraints r r z)MySQLMigrator.drop_foreign_key_constraintX s] 77{KK
d / / 1 1599-..VM**++ -r c d S r r3 r s r rx z MySQLMigrator.add_inline_fk_sql` r r c | || }| | | d | d }t d | j | D }||vr|S || }| || || |||j
|j fS )Nr Fr c 3 ( K | ]
}|j |fV d S r r r fks r <genexpr>z-MySQLMigrator.add_not_null.<locals>.<genexpr>k < = = YO= = = = = =r ) r" rl rR ra rb dictr r r r
dest_table dest_column)r re rq
column_defr
fk_objects fk_metadatas r r zMySQLMigrator.add_not_nullc s 00??
% d&7&7&9&95AA ,, Z^^E^ : :;;
= =m44U;;= = = = =
# # ( 00?? // * + - -. .r c | || }|j rt d | | | d | d S )Nz Primary keys can not be nullr Tr* )r" r r[ rl rR ra rb r s r r zMySQLMigrator.drop_not_nullz s{ ,,UF;; < = ;<< < d / / 1 1599$$VZZZ--.. 0r c L t d | j | D }||v }| || }| | | d t | d | | }|rA|| }| || ||
|||j |j gS |S )Nc 3 ( K | ]
}|j |fV d S r r r, s r r. z.MySQLMigrator.rename_column.<locals>.<genexpr> r/ r z CHANGE r} )rf )
r0 r r r" rl rR ra rb r r r r1 r2 ) r re r r r4 r rq
rename_ctxr5 s r r zMySQLMigrator.rename_column s* = =m44U;;= = = = =
"Z/,,UH==#|D$5$5$7$7?? wz**s6(++,, ws||s6::(:;;<< $X.K00AA// * + - -
r Nc P |t d | }| || d t | d | | S )Nz5alter_column_type() does not support cast with MySQL.r r} )r[ rR rl ra rb r rv r s r r z MySQLMigrator.alter_column_type s & ' '
' !! c5))$$VF^^$$ UYYs^^$$ &r c | d t | d t | S )Nr z ON r r s r r zMySQLMigrator.drop_index sS ''VJ''(( VE]]## %r r )r/ r0 r1 r r rr r; r r" r% r rx r r r r r r3 r r rX rX + s "& "& & & & ( ( Y( " - - Y-
. . Y., 0 0 Y0 Y 6
&
&
& Y
& % % Y% % %r rX c e Zd ZdZ ej d Z ej d Z ej d Z ej dej Z
d Z d Z e
d Zd Ze
dd Ze
dd
Ze
d Ze
d Ze
d Ze
d Ze
dd Ze
d Ze
d Ze
dd ZdS )rZ z
SQLite supports a subset of ALTER TABLE queries, view the docs for the
full details http://sqlite.org/lang_altertable.html
z
(.+?)\((.+)\)z(?:[^,(]|\([^)]*\))+z
["`']?([\w]+)z FOREIGN KEY\s+\("?([\w]+)"?\)\s+c ^ | j d|z }d |j D S )Nzselect * from "%s" limit 1c g | ]
}|d S r r3 )r r* s r r z4SqliteMigrator._get_column_names.<locals>.<listcomp> s 444DQ444r )r r descriptionr re ress r _get_column_namesz SqliteMigrator._get_column_names s3 m''(Du(LMM44CO4444r c | j dd| g }| S )NzBselect name, sql from sqlite_master where type=? and LOWER(name)=?re )r r lowerr r@ s r _get_create_tablez SqliteMigrator._get_create_table s> m''.
ekkmm $& & ||~~ r c t d j | D }| |vrt d|d|d | \ }} j | } j | t j dd| } j
| \ }} j
| } d | D }
g } g } g }
d}|
D ]} j | \ }||k r| ||| }|rm| | |
| j | \ }| | | | | | s*| | |
| t' t) |
| }| | d } sd
}n |k r fd }g }| D ]^} j | }|)| d |k r || }|r| | _|d
z }t j d|z t j }| d|z | }d | }t5 t7 d t9 | g t7 | d|d g}t5 t7 d t9 | t= d | D t7 d t? d |
D t7 d t9 | f }t5 t7 d t9 | g }||| || gz
}tC d | D ]u} || j" vr(| t7 | j# 3 r@ $ | j# | } | "| t7 | v|S )Nc 3 H K | ] }|j V d S r )ru rD r s r r. z0SqliteMigrator._update_column.<locals>.<genexpr> sL F F k'')) F F F F F Fr zColumn "z" does not exist on ""z\s+r} c 6 g | ]}| S r3 stripr cols r r z1SqliteMigrator._update_column.<locals>.<listcomp> s <<<ssyy{{<<<r )zforeign zprimary z constraint zcheck c | S r r3 r3 s r <lambda>z/SqliteMigrator._update_column.<locals>.<lambda> s * r c d S r r3 rO s r rP z/SqliteMigrator._update_column.<locals>.<lambda> s d r c @ j dz | S )NzFOREIGN KEY ("%s") )fk_resub)r3
new_columnr s r rP z/SqliteMigrator._update_column.<locals>.<lambda>
s" djnn%
2/ / r r __tmp__z
("?)%s("?)z\1%s\2, zDROP TABLE IF EXISTSr r z INSERT INTOc , g | ]}t | S r3 r rL s r r z1SqliteMigrator._update_column.<locals>.<listcomp>' s F F FcfSkk F F Fr SELECTc , g | ]}t | S r3 r rL s r r z1SqliteMigrator._update_column.<locals>.<listcomp>) s HHH36#;;HHHr FROMz
DROP TABLEc | j S r )rb )idxs r rP z/SqliteMigrator._update_column.<locals>.<lambda>6 s r )%setr get_columnsrD r[ rE get_indexesr rerT column_researchgroupscolumn_split_refindallcolumn_name_rematchr
startswithr0 zipgetrS compileIrA r r r rK r r r filterrH rb
_fix_index) r re column_to_updater7 rH create_tableindexes
raw_create raw_columns
split_columns column_defsnew_column_defsnew_column_namesoriginal_column_namesconstraint_termsr3 rf new_column_deforiginal_to_new fk_filter_fncleaned_columnsrh
temp_tablergxcreatequeriespopulate_table
drop_originalr rb rU s ` @r _update_columnz SqliteMigrator._update_column sD F F$(M$=$=e$D$DF F F F F ! ! # #7 2 2 * . . . 7 8 8
8 # 4 4U ; ;| -++E22
&&u--- vfc<88 #'."7"7 "E"E"L"L"N"N
K ,44[AA
<<m<<< " L% > >J .44Z@@GGIILK ...!#K!<!<! 9#**>:::)00===#'#6#<#<&$( $((. !K$++K888 &&z222 "''))445EFF >$++K888)00=== s#8:J K KLL$(()9::
44 22LL
+
+
+ L ) 3 3JJ$$Z00E U\\^^A%6:J%J%J )\* 5 5
3 &&z222 Y&
j -rt44 #
))O,, c0116*3E3EF G G V\\^^^^WWW5 6 68
"
: F F5E F F F G G MM HH2GHHH I I KK 5MM# !#l"3"3VE]]!CDD
j% 0 02 2 / /99 - -E u}44s59~~....
-ooei1A:NN?NN3s88,,,r c | | }t | dk r| || S | dd \ }}t | | dk r|d| || S | dd d d }d |D }g }|D ]I} t j d|z | r|| t | d z } | | J|dd d
|D dS ) N (r r r ,c 8 g | ]}| d S )z"`[]' rJ )r parts r r z-SqliteMigrator._fix_index.<locals>.<listcomp>R s$ ;;;T4::i((;;;r z%s(?:[\'"`\]]?\s|$)rW c 3 K | ] }d |z V
dS )z"%s"Nr3 )r r s r r. z,SqliteMigrator._fix_index.<locals>.<genexpr>] s& )D)D&1*)D)D)D)D)D)Dr )splitrB replacersplitra rh r rA )
r rb rp rU r lhsrhsrH cleanrq s
r ro zSqliteMigrator._fix_index@ sf *++ u::??;; /<< < ::c1%%S syy)** + +q 0 0 !cc3;;/?#L#L#LM M
3""1%++C00;;U;;;
! !Fx.1AA6JJ
E#fS1A-B-B-C-C&DD LL 33 )D)De)D)D)D D D D DEEr TFc t j dk ra|s_| }| || d t
| |S | ||d S )N) # r r c d S r r3 )abs r rP z,SqliteMigrator.drop_column.<locals>.<lambda>g s D r )r sqlite_version_inforR rl ra rb r r )r re rf r legacyrk s r r zSqliteMigrator.drop_column_ s &* 4 4V 4##%%C
sE
*
*go&&c&%%&&&J""5+7H7HIIIr c | t j dk r|s| | | d t
| d t
S fd}| ||| S )N)r r r r c 0 | | S r r )rf r3 r s r _renamez-SqliteMigrator.rename_column.<locals>._renamer s %%k8<< <r )r r rl rR ra rb r r )r re r r r r s ` r r z SqliteMigrator.rename_columni s &* 4 4V 4!\$"3"3"5"5u== W .//S))** WV__S))**
, = = = = =""5(G<<<r c 6 d }| ||| S )Nc |dz S )Nz NOT NULLr3 rf r3 s r
_add_not_nullz2SqliteMigrator.add_not_null.<locals>._add_not_nullx s
+ +r r )r re rq r s r r zSqliteMigrator.add_not_nullv s) , , ,""5&-@@@r c 6 d }| ||| S )Nc . | dd S )Nr r r s r _drop_not_nullz4SqliteMigrator.drop_not_null.<locals>._drop_not_null~ s %%j"55 5r r )r re rq r s r r z SqliteMigrator.drop_not_null| s) 6 6 6""5&.AAAr c t d t r
t t r. d s sdz fd}| ||| S )Nr r z'%s'c |dz z S )Nz DEFAULT %sr3 )rf r3 r` s r _add_defaultz7SqliteMigrator.add_column_default.<locals>._add_default s
77 7r )r[ r
r# r r isdigitr )r re rq r` r s ` r r z!SqliteMigrator.add_column_default s ? ?@@ @ W giiG w $ $ 'W-=-=j-I-I 'OO%% ' w&G 8 8 8 8 8 ""5&,???r c 6 d }| ||| S )Nc l t j dd|t j }| S )Nz DEFAULT\s+[\w"\'\(\)]+(\s|$)r )ra rT rm rK )rf r3 rM s r
_drop_defaultz9SqliteMigrator.drop_column_default.<locals>._drop_default s( &8"j"$OOC99;; r r )r re rq r s r r z"SqliteMigrator.drop_column_default s+ ""5&-@@@r Nc |t d | fd}| || S )Nz6alter_column_type() does not support cast with Sqlite.c } t | \ }}|S r )rv rb r r )rf r3 node_listrb r= rq rk rg s r _alter_column_typez<SqliteMigrator.alter_column_type.<locals>._alter_column_type sL #IWWVF^^,,00;;AACCFCJr )r[ rR r )r re rq rg r r rk s `` @r r z SqliteMigrator.alter_column_type sq ' ( (
( !! ""5&2DEEEr c t r r r s r r z SqliteMigrator.add_constraint r r c t r r r s r r z SqliteMigrator.drop_constraint r r c t r r )r re rf rg r r s r r z)SqliteMigrator.add_foreign_key_constraint s
"!r )TF)Fr r )r/ r0 r1 r2 ra rl rb re rg rm rS rB rE r; r ro r r r r r r r r r r r3 r r rZ rZ s
+,,I bj!899O RZ 455N BJ:BD A AE5 5 5 q q Yqf F F F> J J J YJ
=
=
= Y
= A A YA
B B YB
@ @ Y @ A A YA F F F Y F " " Y" " " Y" =A" " " Y" " "r rZ c 8 | D ]}| d S r )r% )r r r; s r migrater s*
r ) r2 collectionsr r9 rC ra peeweer r r r r
r r r
r r r playhouse.cockroachdbr ImportErrorobjectr r; rK rM rV rT r rX rZ r r3 r r <module>r s q qd # " " " " " # # # # # # , , , , , , 7777777 : : : : : : : :6 Q* Q* Q* Q* Q*V Q* Q* Q*h+ + + + + + + +\ & & & & &, & & & **Y )= > > FI% I% I% I% I%N I% I% I%Xw" w" w" w" w"^ w" w" w"t s A# #A-,A-