]j0 U d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZmZmZmZmZmZmZmZ dd lmZ ddlZ d
d l m Z m Z d
d l m!Z! d
d
l"m#Z# d
dl$m%Z%m&Z& d
dl'm(Z( er ddl)m*Z* ddl+m,Z, ej- dej. Z/de0d< e j1 Z2de0d< e2dz Z3de0d< d/dZ4 G d de5 Z6 G d d e6 Z7 G d d e6 Z8d0d#Z9d1d%Z:d& Z;d'd(d2d.Z<dS )3zMigration router. ) annotationsN)cached_property)
import_module)Path)
ModuleType)
TYPE_CHECKINGAnyFinalIterableListOptionalSetTypeUnion)mock )NEWLINE diff_many)logger)Migrator)
MIGRATE_TABLEMigrateHistory)TEMPLATE)Logger)
TModelTypez\s+$r
CLEAN_RECURDIR
migrationsDEFAULT_MIGRATE_DIRc d S N )mdfakes L/opt/imunify360/venv/lib64/python3.11/site-packages/peewee_migrate/router.pyvoidr' " s 4 c e Zd ZdZeddeefd4d
Zed5d Z e
d6d Z e
d7d Z e
d7d Z
ed8d Zdddd9d Zd:d;d Zd! Z d<d=d'Zd;d(Zd)ddd*d>d0Zd?dd1d@d2Zd3 ZdS )A
BaseRouterz Abstract base class for router.Ndatabase Union[pw.Database, pw.Proxy]ignoreOptional[Iterable[str]]schema
Optional[str]r r migrator_classType[Migrator]c , || _ || _ || _ || _ || _ || _ t
| j t j t j f st d|z t | j t st d|z dS )Initialize the router.zInvalid database: %szInvalid migrator_class: %sN)
r+
migrate_tabler/ r- r r1
isinstancepwDatabaseProxy TypeError
issubclassr )selfr+ r5 r- r/ r r1 s r& __init__zBaseRouter.__init__) s !
* ,$-"+rx)@AA ? 2X =>> >$-x88 E 88 CDD D E Er( returnType[MigrateHistory]c t j }| j |_ | j |_ | j |_ t j d t S )z,Initialize and cache MigrationHistory model.T)safe)r _metar+ r5
table_namer/ create_table)r< metas r& modelzBaseRouter.model> sB #
,k #.... r(
Iterable[str]c t )zGet migrations to run.NotImplementedErrorr< s r& todozBaseRouter.todoH s
"!r( List[str]c d | j | j j D S )z Scan migrations in database.c g | ] }|j
S r" name).0mms r&
<listcomp>z#BaseRouter.done.<locals>.<listcomp>P s NNNBNNNr( )rF selectorder_byidrK s r& donezBaseRouter.doneM s9 ON$*"3"3"5"5">">tz}"M"MNNNNr( c R t | j fd| j D S )z'Calculate difference between fs and db.c g | ]}|v| S r" r" )rR rQ rX s r& rT z#BaseRouter.diff.<locals>.<listcomp>V s# ???d$.>.>.>.>.>r( )setrX rL )r< rX s @r& diffzBaseRouter.diffR s. 49~~????????r( r c z | | j }| j D ]}| || |S )z2Create migrator and setup it with fake migrations.)r1 r+ rX run_one)r< migratorrQ s r& r_ zBaseRouter.migratorX sF &&t}55I ) )D LLx ( ( ( (r( autoF)r` rQ strr c X dx}}|rTt j t t |t
r|n|g}t
d |D sv |}t |t r%d t j t g D }d |D }n,# t $ r j d| Y dS w xY w j
r fd|D } j D ] } | j d t# j | }|s j d
dS t# j |d } j d | ||| } j d
| |S )zYCreate a migration.
:param auto: Python module path to scan for models.
c 3 4 K | ]}t | V d S r! ) _check_model)rR r# s r& <genexpr>z$BaseRouter.create.<locals>.<genexpr>j s( 771|A777777r( c g | ]
\ }}}|| S r" r" )rR _r# ispkgs r& rT z%BaseRouter.create.<locals>.<listcomp>n s4 # # #"-!QX]# # # #r( c 6 g | ]}t | D ]}|S r" ) load_models)rR moduler# s r& rT z%BaseRouter.create.<locals>.<listcomp>q s- S S SF{6?R?R S S!a S S S Sr( z Can't import models module: %sNc : g | ]}|j j j v|S r" )rB rQ r- )rR r# r< s r& rT z%BaseRouter.create.<locals>.<listcomp>x s) OOOqw|4;/N/N!/N/N/Nr( Tr% zNo changes found.reversezCreating migration "%s"z"Migration has been created as "%s")syspathappendr r6 listallboolpkgutil iter_modules ImportErrorr exceptionr- r\ r^ r_ compile_migrationswarninginfocompile)r< rQ r` migraterollbackmodelsmodules migrations ` r& createzBaseRouter.create` s
( O HOOM * * * 'd 3 3?TT$F7777777
$G!$-- # #181E&{m1T1T# # # T Sg S S SFF" K))*JDQQQ 44 {
POOOOVOOO !Y
B
B Y
D AAAA( ??G
##$7888t)$-NNNH 2D999||D'844 =tDDD s "AB+ +%CCinitialc t | j }t |t | j j }|s| j d S | | j d| t | j g }|
|||d }t | j }| ||dd | j d| dS )zMerge migrations into one.zCan't merge migrationszMerge migrations into "%s"r Tr% forcez$Migrations has been merged into "%s"N) r r+ r{ rt r_ ormr errorclearr} r~ r^ )r< rQ r_ r r s r& mergezBaseRouter.merge s DM**$XtDM4E/F/FGG ?;$$%=>> >
5t<<<%dmR88||D'8Q77DM** T8$d ;;; ?FFFtr( c \ | j dS )zClear migrations.N)rF deleteexecuterK s r& r zBaseRouter.clear s(
##%%%%%r( rc r r num
Optional[int]c t )Create a migration.rI )r< rQ r r r s r& r~ zBaseRouter.compile s
"!r( c t )Read migration from file.rI )r< rQ s r& readzBaseRouter.read s !!r( T)r% downgrader r_ r% rv r r c | | \ }}|rt j }d|j _ t j d 5 t j d| 5 ||| j | ddd n # 1 swxY w Y ddd n # 1 swxY w Y |r6| j | | j
d| g |_ |S | j 5 |sT| j
d| ||| j | | | j | n| j
d | ||| j | | | j
| j j |k | j
d| |cddd S # 1 swxY w Y dS # t" $ r= | j |sd
nd } | j d | | w xY w)
z(Run/emulate a migration with given name.Nzpeewee.Model.selectzpeewee.Database.execute_sql) return_valuern rP zDone %sz Migrate "%s"zRolling back %s MigrationRollbackz
%s failed: %s)r r Mock fetch_oner patchr+ rF r r r} __ops__ transactionr whererQ r Exceptionr rz )
r< rQ r_ r% r r r r
mocked_cursor operations
r& r^ zBaseRouter.run_one sX $ $ $ GX
$
7;
'4Z 566 @ @
1
9 9 9 @ @ GHdm$????@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @
6J%%4%000K$$Y555#% **,,
QK$$^T:::GHdm$???? HJJJJ%%4%0000K$$%6=== HXt}4@@@@ HJJJJ%%''--djo.EFFNNPPP D111
M " " $ $ $+4D *I K ! !/9d C C C s A
H B &B: B B
B
B
B H B!!H $B!%AH )H C7H9 H H
H
H
H AIrn c | j d g }| j }|s | j d |S | j }|D ]8}| | |||| |r||k r n9|S )zRun migrations.zStarting migrationszThere is nothing to migrater )r r} r\ r_ rs r^ )r< rQ r% rX r\ r_ mnames r& runzBaseRouter.run s ./// y K : ; ; ;K= E KK UH4t LL M M M
r( c | j }|sd}t | |d }| j }| ||dd | j d| dS )z Rollback the latest migration.z There is nothing to rollbackFT)r% r zDowngraded migration: %sN)rX RuntimeErrorr_ r^ r r| )r< rX msgrQ r_ s r& r zBaseRouter.rollback sm y $0C s## #Bx= T8%4 @@@ 6=====r( )
r+ r, r- r. r/ r0 r r r1 r2 )r> r? )r> rG )r> rM )r> r )rQ ra r` r r> r0 )r )rQ ra rc rc N)
rQ ra r ra r ra r r r> ra ) rQ ra r_ r r% rv r rv r rv r> ra r! )rQ r0 r% rv r> rM )__name__
__module__ __qualname____doc__r r r r= r rF propertyrL rX r\ r_ r r r r~ r r^ r r r" r( r& r* r* & s ))
$*. $ )1E E E E E* _ " " " X" O O O XO @ @ @ X@
_' ' ' ' ' ' 'R $& & &
VZ" " " " " " " " " . . . . . .`e $
>
>
>
>
>r( r* c x e Zd ZdZ ej d Z dd fd
Zed Z ddd Zd Z
fd
Z xZ S )RouterzFile system router.z[\d]{3}_[^\.]+\.py$N migrate_dirOptional[Union[str, Path]]c t t | j |fi | |t }n$t |t
rt
| }|| _ dS r4 N)superr r= r r6 ra r r )r< r+ r kwargs __class__s r& r= zRouter.__init__ sd %fd$X88888 -KK
S
)
) , {++K&r( c j s; j d j j d t fdt
j j D S )z Scan migrations in file system.z'Migration directory: %s does not exist.T)parentsc 3 ` K | ](}j | |d d V )d S )N)filemaskmatch)rR fr< s r& rf z Router.todo.<locals>.<genexpr> sA ]]dmFYFYZ[F\F\]af]]]]]]r( )r existsr r| mkdirsortedoslistdirrK s `r& rL z Router.todo s &&(( 1 K I4K[ \ \ \ " "4 " 0 0 0]]]]bj1A&B&B]]]]]]r( rc r> ra c 4 |t | j }d |dz |z }|dz }| j |z }| d 5 }| t
j ||| ddd n # 1 swxY w Y |S )r Nz{:03}_r .pyw)r r rQ )lenrL formatr openwriter )r< rQ r r r filenamerr r s r& r~ zRouter.compile s ;di..CsQw''$.%< (*
YYs^^ Xq
GGHOGhXVVV W W W X X X X X X X X X X X X X X X s +B
BBc Z | j |dz z }| d 5 }| }i }t |ddd }t ||d | dt | d t fcddd S # 1 swxY w Y dS )
r r rz<string>execT) dont_inheritNr r )r r r r~ r getr' )r< rQ rr r codescopes r& r z Router.read s 4%<0
YYs^^ Kq6688DE4V$GGGD ud # # #99Y--uyyT/J/JJ K K K K K K K K K K K K K K K K K Ks A0B B$'B$c t t | | j D ]#}| j |dz z }| $dS )zRemove migrations from fs.r N)r r r rL r unlink)r< rQ rr r s r& r z Router.clear* sZ
fd!!###I D#te|4D KKMMMM r( r! )r r r )r> ra )
r r r r rer~ r r= r rL r r
__classcell__r s @r& r r s rz011H
37 ' ' ' ' ' ' ' ^ ^ X^ K K K r( r c * e Zd ZdZd fd Zd Z xZS ) ModuleRouterzModule based router.r c t t | j |fi | t |t rt | }|| _ dS r )r r r= r6 ra r migrate_module)r< r+ r r r s r& r= zModuleRouter.__init__5 sU *lD!!*8>>v>>> nc * * ; *> : :N ,r( c t | j | }t |dt t |dt fS )z Read migrations from a module.r r )getattrr r' )r< rQ mods r& r zModuleRouter.read> s7 d)400sIt,,gc:t.L.LLLr( )r )r r r r r= r r r s @r& r r 2 sY - - - - - -M M M M M M Mr( r rl Union[str, ModuleType]r> Set[Type[pw.Model]]c h t | t r| gnt | }d |D S )z Load models from given module.c v h | ]5t t fd t D D ]}|6S )c 3 8 K | ]}t | V d S r! )r )rR rQ rl s r& rf z(load_models.<locals>.<setcomp>.<genexpr>K s- &U&Uwvt'<'<&U&U&U&U&U&Ur( )filterre dir)rR r# rl s @r& <setcomp>z load_models.<locals>.<setcomp>H se &U&U&U&UV&U&U&UVV
r( )r6 r _import_submodules)rl r s r& rk rk D sI %VZ88Xvhh>PQW>X>XG r( .c |du rt }t | t rt | } t | d s| hS g }t | j |z r|S |t | j z }t
j | j | j dz D ]i\ }}}| |d }||j
&|j
| }| | |r|t | z
}j|S )N.__path__.)r[ r6 ra r hasattrr rw
walk_packagesr find_specloader load_modulers r )packagepassedr r rQ is_pkgspecrl s r& r r O s!
}}'3 ) (( 7J ' ' yG
7 v%
c'"###F ' 5g6FHX[^H^ _ _ 2 2 f d++ <4;. ((..v 2 )&11 1G Nr( c t | t o)t | t j ot | d S )z/Check object if it's a peewee model and unique.rB )r6 typer; r7 Modelr )objs r& re re k s3 c4 XZRX%>%> X73PWCXCXXr( Fro r_ r r List[TModelType]ra c t | j }|r||}}t ||| | }|sdS t t j d | d z }t d| S )z$Compile migrations for given models.ro rc z
)rt r r r joinsplitr sub)r_ r rp sourcer r s r& r{ r{ p s
(,
F ( 668WEEEJ r W\&++j"9"9"?"?"E"EFF FD <<d # ##r( r! )rl r r> r ).)r_ r r r r> ra )=r
__future__r r rw r rq functoolsr importlibr pathlibr typesr typingr r r
r r r
r r r unittestr peeweer7 r` r r logsr r_ r r r r templater loggingr peewee_migrate.typesr r~ Mr __annotations__cwdr r r' objectr* r r rk r re r{ r" r( r& <module>r s " " " " " "
% % % % % % # # # # # # X X X X X X X X X X X X X X X X X X X X X X $ $ $ $ $ $ $ $ 1 1 1 1 1 1 1 1 0 ////// "*Wbd++ + + + +
#l 2 2 2 2 2 O> O> O> O> O> O> O> O>d7 7 7 7 7Z 7 7 7tM M M M M: M M M$ 8Y Y Y
QV $ $ $ $ $ $ $ $r(