Current File : //opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyo
�
��4]c @ s- d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l
m Z d d
l
m Z e j e � 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 � � 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* e j" d � d$ � � Z+ e j" d � d% � � Z, e j" d& � d' � � Z- e j" d& � d( � � Z. d) � Z/ e j" d& � d* � � Z0 e j" d& � d+ � � Z1 d, � Z2 d- � Z3 e! j" d& � d. � � Z4 e j" d � d/ � � Z5 e j" d � d0 � � Z6 d1 � Z7 d2 � Z8 d S(3 i����Ni ( t config( t enginesi ( t
create_engine( t exc( t text( t url( t compatt 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( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt __init__ s c C s t � j d � | � S( Nt *( R t for_db( t clst fn( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt init s c s � � f d � } | S( Nc s | � j � <� S( N( R ( R ( t dbnameR ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt decorate s
( ( R R R ( ( R R sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR s c G s� t | t j � r$ t j | � } n' t | t j � r? | } n | j j } | j � } | | j k rz | j | | | � S| j d | | � Sd S( NR (
t
isinstanceR t string_typest sa_urlt make_urlt URLt dbR t get_backend_nameR ( R t cfgt argR t backend( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt __call__# s ( t __name__t
__module__R
t classmethodR R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR s c C sD x= t � D]2 } t j d | | j j � t | | j | � q
Wd S( Ns CREATE database %s, URI %r( t _configs_for_db_operationt logt infoR R t
_create_db( t follower_identR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt create_follower_db1 s c C s+ x$ t j j � D] } t | | � q Wd S( N( R t Configt all_configst _configure_follower( R$ R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt configure_follower7 s c C s� | r t | | � } n i } t | | � t j | | � } t | | | � | j � j � t j j | | | | � } | r� t
| | � n | S( N( t _follower_url_from_maint _update_db_optsR t testing_enginet _post_configure_enginet connectt closeR R& R R( ( t db_urlt optionst file_configR$ t db_optst engR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt setup_config< s
c C sD x= t � D]2 } t j d | | j j � t | | j | � q
Wd S( Ns DROP database %s, URI %r( R R! R" R R t _drop_db( R$ R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt drop_follower_dbK s c c s� t � } x$ t j j � D] } | j j � q Wxk t j j � D]Z } | j j } | j � } | | 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( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR Q s c C s t d | j � � d S( Ns" no DB creation routine for cfg: %s( t NotImplementedErrorR ( R R4 t ident( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR# d s c C s t d | j � � d S( Ns no DB drop routine for cfg: %s( R@ R ( R R4 RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR6 i s c C s d S( N( ( R0 R3 ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR+ n s c C s d S( N( ( R RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR( s s c C s d S( N( ( R t engineR$ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR- x s c C s t j | � } | | _ | S( N( R R R< ( R RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR* } s t mssqlc C s t | d <d S( Nt legacy_schema_aliasing( t False( R0 R3 ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _mssql_update_db_opts� 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 R< ( R RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _sqlite_follower_url_from_main� s c s5 d d l m } | j | d � � f d � � } d S( Ni����( t eventR. c s+ � s | j d � n | j d � � d S( Ns/ ATTACH DATABASE "test_schema.db" AS test_schemas2 ATTACH DATABASE "%s_test_schema.db" AS test_schema( t execute( t dbapi_connectiont connection_record( R$ ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR. � s
( t
sqlalchemyRI t listens_for( R RB R$ RI R. ( ( R$ sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _sqlite_post_configure_engine� s t
postgresqlc C s | j j } | j � j d d � �� } y t | | | � Wn t k
rN n X| sg | j d � } n d } x� t ry | j d | | f � Wnv t j
k
r } | d 7} | d k r� � n d t | � k r t j
d | | j | � t j d
� q qp � qp XPqp WWd QXd S( Nt isolation_levelt
AUTOCOMMITs select current_database()i s CREATE DATABASE %s TEMPLATE %si i s accessed by other userssF Waiting to create %s, URI %r, template DB %s is in use sleeping for .5g �?( R1 t postgresql_templatedbR. t execution_optionst _pg_drop_dbt Exceptiont scalart TrueRJ R t OperationalErrort strR! R" R t timet sleep( R R4 RA t template_dbt connt attemptt err( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt
_pg_create_db� s6
t mysqlc C st | j � �b } y t | | | � Wn t k
r6 n X| j d | � | j d | � | j d | � Wd QXd S( Ns( CREATE DATABASE %s CHARACTER SET utf8mb4s4 CREATE DATABASE %s_test_schema CHARACTER SET utf8mb4s6 CREATE DATABASE %s_test_schema_2 CHARACTER SET utf8mb4( R. t _mysql_drop_dbRV RJ ( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/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 RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _mysql_configure_follower� s
c C s d S( N( ( R R4 RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _sqlite_create_db� s c C sO | j � j d d � �1 } | j t d � d | �| j d | � Wd QXd S( NRQ RR 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. RT RJ R ( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyRU � s c C s/ | r t j d | � n t j d | � d S( Ns %s_test_schema.dbs %s.db( t ost remove( R R4 RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _sqlite_drop_db� s c C sL | j � �: } | j d | � | j d | � | j d | � Wd QXd S( Ns DROP DATABASE %s_test_schemas DROP DATABASE %s_test_schema_2s DROP DATABASE %s( R. RJ ( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyRc � s t oraclec C s� | j � �� } | j d | � | j d | � | j d | � | j d | f � | j d | � | j d | � | j d | � Wd QXd S( Ns create user %s identified by xes# create user %s_ts1 identified by xes# create user %s_ts2 identified by xes grant dba to %ss grant unlimited tablespace to %ss$ grant unlimited tablespace to %s_ts1s$ grant unlimited tablespace to %s_ts2( R. RJ ( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _oracle_create_db s c C s d | | _ d | | _ d S( Ns %s_ts1s %s_ts2( Re Rf ( R RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _oracle_configure_follower s
c C sW y) | j d | � t j d | � t SWn' t j k
rR } t j d | � t SXd S( Ns drop user %s cascades
Reaped db: %ss couldn't drop db: %s( RJ R! R" RX R t
DatabaseErrort warningRE ( R^ R R` ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _ora_drop_ignore s c C sH | j � �6 } t | | � t | d | � t | d | � Wd QXd S( Ns %s_ts1s %s_ts2( R. Rr ( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _oracle_drop_db# s
c C s d S( N( ( R0 R3 ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _oracle_update_db_opts0 s c C s- t j d � t j t � } t j t � } t | � � } xu | D]m } | j � } | j d � \ } } t j | � } | j
� | j f } | | j | � | | j | � qA WWd QXxn | D]f } | d } t
| | � d }
| | } | d k r t |
| � q� | d k r� t |
| � q� q� Wd S( Ns Reaping databases...t i Rm RC ( R! R" t collectionst defaultdictR8 t opent stript splitR R R R; R= t listt _reap_oracle_dbst _reap_mssql_dbs( t idents_filet urlst identst file_t linet db_nameR0 t url_objt url_keyR R RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt reap_dbs5 s&
c C sh t j d | � t | � } | j � �:} t j d d j | � � | j d � } d � | D� } t � } x� | D]� } | j d � ss | j d � r� qs qs | | k rs | j | � d | | k r� | j d | � n d | | k r| j d | � qqs qs Wd
} } x9 t | d � D]( \ } }
t
| |
� r | d 7} q q Wt j d | | � Wd QXd S(
Ns db reaper connecting to %rs identifiers in file: %ss , s� select u.username from all_users u where username like 'TEST_%' and not exists (select username from v$session where username=u.username)c S s h | ] \ } | j � � q S( ( t lower( t .0R: ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pys <setcomp>Z s t _ts1t _ts2s %s_ts1s %s_ts2i i s- Dropped %d out of %d stale databases detected( R! R" R R. t joinRJ R8 t endswithR= t enumerateRr ( R R� R4 R^ t to_reapt all_namest to_dropt namet droppedt totalR: ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR| N s.
c C s% t j | � } | | _ d | _ | S( Nt xe( R R R: t password( R RA ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _oracle_follower_url_from_maino s c C s� | j � j d d � �e } | j d | � | j d | � | j d | � | j d | � | j d � | j d � Wd QXd S( NRQ RR s create database %ss1 ALTER DATABASE %s SET ALLOW_SNAPSHOT_ISOLATION ONs0 ALTER DATABASE %s SET READ_COMMITTED_SNAPSHOT ONs use %ss create schema test_schemas create schema test_schema_2( R. RT RJ ( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _mssql_create_dbw s
c C s2 | j � j d d � � } t | | � Wd QXd S( NRQ RR ( R. RT t _mssql_drop_ignore( R R4 RA R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt _mssql_drop_db� s c C sW y) | j d | � t j d | � t SWn' t j k
rR } t j d | � t SXd S( Ns drop database %ss
Reaped db: %ss couldn't drop db: %s( RJ R! R" RX R Rp Rq RE ( R^ RA R` ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR� � s c C s t j d | � t | � } | j � j d d � �� } t j d d j | � � | j d � } d � | D� } t � } x* | D]" } | | k r | j | � q q Wd } } x9 t | d � D]( \ } }
t
| |
� r� | d 7} q� q� Wt j d
| | � Wd QXd S( Ns db reaper connecting to %rRQ RR s identifiers in file: %ss , s� select d.name from sys.databases as d where name like 'TEST_%' and not exists (select session_id from sys.dm_exec_sessions where database_id=d.database_id)c S s h | ] \ } | j � � q S( ( R� ( R� R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pys <setcomp>� s i i s- Dropped %d out of %d stale databases detected( R! R" R R. RT R� RJ R8 R= R� R� ( R R� R4 R^ R� R� R� R� R� R� R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyR} � s"
(9 Rv t loggingRj R[ t R R R R R RB R R t utilR t getLoggerR R! t Nonet FOLLOWER_IDENTt objectR R% R) R5 R7 R R R# R6 R+ R( R- R* R RF RH RO Ra Rd Rg Rh RU Rl Rc Rn Ro Rr Rs Rt R� R| R� R� R� R� R} ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/provision.pyt <module> s\ %
!
Mini Shell