/Xp* @ s d d l Z d d l m Z m Z m Z m Z
m Z m Z m
Z
m Z m Z m Z m Z m Z d d l m Z Gd d d e Z Gd d d e e Z Gd d d e e Z Gd
d d e e
Z d d
Z Gd d d e Z d S) N)
MySQLMigratorPostgresqlMigratorSchemaMigratorSqliteMigrator OperationSQLEntityClausePostgresqlDatabase operationSqliteDatabase
MySQLDatabase)LOGGERc s e Z d Z d Z e f d d Z d d d Z e d d Z d d
Z e d d Z
e d
d Z S)r zImplement migrations.c sa t | t r t | St | t r2 t | St | t rK t | St t | j | S)zInitialize migrator by db.)
isinstancer
r r r r
r superr
from_database)clsdatabase) __class__ /migrator.pyr s
z SchemaMigrator.from_databaseTc s f d d S)Nc s j d S)Ncascade)
drop_tabler )r modelr r <lambda> s z+SchemaMigrator.drop_table.<locals>.<lambda>r )selfr r r )r r r r s zSchemaMigrator.drop_tablec C sA | j | | | g } | j s= | j | j | | g | S)zChange column.)alter_change_columnnullextend add_not_null)r table column_namefieldZ
operationsr r r
change_column s z SchemaMigrator.change_columnc C sY | j d } | _ | j j j | } | | _ t t d t | t d | S)zSupport change columns.Tz ALTER TABLEz ALTER COLUMN)r r compilerfield_definitionr r r )r r columnr"
field_null field_clauser r r r ' s z"SchemaMigrator.alter_change_columnc G s t t | | S)zExecute raw SQL.)r r )r sqlparamsr r r r) . s zSchemaMigrator.sqlc C s | j d } | _ | | _ | j j j | } | | _ t d t | t d | g } t | t j r | j
| j | n | | _ t
| S)zKeep fieldname unchanged.Tz ALTER TABLEz
ADD COLUMN)r db_columnr r$ r% r r r pwForeignKeyFieldr Zget_inline_fk_sqlnamer )r r r! r" r' r( partsr r r alter_add_column3 s z SchemaMigrator.alter_add_column) __name__
__module__ __qualname____doc__ classmethodr r r r# r r) r0 r r )r r r
s r c s( e Z d Z d Z f d d Z S)r z Support the migrations in MySQL.c sH t t | j | | | } | j d } | j j d t d | S)zSupport change columns. TYPE)r r r nodesinsertr )r r r! r" clauser( )r r r r K s
z!MySQLMigrator.alter_change_column)r1 r2 r3 r4 r r r )r r r G s r c s( e Z d Z d Z f d d Z S)r z%Support the migrations in postgresql.c sH t t | j | | | } | j d } | j j d t d | S)zSupport change columns.r6 r7 r8 )r r r r9 r: r )r r r! r" r; r( )r r r r W s
z&PostgresqlMigrator.alter_change_column)r1 r2 r3 r4 r r r )r r r S s r c @ s1 e Z d Z d Z d d d Z d d Z d S)r z!Support the migrations in sqlite.Tc s f d d S)z0SQLite doesnt support cascade syntax by default.c s j d d S)Nr F)r r )r r r r e s z+SqliteMigrator.drop_table.<locals>.<lambda>r )r r r r )r r r c s zSqliteMigrator.drop_tablec s( f d d } j | | | S)zSupport change columns.c s7 j j } | j } | j | \ } } | S)N)r r$ r% Z
parse_node)r! Z
column_defr$ r; r) _)r" r r r _changei s z3SqliteMigrator.alter_change_column.<locals>._change)Z_update_column)r r r& r" r= r )r" r r r g s z"SqliteMigrator.alter_change_columnN)r1 r2 r3 r4 r r r r r r r _ s r c s f d d } | S)z Convert string to model class.c s< t | t r) | | j | | | S | | | | S)N)r strorm)migratorr argskwargs)methodr r wrappers s zget_model.<locals>.wrapperr )rC rD r )rC r get_modelq s rE c @ sW e Z d Z d Z d d Z d d Z d d Z d d Z d
d Z d d
Z e Z
e d d d Z e Z
e d d Z e Z e d d Z e Z e d d Z e Z d d Z e d d Z e Z e d d Z e d d Z e d d Z e d! d" Z e d# d$ Z e d% d& Z d' S)(MigratorzProvide migrations.c C sU t | t j r | j } | | _ t | _ t | _ t j
| j | _ d S)zInitialize the migrator.N) r r, ZProxyobjr dictr? listopsr r r@ )r r r r r __init__~ s zMigrator.__init__c C s^ xM | j D]B } t | t rE t j d | j | j | j q
| q
W| j d S)zRun operations.z%s %sN) rJ r r r inforC rA runclean)r Zopnr r r rM s
z Migrator.runc s& | j j f d d d S)zRun python code.c s
S)Nr r )rA funcrB r r r s z!Migrator.python.<locals>.<lambda>N)rJ append)r rO rA rB r )rA rO rB r python s zMigrator.pythonc G s# | j j | j j | | d S)zExecure raw SQL.N)rJ rP r@ r) )r r) r* r r r r) s z Migrator.sqlc C s t | _ d S)zClean the operations.N)rI rJ )r r r r rN s zMigrator.cleanc C s9 | | j | j j <| j | j _ | j j | j | S)zUCreate model and table in database.
>> migrator.create_table(model)
)r? _metadb_tabler rJ rP create_table)r r r r r rT s zMigrator.create_tableTc C s3 | j | j j =| j j | j j | | d S)zaDrop model and table from database.
>> migrator.drop_table(model, cascade=True)
N)r? rR rS rJ rP r@ r )r r r r r r r s zMigrator.drop_tablec K s x | j D] \ } } | j | | | j j | j j | j j | j | | j r
| j j | j j
| j j | j f d d q
W| S)zCreate new fields.uniqueT) items add_to_classrJ rP r@ Z
add_columnrR rS r+ rU add_index)r r fieldsr. r" r r r add_columns s + #zMigrator.add_columnsc K s x | j D] \ } } | j | | | j j | j j | j j | j | | j r
| j j | j j
| j j | j f d d q
W| S)zChange fields.rU T) rV rW rJ rP r@ r# rR rS r+ rU rX )r r rY r. r" r r r change_columns s #zMigrator.change_columnsc s f d d | j j j D } | j d d } x | D] } | j | | | j r | j j } | j | j j | j
f } | j j | j
j | j j | | j j | j
j | j j | j
d | q> W| S)zRemove fields from model.c s% g | ] } | j k r | q Sr )r. ).0r" )namesr r
<listcomp> s z)Migrator.drop_columns.<locals>.<listcomp>r T)rR rY valuespop
__del_field__rU r r$
index_namerS r+ rJ rP r@
drop_indexZ drop_column) r r r] rB rY r r" r$ rb r )r] r drop_columns s %
% )zMigrator.drop_columnsc C s_ | j j | j t | | j t | t j r[ t | j | j | j j j | j =d S)zDelete field from model.N)
rR Z remove_fieldr. delattrr r, r- Z rel_modelZ related_nameZ reverse_rel)r r r" r r r ra s
zMigrator.__del_field__c C s | j j | } t | t j r+ | j } | j | | | | _ | _ | j | | t | t j r | j d | _ } | j j
| j j | j j
| | | S)zRename field in model._id)rR rY r r, r- r+ ra r. rW rJ rP r@
rename_columnrS )r r Zold_namenew_namer" r r r rg s (zMigrator.rename_columnc C sX | j | j j =| | j _ | | j | j j <| j j | j j | j j | | S)zRename table in database.)r? rR rS rJ rP r@ rename_table)r r rh r r r ri s
%zMigrator.rename_tablec O s | j d d } | j j j | | f g } xL | D]D } | j j j | } t | t j ro | d } | j | q8 W| j j | j
j | j j | d | | S)zCreate indexes.rU Frf )
r` rR indexesrP rY getr r, r- rJ r@ rX rS )r r columnsrB rU columns_colr" r r r rX s
+zMigrator.add_indexc s g } xL D]D } | j j j | } t | t j rD | d } | j | q
W| j j j j
| j j | } f d d | j j D | j _ | j
j | j j | j j | | S)z
Drop indexes.rf c s. g | ]$ \ } } | k r | | f q Sr r )r\ Zcolsr< )rl r r r^ s z'Migrator.drop_index.<locals>.<listcomp>)rR rY rk r r, r- rP r@ r r$ rb rS rj rJ rc )r r rl rm rn r" rb r )rl r rc s
$%%zMigrator.drop_indexc G sV xO | D]G } | j j | } d | _ | j j | j j | j j | j q W| S)z
Add not null.F) rR rY r rJ rP r@ r rS r+ )r r r] r. r" r r r r s
,zMigrator.add_not_nullc G sV xO | D]G } | j j | } d | _ | j j | j j | j j | j q W| S)zDrop not null.T) rR rY r rJ rP r@
drop_not_nullrS r+ )r r r] r. r" r r r ro ' s
,zMigrator.drop_not_nullc C sS | j j | } | | j j | <| _ | j j | j j | j j | | | S)z Add default.) rR rY ZdefaultsdefaultrJ rP r@ Z
apply_defaultrS )r r r. rp r" r r r add_default0 s (zMigrator.add_defaultN) r1 r2 r3 r4 rK rM rQ r) rN rT Z create_modelrE r Z remove_modelrZ Z
add_fieldsr[ Z
change_fieldsrd Z
remove_fieldsra rg Z rename_fieldri rX rc r ro rq r r r r rF z s4
rF )Zpeeweer, Zplayhouse.migrater ZMqMr ZPgMr ZScMr ZSqMr r r r r
r r r
Zpeewee_migrater rE objectrF r r r r <module> s R: