File: //opt/alt/python38/lib/python3.8/site-packages/peewee_migrate/__pycache__/router.cpython-38.pyc
U
ՙ_ � @ 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 �F/opt/alt/python38/lib/python3.8/site-packages/peewee_migrate/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 �_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) �mmr r r �
<listcomp>] s z#BaseRouter.done.<locals>.<listcomp>)r �selectr 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> �rH r r rF c s z#BaseRouter.diff.<locals>.<listcomp>)�setrH r" r r rI r r4 _ s
zBaseRouter.diffc C s& t | j�}| jD ]}| �||� q|S )z2Create migrator and setup it with fake migrations.)r r rH r5 )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)�fakerD zDone %sz Running "%s"zRolling back %szRolled back %szMigration failed: %sN)rC �mock�patchr r rB r r< �cleanZ transaction�run�delete�wherer> �execute� Exceptionr@ � exception) r r> r, rK Z downgrade�forcer? r@ �excr r r r5 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)rK rU )r r< r4 r, r5 �append)r r> rK rH r4 r, Zmnamer r r rO � 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)�striprH r r, r5 r r7 )r r> rH r, r r r r@ � s zBaseRouter.rollback)r# F)r% r% )TFF)NF)�__name__�
__module__� __qualname__�__doc__r r r r �propertyr" rB r= rC rH r4 r, r5 rO 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
)�superr_ r � migrate_dir)r r ra �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�matchr8 )r) �fr r r r+ � s z Router.todo.<locals>.<genexpr>) �osrA �existsra r r7 �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" �formatri rA r8 ra �open�write�MIGRATE_TEMPLATE)r r> r? r@ �num�prefixrA rh 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.ro r? r@ N) rs ri rA r8 ra rC r
�get�VOID)r r> rh �codeZscoper r r rC � s
z Router.read)r% r% )
rZ r[ r\ �re�compilerf � MIGRATE_DIRr r^ r" r= rC �
__classcell__r r rc r r_ � s
r_ c s& e Zd Zd� fdd� Zdd� Z� ZS )� ModuleRouterr c s0 t t| �j|f|� t|t�r&t|�}|| _d S r
)r` r r r r r �migrate_module)r r r� rb rc r r r � s
zModuleRouter.__init__c C s$ t | j|�}t |dt�t |dt�fS )Nr? r@ )r( r� ry )r r> �modr r r rC � s zModuleRouter.read)r )rZ r[ r\ r rC r~ r r rc r r � s r )(ri r{ � importlibr �typesr rL Zpeeweer r �peewee_migrater r Zpeewee_migrate.autor r Zpeewee_migrate.compatr r
Zpeewee_migrate.migratorr r| �Mr: rA r8 �getcwdr} ry rs �abspath�dirname�__file__�trC ru �objectr r_ r r r r r �<module> s& $ %