] @ s} d d l Z d d l Z d d l m Z d d l m Z d d l Z d d l Z d d l m Z d d l
m Z m Z d d l
m Z m Z d d l m Z m Z d d l m Z e j d e j Z e j j e j d
Z d d Z e e j j e j j e j j e d
Z! e! j" Z# Wd QRXGd d d e$ Z% Gd d d e% Z& Gd d d e% Z' d S) N)
import_module)
ModuleType)cached_property)LOGGERMigrateHistory) diff_manyNEWLINE) string_typesexec_in)Migratorz\s+$
migrationsc C s d S)N )mdr
r
/router.py<lambda> s r z template.txtc @ s e Z d Z d Z d e d d Z e d d Z e d d Z d d
d d Z
d
d
d d Z d d Z e d d Z
e d d Z e d d Z d d
d
d d Z d d
d d Z d d Z d S)
BaseRouterz Abstract base class for router.Zmigratehistoryc C sM | | _ | | _ | | _ t | j t j t j f sI t d | d S)NzInvalid database: %s)database
migrate_tablelogger
isinstancepwZDatabaseZProxy RuntimeError)selfr r r r
r
r __init__ s
zBaseRouter.__init__c C s/ | j t j _ | j t j _ t j d t S)z+Ensure that migrations has prepared to run.T)r r Z_metar Zdb_tableZ create_table)r r
r
r model" s
zBaseRouter.modelc C s
t d S)N)NotImplementedError)r r
r
r todo+ s zBaseRouter.todoautoFc s d } } rt t rW y t Wn% t k
rV | j j d SYn Xt t r t t d d f d d t D x$ | j
D] } | j | | j q W } t | j j
j } t | | d | j } | s
| j j d St t j d j | j d
} t j d
| } t | | d | j d d } t t j d j | j d
} t j d
| } | j j d
| | j | | | } | j j d | | S)zCreate a migration. z Can't import models module: %sc S s t | t o t | t j S)N)r type
issubclassr ZModel)r r
r
r r ; s z#BaseRouter.create.<locals>.<lambda>c 3 s | ] } t | Vq d S)N)getattr).0r )r r
r <genexpr>< s z$BaseRouter.create.<locals>.<genexpr>migratorzNo changes found.z
reverseTzCreating migration "%s"zMigration created %s)r r r ImportErrorr errorr listfilterdirdiffrun_oner% Zormvaluesr warnr joinsplitCLEAN_REsubinfo_create) r namer migraterollbackZ migrationZmodels1Zmodels2pathr
)r r create/ s6
%% %zBaseRouter.creater c C s
t d S)N)r )r r7 r8 r9 r
r
r r6 T s zBaseRouter._createc C s
t d S)N)r )r r7 r
r
r readW s zBaseRouter.readc C s d d | j j D S)z Scan migrations in database.c S s g | ] } | j q Sr
)r7 )r# Zmmr
r
r
<listcomp>] s z#BaseRouter.done.<locals>.<listcomp>)r select)r r
r
r doneZ s zBaseRouter.donec s) t | j f d d | j D S)z'Calculate difference between fs and db.c s" g | ] } | k r | q Sr
r
)r# r7 )r? r
r r= c s z#BaseRouter.diff.<locals>.<listcomp>)setr? r )r r
)r? r r- _ s zBaseRouter.diffc C s7 t | j } x! | j D] } | j | | q W| S)z2Create migrator and setup it with fake migrations.)r r r? r. )r r% r7 r
r
r r% e s zBaseRouter.migratorTc C s y| j | \ } } | r t j d 2 t j d | | | j d | Wd QRXWd QRX| r | j j d | | j j d | | j | S| j j d | | j j | s| | | j d | | j
| j j d | | j j d | nn | j j d | | | | j d | | j
| j j j | j j
| k j | j j d | Wd QRXWnH t k
r} z( | j j | j j d
| WYd d } ~ Xn Xd S) zRun a migration.zpeewee.Model.selectzpeewee.Query._executefakeNr7 zDone %sz Running "%s"zRolling back %szRolled back %szMigration failed: %s)r< mockZpatchr r r; r r5 ZcleanZ transactionrundeleteZwherer7 Zexecute Exceptionr9 exception) r r7 r% rA Z downgradeforcer8 r9 excr
r
r r. m s6 $
(
zBaseRouter.run_oneNc C s | j j d g } | j } | s9 | j j d | S| j } xJ | D]B } | j | | d | d | | j | | rI | | k rI PqI W| S)zRun migrations.zStarting migrationszThere is nothing to migraterA rG )r r5 r- r% r. append)r r7 rA r? r- r% Zmnamer
r
r rC s
zBaseRouter.runc C sy | j } | j } | s' t d | | d k rC t d | j } | j | | d d | j j d | d S)NzNo migrations are found. z$Only last migration can be canceled.FTzDowngraded migration: %s)stripr? r r% r. r r0 )r r7 r? r% r
r
r r9 s zBaseRouter.rollback)__name__
__module__ __qualname____doc__r r r r propertyr r; r6 r< r? r- r% r. rC r9 r
r
r
r r s % $r c sd e Z d Z e j d Z e f d d Z e d d Z d d d d Z
d d
Z S) Routerz[\d]{3}_[^\.]+\.py$c s& t t | j | | | | _ d S)N)superrR r migrate_dir)r r rT kwargs) __class__r
r r s zRouter.__init__c sd t j j j s; j j d j t j j t f d d t j j D S)z Scan migrations in file system.z'Migration directory: %s does not exist.c 3 s: | ]0 } j j | r d j | d d Vq d S)r N )filemaskmatchr1 )r# f)r r
r r$ s z Router.todo.<locals>.<genexpr>) osr: existsrT r r0 makedirssortedlistdir)r r
)r r r s
z Router.todor c
C s t | j } d j | d } | | d } t j j | j | } t | d & } | j t j d | d | Wd QRX| S)zCreate a migration.z{:03}_rJ z.pywr8 r9 N)
lenr formatr\ r: r1 rT openwriteMIGRATE_TEMPLATE)r r7 r8 r9 numprefixr: r[ r
r
r r6 s &zRouter._createc C sq t t j j | j | d H } | j } i } t | | | j d t | j d t f SWd QRXd S)zRead migration from file.z.pyr8 r9 N) rd r\ r: r1 rT r< r
getVOID)r r7 r[ codeZscoper
r
r r< s
%
z Router.read) rM rN rO recompilerY MIGRATE_DIRr rQ r r6 r< r
r
)rV r rR s
rR c s1 e Z d Z d f d d Z d d Z S) ModuleRouterr c sA t t | j | | t | t r4 t | } | | _ d S)N)rS ro r r r r migrate_module)r r rp rU )rV r
r r s zModuleRouter.__init__c C s4 t | j | } t | d t t | d t f S)Nr8 r9 )r" rp rj )r r7 modr
r
r r< s zModuleRouter.read)rM rN rO r r< r
r
)rV r ro s ro )(r\ rl importlibr typesr rB Zpeeweer r Zpeewee_migrater r Zpeewee_migrate.autor r Zpeewee_migrate.compatr r
Zpeewee_migrate.migratorr rm Mr3 r: r1 getcwdrn rj rd abspathdirname__file__tr< rf objectr rR ro r
r
r
r <module> s$ 6%