Edit File: provision.cpython-37.opt-1.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 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_opts�optionsZ file_configr% �engr r r r � setup_config4 s r0 c C s" x t � D ]}t||j| � qW d S )N)r# �_drop_dbr )r% r r r r �drop_follower_db? s r2 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 Zusername�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 r/ �identr r r r$ Y s r$ c C s t d|j ��d S )Nz no DB drop routine for cfg: %s)r7 r )r r/ r8 r r r r1 ^ s r1 c C s d S )Nr )r r8 r r r r) c s r) c C s t �| �} || _| S )N)r r r5 )r r8 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 r5 )r r8 r r r � _sqlite_follower_url_from_maino s r9 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/ r8 �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/ r8 r? r r r �_mysql_create_db� s rB c C s d| | _ d| | _d S )Nz%s_test_schemaz%s_test_schema_2)Z test_schemaZ test_schema_2)r r8 r r r �_mysql_configure_follower� s rC c C s d S )Nr )r r/ r8 r r r �_sqlite_create_db� s rD 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< r> r )r r/ r8 r? r r r r= � s r= c C s d S )Nr )r r/ r8 r r r �_sqlite_drop_db� s rE 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/ r8 r? r r r rA � s rA )"�__doc__Zsqlalchemy.enginer r Z sqlalchemyr �utilr � r r r ZFOLLOWER_IDENT�objectr r& r* r0 r2 r# r r$ r1 r) r+ r r9 r@ rB rC rD r= rE rA r r r r �<module> s0