Your IP : 216.73.216.213
���] � @ 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
�F/opt/alt/python35/lib/python3.5/site-packages/peewee_migrate/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 �_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 r8 r9 r: r
r
r r7 T s zBaseRouter._createc C s
t � d S)N)r )r r8 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
)r8 )r$ �mmr
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$ r8 )rA r
r r? c s z#BaseRouter.diff.<locals>.<listcomp>)�setrA r )r r
)rA 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 rA r/ )r r& r8 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�fakeNr8 zDone %sz Running "%s"zRolling back %szRolled back %szMigration failed: %s)r= �mock�patchr r r< r r6 �cleanZ transaction�run�delete�wherer8 �execute� Exceptionr: � exception) r r8 r& rC Z downgrade�forcer9 r: �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 migraterC rM )r r6 r. r& r/ �append)r r8 rC rA r. r& Zmnamer
r
r rG � 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���)�striprA r r& r/ r r1 )r r8 rA r& r
r
r r: � s zBaseRouter.rollback)�__name__�
__module__� __qualname__�__doc__r r r r �propertyr r< r7 r= rA r. r& r/ rG r: 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)�superrX r � migrate_dir)r r rZ �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�matchr2 )r$ �f)r r
r r% � s z Router.todo.<locals>.<genexpr>) �osr; �existsrZ r r1 �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}_rP z.py�wr9 r: N)
�lenr �formatrb r; r2 rZ �open�write�MIGRATE_TEMPLATE)r r8 r9 r: �num�prefixr; ra r
r
r r7 � 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.pyr9 r: N) rj rb r; r2 rZ r= r
�get�VOID)r r8 ra �codeZscoper
r
r r= � s
%
z Router.read) rS rT rU �re�compiler_ � MIGRATE_DIRr rW r r7 r= r
r
)r\ r rX � s
rX 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)rY ru r r r r �migrate_module)r r rv r[ )r\ r
r r � s zModuleRouter.__init__c C s4 t | j | � } t | d t � t | d t � f S)Nr9 r: )r# rv rp )r r8 �modr
r
r r= � s zModuleRouter.read)rS rT rU r r= r
r
)r\ r ru � s ru )(rb rr � importlibr �typesr rD Zpeeweer r �peewee_migrater r Zpeewee_migrate.autor r Zpeewee_migrate.compatr r
Zpeewee_migrate.migratorr rs �Mr4 r; r2 �getcwdrt rp rj �abspath�dirname�__file__�tr= rl �objectr rX ru r
r
r
r �<module> s$ 6�%