Edit File: 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