���] � @ 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)�LOGGER�MigrateHistory)� diff_many�NEWLINE)� string_types�exec_in)�Migratorz\s+$�
migrationsc C s d S)N� )�m�dr
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_table�logger�
isinstance�pwZDatabaseZProxy� 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.todo�autoFc s� d } } � r�t � 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 �list�filter�dir�diff�run_oner% Zorm�valuesr �warnr �join�split�CLEAN_RE�sub�info�_create) r �namer �migrate�rollbackZ migrationZmodels1Zmodels2�pathr
)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._execute�fakeNr7 zDone %sz Running "%s"zRolling back %szRolled back %szMigration failed: %s)r< �mockZpatchr r r; r r5 ZcleanZ transaction�run�deleteZwherer7 Zexecute� Exceptionr9 � exception) r r7 r% rA Z downgrade�forcer8 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� ���)�filemask�matchr1 )r# �f)r r
r r$ � s z Router.todo.<locals>.<genexpr>) �osr: �existsrT r r0 �makedirs�sorted�listdir)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.py�wr8 r9 N)
�lenr �formatr\ r: r1 rT �open�write�MIGRATE_TEMPLATE)r r7 r8 r9 �num�prefixr: 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
�get�VOID)r r7 r[ �codeZscoper
r
r r< � s
%
z Router.read) rM rN rO �re�compilerY � 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 �abspath�dirname�__file__�tr< rf �objectr rR ro r
r
r
r �<module> s$ 6�%| Name | Type | Size | Permission | Actions |
|---|---|---|---|---|
| __init__.cpython-35.opt-1.pyc | File | 1 KB | 0644 |
|
| __init__.cpython-35.pyc | File | 1 KB | 0644 |
|
| __main__.cpython-35.opt-1.pyc | File | 154 B | 0644 |
|
| __main__.cpython-35.pyc | File | 154 B | 0644 |
|
| auto.cpython-35.opt-1.pyc | File | 7.5 KB | 0644 |
|
| auto.cpython-35.pyc | File | 7.5 KB | 0644 |
|
| cli.cpython-35.opt-1.pyc | File | 2.71 KB | 0644 |
|
| cli.cpython-35.pyc | File | 2.71 KB | 0644 |
|
| compat.cpython-35.opt-1.pyc | File | 6.18 KB | 0644 |
|
| compat.cpython-35.pyc | File | 6.18 KB | 0644 |
|
| migrator.cpython-35.opt-1.pyc | File | 11.65 KB | 0644 |
|
| migrator.cpython-35.pyc | File | 11.65 KB | 0644 |
|
| router.cpython-35.opt-1.pyc | File | 8.41 KB | 0644 |
|
| router.cpython-35.pyc | File | 8.41 KB | 0644 |
|