�
�M!Vc@s�
dZd
dl
mZ
d
dlmZ
ddlmZ
ddlm	Z	m
Z

ddlmZ
dZ
d	ef
d
��YZd�Zd�Zd
�Zd�Zd�Zejd��
Zejd��
Zejd��
Zejd��
Zejd�
d��
Zejd�
d��
Zejd�
d��
Zejd�
d��
Zejd�
d��
Zejd�
d��
Z ejd�
d��
Z!ejd�
d��
Z"dS( s�NOTE:  copied/adapted from SQLAlchemy master for backwards compatibility;
   this should be removable when Alembic targets SQLAlchemy 1.0.0
i����(
turl(
ttexti(
tcompati
(tconfigtengines(
tget_url_backend_nametregistercBs2eZ
d�Zed
��
Zd�Zd�ZRS(c

Cs
i|_dS(
N(
tfns(
tself((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt__init__s
cCst�j
d
�
|
�
S(Nt
*(Rtfor_db(tclstfn((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pytinitscs��
fd
�}|S(Nc

s|�
j�<�
S(
N(
R(
R
(tdbnameR(sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pytdecorates


((RRR((RRsJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRs
cGs�t|
t
j�r$tj|
�
}n't|
tj�r?|
}n|
jj}t|�
}||j	krz|j	||
|�
S|j	d
|
|�
SdS(NR
(
t
isinstanceRtstring_typestsa_urltmake_urltURLtdbRRR(RtcfgtargRtbackend((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt__call__s



	


(t__name__t
__module__R	tclassmethodRRR(((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR
s
		c
Cs+x$t�D]}
t
|
|
j|�
q
WdS(
N(t_configs_for_db_operationt
_create_dbR(tfollower_identR((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pytcreate_follower_db)s
c
Cs+x$tj
j�D]}
t|
|�
qWdS(
N(RtConfigtall_configst_configure_follower(R R((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pytconfigure_follower/s

cCso|rt||�}nt
j||
�}|j�j�
tjj||
||�}|rkt||�
n|S(
N(	t_follower_url_from_mainRttesting_enginetconnecttcloseRR"RR$(tdb_urltdb_optstoptionstfile_configR tengR((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pytsetup_config4s







c
Cs+x$t�D]}
t
|
|
j|�
q
WdS(
N(Rt_drop_dbR(R R((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pytdrop_follower_db?s

ccs�t�}x$t
jj�D]}
|
jj�
qWxkt
jj�D]Z}
|
jj}t|�
}||j|j	|j
f}||kr@|
V
|j|�

q@q@Wx$t
jj�D]}
|
jj�
q�WdS(
N(tsetRR"R#RtdisposeRRtusernamethosttdatabasetadd(thostsRRRt	host_conf((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRDs
	






cCstd
|
j
�
�
dS(Ns"no DB creation routine for cfg: %s(tNotImplementedErrorR(RR.tident((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRYscCstd
|
j
�
�
dS(Nsno DB drop routine for cfg: %s(R:R(RR.R;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR0^sc
CsdS(
N((RR;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR$cscCstj
|�
}|
|_|S(
N(RRR6(RR;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyR&hs
	
tsqlitecCsAtj
|�
}|js(|jd
kr,|Stj
d|
�
SdS(Ns:memory:ssqlite:///%s.db(RRR6(RR;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt_sqlite_follower_url_from_mainos

t
postgresqlcCsi|
j�j
d
d�
�K}yt|||�
Wn


n
X|jd�
}|jd||f�

WdQXdS(Ntisolation_levelt
AUTOCOMMITsselect current_database()sCREATE DATABASE %s TEMPLATE %s(R(texecution_optionst_pg_drop_dbtscalartexecute(RR.R;tconnt	currentdb((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt
_pg_create_dbxs






tmysqlcCsj|
j��X}yt
|||�
Wn


n
X|jd
|�

|jd|�

|jd|�

WdQXdS(NsCREATE DATABASE %ssCREATE DATABASE %s_test_schemas CREATE DATABASE %s_test_schema_2(R(t_mysql_drop_dbRD(RR.R;RE((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt_mysql_create_db�s






cCsd
|
|_d|
|_
dS(Ns%s_test_schemas%s_test_schema_2(ttest_schemat
test_schema_2(RR;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt_mysql_configure_follower�s

c
CsdS(
N((RR.R;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt_sqlite_create_db�scCsO|
j�j
d
d�
�1}|jtd�
d|�


|jd|�

WdQXdS(NR?R@s�select pg_terminate_backend(pid) from pg_stat_activity where usename=current_user and pid != pg_backend_pid() and datname=:dnametdnamesDROP DATABASE %s(R(RARDR
(RR.R;RE((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRB�s



	
c
CsdS(
N((RR.R;((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt_sqlite_drop_db�scCsv|
j��d}y|j
d
|�

Wn


n
Xy|j
d|�

Wn


n
Xy|j
d|�

Wn


n
XWdQXdS(NsDROP DATABASE %s_test_schemasDROP DATABASE %s_test_schema_2sDROP DATABASE %s(R(RD(RR.R;RE((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyRI�s











N(#t__doc__tsqlalchemy.engineRRt
sqlalchemyR
tutilRtRRRtNonetFOLLOWER_IDENTtobjectRR!R%R/R1RRRR0R$R&RR=RGRJRMRNRBRPRI(((sJ/opt/alt/python27/lib/python2.7/site-packages/alembic/testing/provision.pyt<module>s0