~~` d Z ddlmZ ddlmZ ddlmZmZmZm Z m
Z
m Z m Z m
Z
ddlZddlmZmZmZmZmZmZmZmZ ddlmZ ddlmZ dd lm Z dd
lm Z dd l!m"Z" d d
l#m$Z$ erd dl%m&Z&m'Z' G d d Z( G d d Z) G d de Z G d de e Z G d de e Z G d de e Z G d d Z*dS ) zRun migrations. ) annotations)suppress)
TYPE_CHECKINGAnyCallableDictListUnioncastoverloadN)SQLContext
MySQLDatabase OperationPostgresqlDatabaseSqliteDatabasemake_index_name operation)
MySQLMigrator)PostgresqlMigrator)SchemaMigrator)SqliteMigrator)depricated_method )logger)
TModelType TVModelTypec @ e Zd ZdZddZddZddZdd Zdd Zd
Z dS )ORM
__tables__
__models__selfc " i | _ i | _ d S Nr r# s w/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/peewee_migrate/migrator.py__init__z ORM.__init__# s 1313 modelr c L || j |j < || j |j j < d S r% r" __name__r! _meta
table_namer# r* s r' addzORM.add' s' */'27 .///r) c D | j |j = | j |j j = d S r% r, r0 s r' removez
ORM.remove+ s# OEN + OEK 2 3 3 3r) namestrreturnc | j | S r% )r" r# r4 s r' __getattr__zORM.__getattr__/ t$$r) c | j | S r% )r! r8 s r' __getitem__zORM.__getitem__2 r: r) c N t | j S r% )iterr" valuesr& s r' __iter__z ORM.__iter__5 s DO**,,---r) N)r# r )r* r )r4 r5 r6 r )
r-
__module__ __qualname__ __slots__r( r1 r3 r9 r< r@ r) r' r r s ,I4 4 4 48 8 8 84 4 4 4% % % %% % % %. . . . .r) r c e Zd ZdZd?dZd Zd Zd Zed@d Z edAd Z dBdZ dCdZ
dDdZ e e d Z
d@dZ e ed ZdddEd Z e ed ZdFd#Z e ed$ ZdFd%Z e ed& ZdddGd(Z e ed) ZdHd,Z e ed- ZdId/ZdJd0Zd1d2dKd4Z dKd5Z dLd6Z dLd7Z dMd;Z dNd<Z!dLd=Z"d>S )OMigratorzProvide migrations.database Union[pw.Database, pw.Proxy]c t | _ t |t j r|j }|| _ g | _ t | | _
dS )zInitialize the migrator.N) r orm
isinstancepwProxyobj __database____ops__r
from_database __migrator__)r# rG s r' r( zMigrator.__init__< sS h ) ) $ |H $9; * 8 8 B Br) c | j D ]p}t |t r5t j d|j |j | Lt j d|j | qg | _ dS )zRun operations.z%s %szRun %sN) rP rK r r infomethodargsrunr- )r# ops r' __call__zMigrator.__call__G sx , B"i((
GRY888 Hbk222 r) c * t | j S )zIterate over models.)r> rJ r& s r' r@ zMigrator.__iter__R s DH~~ r) c t | S )zySync the current snapshot but not run operations.
>> with migrator.sync():
>> # ... do changes
) SyncContextr& s r' fakez
Migrator.fakeV s 4 r) r* r r6 c d S r% rD r0 s r'
__get_model__zMigrator.__get_model__^ r) r5 r c d S r% rD r0 s r' r_ zMigrator.__get_model__b r` r) Union[TVModelType, str] Union[TVModelType, TModelType]c t |t rM|| j j v r| j j | S || j j v r
| j | S t d|z |S )zGet model by name.zModel %s not found)rK r5 rJ r" r!
ValueErrorr0 s r' r_ zMigrator.__get_model__f si eS ! ! ;+++x*511+++x& 1E 9:: : r) sqlc t t | j j |g|R }| j | dS )Execute raw SQL.N)r r rR rf rP append)r# rf paramsrX s r' rf z Migrator.sqlr sF
) 2T . 23 @ @ @ @
A
A B r) funcr c J | j fd dS )zRun a python function.c i S r% rD )rV rk kwargss r' z Migrator.run..y s DD$$9&$9$9 r) N)rP ri )r# rk rV rn s ```r' rW z Migrator.runw s1 9 9 9 9 9 9:::::r) pythonc |j }| j | | j |_ | j |j |S )zxCreate model and table in database.
:param model: Model class
>> migrator.create_model(Model)
)r. rJ r1 rO rG rP ri create_table)r# r* metas r' create_modelzMigrator.create_model} sG { U )
E ./// r) rr TcascadeUnion[str, TModelType]rv boolc | | }| j | | j | j || dS )zDrop model and table from database.
:param model: Model class or table name
:param cascade(true): Drop table with cascade
>> migrator.remove_model(Model, cascade=True)
ru N)r_ rJ r3 rP ri rR
drop_tabler# r* rv s r' remove_modelzMigrator.remove_model s] ""5)) D - 8 8 8 P PQQQQQr) rz fieldspw.Fieldc | | }|j }| D ]\ }}| || | j | j |j |j | |j
r@| j | j |j |j fd |S )Change fields.
:param model: Model class or table name
>> migrator.change_fields(Model, name=pw.CharField(null=True))
Tunique) r_ r. items add_fieldrP ri rR
add_columnr/ column_namer add_index)r# r* r} rs r4 fields r'
add_fieldszMigrator.add_fields s ""5)){!<<>> KD% NN4 ' ' ' L !,,OU%6
|
##%//%BSAU^b/cc r) add_columnsc | | }|j }| D ]\ }}|j || }|o|j }| || t |t j r8| j
| j
|j | ||j k r>| j
| j |j ||j t |t j r||j r|j nd}|j r|j nd} | j
| j |j |j |j j j |j j || q| j
| j |j |j | |j |j k r|j r[|j f|j f}
|j |
| j
| j j |j g|
R #|j }
t5 t6 5 |j |j fdf ddd n # 1 swxY w Y | j
| j |j |
|S )r RESTRICTTN) r_ r. r r} getr r rK rL ForeignKeyFieldrP ri rR drop_foreign_key_constraintr/
rename_column on_delete on_updateadd_foreign_key_constraint rel_model rel_fieldr4
change_columnr indexesr r re r3
drop_index) r# r* r} rs r4 r old_fieldold_column_namer r indexs r'
change_fieldszMigrator.change_fields s# ""5)) !K!<<>> 2 Z 2 ZKD% e44I ' AI,AO NN4 ' ' ')R%788
##%AA$/Sbcc %"333 ##%33 %:K %!344
/4 NEOOJ /4 NEOOJ ##%@@ ) -8 ,!!
L !//OU%6
|y ///|
Z *,el: ##E*** ##$?D$5$?$XRW$X$X$XYYYY ) j)) F FL''%*;)=t(DEEEF F F F F F F F F F F F F F F ##D$5$@$@ RW$X$XYYYY s /#J J" %J" change_columnsnamesc | | }|j }fd|j D }|D ]}| || |j rSt
|j |j g }| j
| j |j | | j
| j
|j |j | |S )zRemove fields from model.
:param model: Model class or table name
:param cascade(true): Drop columns with cascade
>> migrator.remove_fields(Model, "name", "age", cascade=True)
c & g | ]
}|j v |S rD )r4 ).0r r s r'
z*Migrator.remove_fields.. s% QQQEUZ5=P=P%=P=P=Pr) ru )r_ r. r} r?
__del_field__r r r/ r rP ri rR r drop_column)r# r* rv r rs r} r
index_names ` r'
remove_fieldszMigrator.remove_fields s ""5)){QQQQT[%7%7%9%9QQQ E ue , , ,|
_ ,T_u?P>Q R R
##D$5$@$@ R\$]$]^^^ L !--OU%6 .
r) drop_columnsold_namenew_namec | | }|j }|j | }t |t j r|j }| || |x|_ |_ t |t j r|j dz |_ | || | j
| j
|j ||j |S )zRename field in model.
:param model: Model class or table name
:param old_name: Old field name
:param new_name: New field name
>> migrator.rename_field(Model, "name", "full_name")
_id)r_ r. r} rK rL r r r r4 r rP ri rR r r/ )r# r* r r rs r s r' rename_fieldzMigrator.rename_field s ""5)){ H% eR / 0 0 ) (H 5%((()11
U& eR / 0 0 : % 1E 9E x''' + +DOXuGX Y Y
r) r r c V |j }| |j t ||j t |t
j rX|j }|j |j k r|dz
}t || rt || t |j |j
dS dS )zDelete field from model.r N) r. remove_fieldr4 delattrrK rL r r hasattrr backref)r# r* r rs obj_id_names r' r zMigrator.__del_field__, s { %*%%%uz""" eR / 0 0 4 +K EJ..u$ uk**
,{+++ EOU] 3 3 3 3 3
4 4r) c ( | | }|j }|j }| j | ||_ | j | | j | j || |S )zRename table in database.)
r_ r. r/ rJ r3 r1 rP ri rR rename_table)r# r* r rs r s r' r zMigrator.rename_table9 s ""5)){? " U D - : :8X N NOOO r) Fr columnsc | | }|j }|j ||f g }|D ]X}|j | }t
| dk r||_ | |_ | |j Y| j
| j |j
|| |S )zCreate indexes.r r )r_ r. r ri r} r lenr r r rP rR r r/ )r# r* r r rs columns_colr s r' r zMigrator.add_indexD s ""5)){ Wf -... / /CKOOC((E7||q % "(j OOE - . . . . D - 7 7Z` 7 a abbb r) c | | }|j }g }D ]Z}|j | }|s t dk rdx|_ |_ | |j [t |j
| }fd|j D |_ | j | j
|j
| |S )z
Drop indexes.r Fc * g | ]\ }}|k ||fS rD rD )r cols_r s r' r z'Migrator.drop_index..f s& SSSitQ7d??q ???r) )r_ r. r} r r r r ri r r r/ r rP rR r )r# r* r rs r r r r s ` r' r zMigrator.drop_indexV s ""5)){ / /CKOOC((E
7||q -22 u{ OOE - . . . .$T_h??
SSSS4<SSS D - 8 8 * U UVVV r) c | | }|j }|D ]S}|j | }d|_ | j | j |j |j T|S )z
Add not null.F)
r_ r. r} nullrP ri rR add_not_nullr/ r r# r* r rs r4 r s r' r zMigrator.add_not_nullj sw ""5)){ d dDK%E EJ L 1 > >tPUPa b b c c c c r) c | | }|j }|D ]S}|j | }d|_ | j | j |j |j T|S )zDrop not null.T)
r_ r. r} r rP ri rR
drop_not_nullr/ r r s r' r zMigrator.drop_not_nullt sw ""5)){ e eDK%E EJ L 1 ? ?QVQb c c d d d d r) r4 defaultr c | | }|j }|j | }|x|j |< |_ | j | j |j || |S )z Add default.)
r_ r. r} defaultsr rP ri rR
apply_defaultr/ )r# r* r4 r rs r s r' add_defaultzMigrator.add_default~ sn ""5)){ D!/66
e u} D - ; ;DOTSX Y YZZZ r) c | | }|j }| j | j |j || |S )zAdd constraint.)r_ r. rP ri rR add_constraintr/ )r# r* r4
constraintrs s r' r zMigrator.add_constraint sO ""5)){ D - < . s. X X X$T
.
.t
E
E X X Xr) )r_ r. rP extend)r# r* r rs s ` @r' drop_constraintszMigrator.drop_constraints sZ ""5)){ X X X X XRW X X X
r) N)rG rH )r* r r6 r )r* r5 r6 r )r* rb r6 rc )rf r5 )rk r )r* rw rv rx )r* rw r} r~ r6 r )r* rw r r5 rv rx r6 r )r* rw r r5 r r5 r6 r )r* r r r~ )r* rw r r5 r6 r )r* rw r r5 r6 r )r* rw r r5 r6 r )r* rw r4 r5 r r r6 r )r* rw )#r- rA rB __doc__r( rY r@ r] r r_ rf rW r rp rt rr r| rz r r r r r r r r r r r r r r r r r rD r) r' rF rF 9 s C C C C ! ! ! X X
; ; ; ;
sH
-
-F %$\>BBLMQ
R
R
R
R
R
R #"< >>J . $#J
>>K= = = =~ '&}6FGGN KO 2 %$]NCCL 4 &%lODDM 4 4 4 4 NS $ ( r) rF c e Zd ZdZed d Zddd d
Zed d Zed d Z ed!d Z
d"dZ d# fd Z xZ
S )$r zImplement migrations.rG rH r6 c t |t rt | S t |t rt | S t |t
rt
| S t d|z )zInitialize migrator by db.zUnsupported database: %s)rK r r r r r r re )clsrG s r' rQ z SchemaMigrator.from_database sy h 2 3 3 0%h// / h / / ,!(++ + h
. . + ** *3h>???r) Tru r* r rv rx Callable[[], Any]c fdS )z Drop table.c 0 S )Nru rz )rv r* s r' ro z+SchemaMigrator.drop_table.. s u'''88 r) rD r{ s ``r' rz zSchemaMigrator.drop_table s 888888r) rf r5 r
c t |g|R S )rh )r
)r# rf rj s r' rf zSchemaMigrator.sql s 3 r) tabler r r~ List[Union[Context, Operation]]c | ||| }|j s)| | || |S )zChange column.)alter_change_columnr ri r )r# r r r
operationss r' r z SchemaMigrator.change_column sW
7;6N6N ;7
7
z E d / /{ C C D D Dr) columnc |j }t | r
| }| | || }| d }| | | S )Nz
SET DEFAULT )r callable
_alter_column make_contextliteralrf db_value)r# r r r r alter_columnctxs r' r zSchemaMigrator.add_default sy - G giiG#'#5#5d6G6G6I6I5RX#Y#Y &..??wwu~~g..///r) c | }|j dc}|_ | || d | | }||_ |gS )Support change columns.T ALTER COLUMN r r _alter_tabler rf ddlr# r r r r
field_nulls r' r z"SchemaMigrator.alter_change_column sp !!!&T
EJ U++334DEEII%))TW..YY
u r) r c |j } t t | j |||fi |}t |t
j r||_ |S )z Fix fieldname for ForeignKeys.)r4 superr alter_add_columnrK rL r )r# r r r rn r4 rX __class__s r' r z SchemaMigrator.alter_add_column sV z
9U>4
(
(
9%e
^
^W]
^
^ eR / 0 0 EJ r) )rG rH r6 r )r* r rv rx r6 r )rf r5 r6 r
)r r5 r r5 r r~ r6 r )r r5 r r5 r r~ r r5 r r5 r r~ r6 r )r r5 r r5 r r~ r6 r )r- rA rB r classmethodrQ rz r rf r r r r
__classcell__)r s @r' r r s @ @ @ [ @ @D 9 9 9 9 9 9 ! ! ! Y! Y 0 0 0 Y0 r) r c e Zd ZdZd d Zd
S ) r zSupport MySQL.r r5 r r r~ r6 r c | }|j dc}|_ | || d | | }||_ |gS )r Tz MODIFY COLUMN r r s r' r z!MySQLMigrator.alter_change_column sp !!!&T
EJ U++334EFFJJ599UX>>ZZ
u r) Nr r- rA rB r r rD r) r' r r s. r) r c e Zd ZdZd d Zd
S ) r z%Support the migrations in postgresql.r r5 r r r~ r6 r c | }|j dc}|_ |j g c}|_ | | }|j dt
j d | || d
| }||c|_ |_ |g}|j *| |
||| |S )r Tr TYPEr )r r constraintsr nodesinsertrL r
r r rf r ri r ) r# r r r r fnfcr ress r' r z&PostgresqlMigrator.alter_change_column s !! TEJ % 12 E iinn BF6NN+++ U++334DEEII#NN(*B%
E%e = $ JJt''vu== > > >
r) Nr r rD r) r' r r s. // r) r c ( e Zd ZdZdddd
ZddZdS )r z!Support the migrations in sqlite.Tru r* pw.Modelrv rx r6 r c fdS )z0Sqlite doesnt support cascade syntax by default.c 0 d S )NFru r )r* s r' ro z+SqliteMigrator.drop_table.. s u'''66 r) rD r{ s ` r' rz zSqliteMigrator.drop_table s 66666r) r r5 r r r~ List[Union[Operation, Context]]c B fd} ||| gS )r c }| | | d S )Nr )r rf r query)c_namec_defr r r# s r' r z.SqliteMigrator.alter_change_column..fn
sB ##%%C GGEIIcNN # # #99;;q> !r) )_update_column)r# r r r r s ` ` r' r z"SqliteMigrator.alter_change_column s?
" " " " " "
##E626677r) N)r* r rv rx r6 r )r r5 r r5 r r~ r6 r )r- rA rB r rz r rD r) r' r r sO ++=A 7 7 7 7 7 7
8
8
8
8
8
8r) r c e Zd Zd Zd Zd ZdS )r\ c " || _ d | _ d S r% )migratorops)r# r s r' r( zSyncContext.__init__ s
r) c B t | j j | _ | S r% )listr rP r r& s r' __enter__zSyncContext.__enter__ s
-.. r) c ( | j | j _ d S r% )r r rP )r# exc_typeexc_valexc_tbs r' __exit__zSyncContext.__exit__ s $
r) N)r- rA rB r( r r rD r) r' r\ r\ sA ) ) ) ) )r) r\ )+r
__future__r
contextlibr typingr r r r r r
r r peeweerL playhouse.migrater
r r r r r r r r MqMr PgMr ScMr SqMpeewee_migrate.utilsr logsr typesr r r rF r\ rD r) r' r& s " " " " " " R R R R R R R R R R R R R R R R R R R R 3 2 2 2 2 2 7 7 7 7 7 7 3 3 3 3 3 3 3 3 3 3 3 3 2 2 2 2 2 2 /......... . . . . . . .2\ \ \ \ \ \ \ \~
A A A A AS A A AH NC (8 8 8 8 8^S 8 8 8(
)
)
)
)
)
)
)
)
)
)r)