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
�J/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/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 � string_types�sa_url�make_url�URL�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_opts�optionsZ file_configr' �engr r
r
r � setup_config4 s
r2 c C s" x t � D ]}t||j| � qW d S )N)r% �_drop_dbr )r' r r
r
r �drop_follower_db? s r4 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 �username�host�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 r1 �identr
r
r r& Y s r& c C s t d|j ��d S )Nz no DB drop routine for cfg: %s)r: r )r r1 r; r
r
r r3 ^ s r3 c C s d S )Nr
)r r; r
r
r r+ c s r+ c C s t �| �} || _| S )N)r r r8 )r r; 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 r8 )r r; r
r
r � _sqlite_follower_url_from_maino s
r< 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 r1 r; �connZ currentdbr
r
r �
_pg_create_dbx s
rC 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_dbrA )r r1 r; rB r
r
r �_mysql_create_db� s
rE c C s d| | _ d| | _d S )Nz%s_test_schemaz%s_test_schema_2)Z test_schemaZ
test_schema_2)r r; r
r
r �_mysql_configure_follower� s
rF c C s d S )Nr
)r r1 r; r
r
r �_sqlite_create_db� s rG c C s@ |� � jdd��&}|jtd�|d� |�d| � W d Q R X d S )Nr= )r> 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? rA r )r r1 r; rB r
r
r r@ � s
r@ c C s d S )Nr
)r r1 r; r
r
r �_sqlite_drop_db� s rH 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. rA )r r1 r; rB r
r
r rD � s
rD )"�__doc__Zsqlalchemy.enginer r Z
sqlalchemyr �utilr � r r r ZFOLLOWER_IDENT�objectr
r( r, r2 r4 r% r r& r3 r+ r- r r< rC rE rF rG r@ rH rD r
r
r
r �<module> s0