/�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) �
MySQLMigrator�PostgresqlMigrator�SchemaMigrator�SqliteMigrator� Operation�SQL�Entity�Clause�PostgresqlDatabase� operation�SqliteDatabase�
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)�cls�database)� __class__� � /migrator.pyr s
z SchemaMigrator.from_databaseTc s � � f d d � S)Nc s � j d � � S)N�cascade)�
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_column�null�extend� add_not_null)r �table� column_name�fieldZ
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 �compiler�field_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 �sql�paramsr 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 �pw�ForeignKeyFieldr Zget_inline_fk_sql�namer )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 �nodes�insertr )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 �str�orm)�migratorr �args�kwargs)�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, ZProxy�objr �dictr? �list�opsr 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 �run�clean)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? �_meta�db_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 �values�pop�
__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_name�new_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 Zdefaults�defaultrJ 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: