آگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآ U /“Xp*م@s®ddlZddlmZmZmZm Z mZmZm Z mZmZmZmZmZddlmZGdd„deƒZGdd„deeƒZGdd „d eeƒZGd d„dee ƒZ dd „ZGdd„deƒZdS)éN)ع MySQLMigratorعPostgresqlMigratorعSchemaMigratorعSqliteMigratorع OperationعSQLعEntityعClauseعPostgresqlDatabaseع operationعSqliteDatabaseع MySQLDatabase)عLOGGERcsZeZdZdZe‡fdd„ƒZddd„Zedd„ƒZd d „Z edd„ƒZ ed d„ƒZ‡ZS)rzImplement migrations.csFt|tƒrt|ƒSt|tƒr$t|ƒSt|tƒr6t|ƒStt|ƒ |،S)zInitialize migrator by db.) ع isinstancer rrrr rعsuperrع from_database)عclsعdatabase©ع __class__©ْH/opt/alt/python38/lib/python3.8/site-packages/peewee_migrate/migrator.pyrs zSchemaMigrator.from_databaseTcs‡‡fdd„S)NcsˆjˆdچS)N©عcascade©ع drop_tabler©rعmodelrrعَz+SchemaMigrator.drop_table..r©عselfrrrrrrszSchemaMigrator.drop_tablecCs.| |||،g}|js*| | ||،g،|S)zChange column.)عalter_change_columnعnullعextendعadd_not_null)r!عtableعcolumn_nameعfieldZ operationsrrrع change_columnszSchemaMigrator.change_columncCs>|jd}|_|j ، |،}||_ttdƒt|ƒtdƒ|ƒS)ْSupport change columns.TْALTER TABLEzALTER COLUMN)r#rعcompilerعfield_definitionr rr)r!r&عcolumnr(ع field_nullعfield_clauserrrr"'sz"SchemaMigrator.alter_change_columncGstt|f|‍ژƒS)zExecute raw SQL.)r r©r!عsqlعparamsrrrr2.szSchemaMigrator.sqlcCsn|jd}|_||_|j ، |،}||_tdƒt|ƒtdƒ|g}t|tj ƒr`| | |،،n||_t |ژS)zKeep fieldname unchanged.Tr+z ADD COLUMN)r#ع db_columnrr,r-rrrعpwعForeignKeyFieldr$Zget_inline_fk_sqlعnamer )r!r&r'r(r/r0عpartsrrrعalter_add_column3süzSchemaMigrator.alter_add_column)T) ع__name__ع __module__ع__qualname__ع__doc__عclassmethodrrrr)r"r2r9ع __classcell__rrrrr s rcs eZdZdZ‡fdd„Z‡ZS)rz Support the migrations in MySQL.cs4tt|ƒ |||،}|jd}|j dtdƒ،|S©r*éےےےےéZTYPE)rrr"عnodesعinsertr©r!r&r'r(عclauser0rrrr"Ks z!MySQLMigrator.alter_change_column©r:r;r<r=r"r?rrrrrGsrcs eZdZdZ‡fdd„Z‡ZS)rz%Support the migrations in postgresql.cs4tt|ƒ |||،}|jd}|j dtdƒ،|Sr@)rrr"rCrDrrErrrr"Ws z&PostgresqlMigrator.alter_change_columnrGrrrrrSsrc@s"eZdZdZddd„Zdd„ZdS) rz!Support the migrations in sqlite.Tcs‡fdd„S)z0SQLite doesnt support cascade syntax by default.csˆjddچS)NFrrr©rrrrerz+SqliteMigrator.drop_table..rr rrHrrcszSqliteMigrator.drop_tablecs‡‡fdd„}ˆ |||،S)r*cs&ˆj ،}| ˆ،}| |،\}}|S©N)rr,r-Z parse_node)r'Z column_defr,rFr2ع_©r(r!rrع_changeis z3SqliteMigrator.alter_change_column.._change)Z_update_column)r!r&r.r(rLrrKrr"gsz"SqliteMigrator.alter_change_columnN)T)r:r;r<r=rr"rrrrr_s rcs‡fdd„}|S)zConvert string to model class.cs4t|tƒr"ˆ||j|f|‍|ژSˆ||f|‍|ژSrI)rعstrعorm)عmigratorrعargsعkwargs©عmethodrrعwrapperss zget_model..wrapperr)rSrTrrRrع get_modelqsrUc@sوeZdZdZdd„Zdd„Zdd„Zdd „Zd d„Zdd „Z e Z ed(dd„ƒZeZ edd„ƒZeZedd„ƒZeZedd„ƒZeZdd„Zedd„ƒZeZedd„ƒZedd„ƒZedd „ƒZed!d"„ƒZed#d$„ƒZed%d&„ƒZd'S))عMigratorzProvide migrations.cCs:t|tjƒr|j}||_tƒ|_tƒ|_t |j،|_dS)zInitialize the migrator.N)rr5ZProxyعobjrعdictrNعlistعopsrrrO)r!rrrrع__init__~szMigrator.__init__cCsD|jD]0}t|tƒr0t d|j|j،| ،q|ƒq| ،dS)zRun operations.z%s %sN) rZrrrعinforSrPعrunعclean)r!Zopnrrrr]ˆs zMigrator.runcs|j ‡‡‡fdd„،dS)zRun python code.cs ˆˆˆژSrIrr©rPعfuncrQrrr”rz!Migrator.python..N)rZعappend)r!r`rPrQrr_rعpython’szMigrator.pythoncGs|j |jj|f|‍ژ،dS)zExecure raw SQL.N)rZrarOr2r1rrrr2–szMigrator.sqlcCstƒ|_dS)zClean the operations.N)rYrZ)r!rrrr^ڑszMigrator.cleancCs*||j|jj<|j|j_|j |j،|S)zUCreate model and table in database. >> migrator.create_table(model) )rNع_metaعdb_tablerrZraعcreate_table)r!rrrrre‍s zMigrator.create_tableTcCs&|j|jj=|j |j ||،،dS)zaDrop model and table from database. >> migrator.drop_table(model, cascade=True) N)rNrcrdrZrarOrr rrrrھszMigrator.drop_tablecKsh| ،D]Z\}}| ||،|j |j |jj|j|،،|j r|j |jj |jj|jfddچ،q|S)zCreate new fields.T©عunique)عitemsعadd_to_classrZrarOZ add_columnrcrdr4rgع add_index©r!rعfieldsr7r(rrrعadd_columnsµsے zMigrator.add_columnscKsh| ،D]Z\}}| ||،|j |j |jj|j|،،|j r|j |jj |jj|jfddچ،q|S)zChange fields.Trf)rhrirZrarOr)rcrdr4rgrjrkrrrعchange_columnsآsےے zMigrator.change_columnsc s‍‡fdd„|jj ،Dƒ}| dd،}|D]n}| ||،|jrx|j ،}| |jj |j f،}|j |j |jj |،،|j |j j|jj |j |dچ،q*|S)zRemove fields from model.csg|]}|jˆkr|‘qSr)r7)ع.0r(©عnamesrrع سs z)Migrator.drop_columns..rTr)rcrlعvaluesعpopع __del_field__rgrr,ع index_namerdr4rZrarOع drop_indexZdrop_column) r!rrqrQrlrr(r,rvrrprعdrop_columnsذs ےzMigrator.drop_columnscCsF|j |j،t||jƒt|tjƒrBt|j|jƒ|jjj |j=dS)zDelete field from model.N) rcZremove_fieldr7عdelattrrr5r6Z rel_modelZrelated_nameZreverse_rel)r!rr(rrrruلs zMigrator.__del_field__cCs~|jj|}t|tjƒr|j}| ||،||_|_| ||،t|tjƒr^|jd|_}|j |j |jj ||،،|S)zRename field in model.ع_id)rcrlrr5r6r4rur7rirZrarOع rename_columnrd)r!rZold_nameعnew_namer(rrrr{észMigrator.rename_columncCs@|j|jj=||j_||j|jj<|j |j |jj|،،|S)zRename table in database.)rNrcrdrZrarOعrename_table)r!rr|rrrr}ùs zMigrator.rename_tablecOsz| dd،}|jj ||f،g}|D]0}|jj |،}t|tjƒrL|d}| |،q&|j |j j|jj||dچ،|S)zCreate indexes.rgFrzrf) rtrcعindexesrarlعgetrr5r6rZrOrjrd)r!rعcolumnsrQrgعcolumns_عcolr(rrrrjszMigrator.add_indexcsٹg}ˆD]0}|jj |،}t|tjƒr.|d}| |،q|jj ، |jj|،}‡fdd„|jjDƒ|j_|j |j |jj|،،|S)z Drop indexes.rzcs g|]\}}ˆ|kr||f‘qSrr)roعcolsrJ©r€rrrrsz'Migrator.drop_index..)rcrlrrr5r6rarOrr,rvrdr~rZrw)r!rr€rپr‚r(rvrr„rrwszMigrator.drop_indexcGs<|D]2}|jj|}d|_|j |j |jj|j،،q|S)z Add not null.F) rcrlr#rZrarOr%rdr4©r!rrqr7r(rrrr%s zMigrator.add_not_nullcGs<|D]2}|jj|}d|_|j |j |jj|j،،q|S)zDrop not null.T) rcrlr#rZrarOع drop_not_nullrdr4r…rrrr†'s zMigrator.drop_not_nullcCs>|jj|}||jj|<|_|j |j |jj||،،|S)zAdd default.) rcrlعdefaultsعdefaultrZrarOZ apply_defaultrd)r!rr7rˆr(rrrعadd_default0szMigrator.add_defaultN)T)r:r;r<r=r[r]rbr2r^reZcreate_modelrUrZremove_modelrmZ add_fieldsrnZ change_fieldsrxZ remove_fieldsrur{Zrename_fieldr}rjrwr%r†r‰rrrrrVzsH rV)Zpeeweer5Zplayhouse.migraterZMqMrZPgMrZScMrZSqMrrrr r rrr عpeewee_migraterrUعobjectrVrrrrعs8: