Current File : //opt/alt/python37/lib/python3.7/site-packages/alembic/testing/__pycache__/provision.cpython-37.pyc
B
�M!V� � @ sL d Z ddlmZ ddlmZ ddlmZ ddlm Z m
Z
ddlm Z d Z G d
d � d e
�Zd d
� Zdd� Zdd� Zdd� Zdd� Zejdd� �Zejdd� �Zejdd� �Zejd d � �Ze�d �d d � �Ze�d!�d"d#� �Ze�d$�d%d&� �Z e�d$�d'd(� �Z e�d �d)d*� �Z e�d!�d+d,� �Z e�d �d-d.� �Z e�d$�d/d0� �Z!d S )1z�NOTE: copied/adapted from SQLAlchemy master for backwards compatibility;
this should be removable when Alembic targets SQLAlchemy 1.0.0
� )�url)�text� )�compat� )�config�engines)�get_url_backend_nameNc @ s0 e Zd Zdd� Zedd� �Zdd� Zdd� Zd S )
�registerc C s
i | _ d S )N)�fns)�self� r
�
/provision.py�__init__ s zregister.__init__c C s t � �d�|�S )N�*)r
�for_db)�cls�fnr
r
r �init s z
register.initc s � �fdd�}|S )Nc s | �j � < �S )N)r )r )�dbnamer r
r �decorate s
z!register.for_db.<locals>.decorater
)r r r r
)r r r r s zregister.for_dbc G sp t |tj�rt�|�}nt |tj�r*|}n|jj}t|�}|| j krX| j | |f|�� S | j d |f|�� S d S )Nr )
�
isinstancer Z string_types�sa_url�make_urlZURL�dbr r r )r �cfg�argr �backendr
r
r �__call__ s
zregister.__call__N)�__name__�
__module__� __qualname__r � classmethodr r r r
r
r
r r
s r
c C s" x t � D ]}t||j| � qW d S )N)�_configs_for_db_operation�
_create_dbr )�follower_identr r
r
r �create_follower_db) s r& c C s" x t j�� D ]}t|| � q W d S )N)r �Config� all_configs�_configure_follower)r% r r
r
r �configure_follower/ s r* c C sJ |rt | |�} t�| |�}|�� �� tj�||||�}|rFt||� |S )N) �_follower_url_from_mainr Ztesting_engine�connect�closer r' r
r) )Zdb_urlZdb_optsZoptionsZ file_configr% �engr r
r
r � setup_config4 s
r/ c C s" x t � D ]}t||j| � qW d S )N)r# �_drop_dbr )r% r r
r
r �drop_follower_db? s r1 c c s� t � } x tj�� D ]}|j�� qW xLtj�� D ]>}|jj}t|�}||j|j |j
f}|| kr0|V | � |� q0W x tj�� D ]}|j�� q~W d S )N) �setr r' r( r Zdisposer r ZusernameZhost�database�add)Zhostsr r r Z host_confr
r
r r# D s r# c C s t d|j ��d S )Nz"no DB creation routine for cfg: %s)�NotImplementedErrorr )r r. �identr
r
r r$ Y s r$ c C s t d|j ��d S )Nz no DB drop routine for cfg: %s)r5 r )r r. r6 r
r
r r0 ^ s r0 c C s d S )Nr
)r r6 r
r
r r) c s r) c C s t �| �} || _| S )N)r r r3 )r r6 r
r
r r+ h s
r+ Zsqlitec C s0 t �| �} | jr| jdkr | S t �d| �S d S )Nz:memory:zsqlite:///%s.db)r r r3 )r r6 r
r
r � _sqlite_follower_url_from_maino s
r7 Z
postgresqlc C sZ |� � jdd��@}yt| ||� W n Y nX |�d�}|�d||f � W d Q R X d S )N�
AUTOCOMMIT)�isolation_levelzselect current_database()z CREATE DATABASE %s TEMPLATE %s)r, �execution_options� _pg_drop_dbZscalar�execute)r r. r6 �connZ currentdbr
r
r �
_pg_create_dbx s
r> Zmysqlc C s` |� � �N}yt| ||� W n Y nX |�d| � |�d| � |�d| � W d Q R X d S )NzCREATE DATABASE %sz CREATE DATABASE %s_test_schemaz CREATE DATABASE %s_test_schema_2)r, �_mysql_drop_dbr< )r r. r6 r= r
r
r �_mysql_create_db� s
r@ c C s d| | _ d| | _d S )Nz%s_test_schemaz%s_test_schema_2)Z test_schemaZ
test_schema_2)r r6 r
r
r �_mysql_configure_follower� s
rA c C s d S )Nr
)r r. r6 r
r
r �_sqlite_create_db� s rB c C s@ |� � jdd��&}|jtd�|d� |�d| � W d Q R X d S )Nr8 )r9 z�select pg_terminate_backend(pid) from pg_stat_activity where usename=current_user and pid != pg_backend_pid() and datname=:dname)ZdnamezDROP DATABASE %s)r, r: r< r )r r. r6 r= r
r
r r; � s
r; c C s d S )Nr
)r r. r6 r
r
r �_sqlite_drop_db� s rC c C sx |� � �f}y|�d| � W n Y nX y|�d| � W n Y nX y|�d| � W n Y nX W d Q R X d S )Nz DROP DATABASE %s_test_schemaz DROP DATABASE %s_test_schema_2zDROP DATABASE %s)r, r< )r r. r6 r= r
r
r r? � s
r? )"�__doc__Zsqlalchemy.enginer r Z
sqlalchemyr �utilr � r r r ZFOLLOWER_IDENT�objectr
r& r* r/ r1 r# r r$ r0 r) r+ r r7 r> r@ rA rB r; rC r? r
r
r
r �<module> s0
Mini Shell