U
p_ @ s d dl Z d dlZd dlmZ d dlmZ d dlZd dlZd dl m Z d dl
m Z m Z d dl
mZmZ d dlmZmZ d dlmZ ed ejZe je d
Zd d Z e e je j e j e d
Z!e!" Z#W 5 Q R X G dd de$Z%G dd de%Z&G dd de%Z'dS ) 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> r z template.txtc @ s e Zd ZdZdefddZedd Zedd Z d d d Z
d!ddZ dd Z edd Z
edd Zedd Zd"ddZd#d d Zd d ZdS )$
BaseRouterz Abstract base class for router.Zmigratehistoryc C s6 || _ || _|| _t| j tjtjfs2td| d S )NzInvalid database: %s)database
migrate_tablelogger
isinstancepwZDatabaseZProxy RuntimeError)selfr r r r r r __init__ s
zBaseRouter.__init__c C s" | j tj_ | jtj_td tS )z+Ensure that migrations has prepared to run.T)r r Z_metar Zdb_tableZ create_tabler r r r model" s
zBaseRouter.modelc C s t d S r
NotImplementedErrorr r r r todo+ s zBaseRouter.todoautoFc sP d }} r"t trHz t W n" tk
rF | jd Y S X t trvttdd fddt D | j
D ]}| || j q| }t| j j
}t||| j d}|s| jdS ttd |d
}td
|}t||| j d d }ttd |d
}td
|}| jd
| | |||}| jd| |S )Create a migration. z Can't import models module: %sc S s t | tot| tjS r
)r type
issubclassr ZModel)r r r r r ; r z#BaseRouter.create.<locals>.<lambda>c 3 s | ]}t |V qd S r
)getattr).0r r" r r <genexpr>< s z$BaseRouter.create.<locals>.<genexpr>)migratorzNo changes found.z
T)r+ reversezCreating migration "%s"zMigration created %s)r r r ImportErrorr errorr listfilterdirdiffrun_oner+ Zormvaluesr warnr joinsplitCLEAN_REsubinfo_create) r namer" migraterollbackZ migrationZmodels1Zmodels2pathr r) r create/ s8
zBaseRouter.creater$ c C s t d S r
r )r r= r> r? r r r r< T s zBaseRouter._createc C s t d S r
r )r r= r r r readW s zBaseRouter.readc C s dd | j D S )z Scan migrations in database.c S s g | ]
}|j qS r r= )r( Zmmr r r
<listcomp>] s z#BaseRouter.done.<locals>.<listcomp>)r Zselectr r r r doneZ s zBaseRouter.donec s t | j fdd| jD S )z'Calculate difference between fs and db.c s g | ]}| kr|qS r r )r( r= rE r r rD c s z#BaseRouter.diff.<locals>.<listcomp>)setrE r! r r rF r r3 _ s
zBaseRouter.diffc C s& t | j}| jD ]}| || q|S )z2Create migrator and setup it with fake migrations.)r r rE r4 )r r+ r= r r r r+ e s
zBaseRouter.migratorTc
C st z.| |\}}|rtd, td ||| j|d W 5 Q R X W 5 Q R X |rr| jj|d | jd| | |W S | jd| | j |s||| j|d |
| jj|d | jd| nP| jd| ||| j|d |
| j | jj
|k | jd| W 5 Q R X W n> tk
rn } z | j | jd | W 5 d
}~X Y nX d
S ) zRun a migration.zpeewee.Model.selectzpeewee.Query._execute)fakerC zDone %sz Running "%s"zRolling back %szRolled back %szMigration failed: %sN)rB mockZpatchr r rA r r; ZcleanZ transactionrundeletewherer= Zexecute Exceptionr? Z exception) r r= r+ rH Z downgradeforcer> r? excr r r r4 m s6 $
zBaseRouter.run_oneNc C sj | j d g }| j}|s*| j d |S | j}|D ]0}| j||||d || |r4||kr4 qfq4|S )zRun migrations.zStarting migrationszThere is nothing to migrate)rH rN )r r; r3 r+ r4 append)r r= rH rE r3 r+ Zmnamer r r rJ s
zBaseRouter.runc C sV | }| j}|std||d kr.td| j}| ||dd | jd| d S )NzNo migrations are found.z$Only last migration can be canceled.FTzDowngraded migration: %s)striprE r r+ r4 r r6 )r r= rE r+ r r r r? s zBaseRouter.rollback)r" F)r$ r$ )TFF)NF)__name__
__module__ __qualname____doc__r r r r propertyr! rA r< rB rE r3 r+ r4 rJ r? r r r r r s$
%
$
r c sH e Zd ZedZef fdd Zedd Z d ddZ
d d
Z Z S ) Routerz[\d]{3}_[^\.]+\.py$c s t t| j|f| || _d S r
)superrX r migrate_dir)r r rZ kwargs __class__r r r s zRouter.__init__c sH t j js* jd j t j t fddt jD S )z Scan migrations in file system.z'Migration directory: %s does not exist.c 3 s, | ]$} j |rd |dd V qdS )r$ N)filemaskmatchr7 )r( fr r r r* s z Router.todo.<locals>.<genexpr>) osr@ existsrZ r r6 makedirssortedlistdirr r r r r! s
z Router.todor$ c C sb t | j}d|d }|| d }tj| j|}t|d}|t j||d W 5 Q R X |S )r# z{:03}_ .pyw)r> r? )
lenr! formatrb r@ r7 rZ openwriteMIGRATE_TEMPLATE)r r= r> r? Znumprefixr@ ra r r r r< s
zRouter._createc
C s^ t tj| j|d <}| }i }t|| |dt|dtfW 5 Q R S Q R X dS )zRead migration from file.rh r> r? N) rl rb r@ r7 rZ rB r
getVOID)r r= ra codeZscoper r r rB s
z Router.read)r$ r$ )
rS rT rU recompiler_ MIGRATE_DIRr rW r! r< rB
__classcell__r r r\ r rX s
rX c s& e Zd Zd fdd Zdd Z ZS ) ModuleRouterr c s0 t t| j|f| t|tr&t|}|| _d S r
)rY rw r r r r migrate_module)r r rx r[ r\ r r r s
zModuleRouter.__init__c C s$ t | j|}t |dtt |dtfS )Nr> r? )r' rx rq )r r= modr r r rB s zModuleRouter.read)r )rS rT rU r rB rv r r r\ r rw s rw )(rb rs importlibr typesr rI Zpeeweer r Zpeewee_migrater r Zpeewee_migrate.autor r Zpeewee_migrate.compatr r
Zpeewee_migrate.migratorr rt Mr9 r@ r7 getcwdru rq rl abspathdirname__file__trB rn objectr rX rw r r r r <module> s& $ %