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 e�d ej�Ze j�e �� d
�Zd d � Z e e j�e 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)�LOGGER�MigrateHistory)� diff_many�NEWLINE)� string_types�exec_in)�Migratorz\s+$�
migrationsc C s d S �N� )�m�dr r �
/router.py�<lambda> � r z template.txtc @ s� e Zd ZdZdefdd�Zedd� �Zedd� �Z d d d �Z
d!dd�Z dd� Z edd� �Z
edd� �Zedd� �Zd"dd�Zd#d d �Zd d � ZdS )$�
BaseRouterz Abstract base class for router.Zmigratehistoryc C s6 || _ || _|| _t| j tjtjf�s2td| ��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 tj_ | jtj_t�d� tS )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 r
��NotImplementedErrorr r r r �todo+ s zBaseRouter.todo�autoFc sP d }}� �r"t � t�rHz t� �� W n" tk
rF | j�d� � Y S X t � t�rvttdd� � fdd�t � �D ���� | j
D ]}| � || j � q|� }t| j j
�� �}t||| j d�}|s�| j�d�S tt�d �|��d
�� }t�d
|�}t||| j d d �}tt�d �|��d
�� }t�d
|�}| j�d
|� | �|||�}| j�d|� |S )�Create a migration.� z Can't import models module: %sc S s t | t�ot| tj�S 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 �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/ 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.| � |�\}}|r�t�d��, t�d�� ||| j|d� W 5 Q R X W 5 Q R X |rr| jj|d� | j�d|� |�� |W S | j�d|� | j� � �� |s�||| j|d� |�
� | jj|d� | j�d|� nP| j�d|� ||| j|d� |�
� | j� � � | jj
|k��� | j�d|� W 5 Q R X W n> tk
�rn } z | j�� | j�d |� � 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 transaction�run�delete�wherer= Zexecute� Exceptionr? Z exception) r r= r+ rH Z downgrade�forcer> 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� | j�d|� 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 Ze�d�Zef� fdd� Zedd� �Z d dd�Z
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�� j�s*� j�d� j� t �� j� t� fdd�t �� j�D ��S )z Scan migrations in file system.z'Migration directory: %s does not exist.c 3 s, | ]$}� j �|�rd �|dd� �V qdS )r$ N���)�filemask�matchr7 )r( �fr r r r* � s z Router.todo.<locals>.<genexpr>) �osr@ �existsrZ r r6 �makedirs�sorted�listdirr 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}_� �.py�w)r> r? )
�lenr! �formatrb r@ r7 rZ �open�write�MIGRATE_TEMPLATE)r r= r> r? Znum�prefixr@ ra r r r r< � s
zRouter._createc
C s^ t tj�| j|d ���<}|�� }i }t||� |�dt�|�dt�fW 5 Q R � S Q R X dS )zRead migration from file.rh r> r? N) rl rb r@ r7 rZ rB r
�get�VOID)r r= ra �codeZscoper r r rB � s
z Router.read)r$ r$ )
rS rT rU �re�compiler_ � 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|t�r&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 |dt�t |dt�fS )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 �abspath�dirname�__file__�trB rn �objectr rX rw r r r r �<module> s& $ %| Name | Type | Size | Permission | Actions |
|---|---|---|---|---|
| __init__.cpython-38.opt-1.pyc | File | 979 B | 0644 |
|
| __init__.cpython-38.pyc | File | 979 B | 0644 |
|
| __main__.cpython-38.opt-1.pyc | File | 157 B | 0644 |
|
| __main__.cpython-38.pyc | File | 157 B | 0644 |
|
| auto.cpython-38.opt-1.pyc | File | 6.89 KB | 0644 |
|
| auto.cpython-38.pyc | File | 6.89 KB | 0644 |
|
| cli.cpython-38.opt-1.pyc | File | 2.44 KB | 0644 |
|
| cli.cpython-38.pyc | File | 2.44 KB | 0644 |
|
| compat.cpython-38.opt-1.pyc | File | 5.88 KB | 0644 |
|
| compat.cpython-38.pyc | File | 5.88 KB | 0644 |
|
| migrator.cpython-38.opt-1.pyc | File | 10.64 KB | 0644 |
|
| migrator.cpython-38.pyc | File | 10.64 KB | 0644 |
|
| router.cpython-38.opt-1.pyc | File | 7.83 KB | 0644 |
|
| router.cpython-38.pyc | File | 7.83 KB | 0644 |
|