Current File : //opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyo
�
�M!Vc @ s� d Z d d l m Z d d l m Z d d l m Z d d l m Z m
Z
d d l m Z d Z
d e f d
� � YZ d � Z d � Z d
� Z d � Z d � Z e j d � � Z e j d � � Z e j d � � Z e j d � � Z e j d � d � � Z e j d � d � � Z e j d � d � � Z e j d � d � � Z e j d � d � � Z e j d � d � � Z e j d � d � � Z! e j d � d � � Z" d S( s� NOTE: copied/adapted from SQLAlchemy master for backwards compatibility;
this should be removable when Alembic targets SQLAlchemy 1.0.0
i����( t url( t texti ( t compati ( t configt engines( t get_url_backend_namet registerc B s2 e Z d � Z e d � � Z d � Z d � Z RS( c C s
i | _ d S( N( t fns( t self( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt __init__ s c C s t � j d � | � S( Nt *( R t for_db( t clst fn( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt init s c s � � f d � } | S( Nc s | � j � <� S( N( R ( R
( t dbnameR ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt decorate s
( ( R R R ( ( R R sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR s c G s� t | t j � r$ t j | � } n' t | t j � r? | } n | j j } t | � } | | j k rz | j | | | � S| j d | | � Sd S( NR
(
t
isinstanceR t string_typest sa_urlt make_urlt URLt dbR R R ( R t cfgt argR t backend( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt __call__ s ( t __name__t
__module__R t classmethodR R R ( ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR
s c C s+ x$ t � D] } t | | j | � q
Wd S( N( t _configs_for_db_operationt
_create_dbR ( t follower_identR ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt create_follower_db) s c C s+ x$ t j j � D] } t | | � q Wd S( N( R t Configt all_configst _configure_follower( R R ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt configure_follower/ s c C so | r t | | � } n t j | | � } | j � j � t j j | | | | � } | rk t | | � n | S( N( t _follower_url_from_mainR t testing_enginet connectt closeR R" R R$ ( t db_urlt db_optst optionst file_configR t engR ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt setup_config4 s c C s+ x$ t � D] } t | | j | � q
Wd S( N( R t _drop_dbR ( R R ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt drop_follower_db? s c c s� t � } x$ t j j � D] } | j j � q Wxk t j j � D]Z } | j j } t | � } | | j | j | j
f } | | k r@ | V| j | � q@ q@ Wx$ t j j � D] } | j j � q� Wd S( N( t setR R" R# R t disposeR R t usernamet hostt databaset add( t hostsR R R t host_conf( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR D s c C s t d | j � � d S( Ns" no DB creation routine for cfg: %s( t NotImplementedErrorR ( R R. t ident( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR Y s c C s t d | j � � d S( Ns no DB drop routine for cfg: %s( R: R ( R R. R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR0 ^ s c C s d S( N( ( R R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR$ c s c C s t j | � } | | _ | S( N( R R R6 ( R R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR& h s t sqlitec C sA t j | � } | j s( | j d k r, | St j d | � Sd S( Ns :memory:s sqlite:///%s.db( R R R6 ( R R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt _sqlite_follower_url_from_maino s t
postgresqlc C si | j � j d d � �K } y t | | | � Wn n X| j d � } | j d | | f � Wd QXd S( Nt isolation_levelt
AUTOCOMMITs select current_database()s CREATE DATABASE %s TEMPLATE %s( R( t execution_optionst _pg_drop_dbt scalart execute( R R. R; t connt currentdb( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt
_pg_create_dbx s t mysqlc C sj | j � �X } y t | | | � Wn n X| j d | � | j d | � | j d | � Wd QXd S( Ns CREATE DATABASE %ss CREATE DATABASE %s_test_schemas CREATE DATABASE %s_test_schema_2( R( t _mysql_drop_dbRD ( R R. R; RE ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt _mysql_create_db� s c C s d | | _ d | | _ d S( Ns %s_test_schemas %s_test_schema_2( t test_schemat
test_schema_2( R R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt _mysql_configure_follower� s
c C s d S( N( ( R R. R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt _sqlite_create_db� s c C sO | j � j d d � �1 } | j t d � d | �| j d | � Wd QXd S( NR? R@ s� select pg_terminate_backend(pid) from pg_stat_activity where usename=current_user and pid != pg_backend_pid() and datname=:dnamet dnames DROP DATABASE %s( R( RA RD R ( R R. R; RE ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRB � s c C s d S( N( ( R R. R; ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt _sqlite_drop_db� s c C sv | j � �d } y | j d | � Wn n Xy | j d | � Wn n Xy | j d | � Wn n XWd QXd S( Ns DROP DATABASE %s_test_schemas DROP DATABASE %s_test_schema_2s DROP DATABASE %s( R( RD ( R R. R; RE ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRI � s N(# t __doc__t sqlalchemy.engineR R t
sqlalchemyR t utilR t R R R t Nonet FOLLOWER_IDENTt objectR R! R% R/ R1 R R R R0 R$ R& R R= RG RJ RM RN RB RP RI ( ( ( sJ /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt <module> s0
Mini Shell