ó ¿b›]c@s-ddlZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z dd l m Z dd l mZejeƒZdZd efd „ƒYZd „Zd„Zd„Zd„Zd„Zejd„ƒZejd„ƒZejd„ƒZejd„ƒZejd„ƒZ ejd„ƒZ!ej"dƒd„ƒZ#e!j"dƒd„ƒZ$e j"dƒd„ƒZ%ej"dƒd„ƒZ&ej"dƒd „ƒZ'ej"dƒd!„ƒZ(ej"dƒd"„ƒZ)ej"dƒd#„ƒZ*ej"dƒd$„ƒZ+ej"dƒd%„ƒZ,ej"d&ƒd'„ƒZ-ej"d&ƒd(„ƒZ.d)„Z/ej"d&ƒd*„ƒZ0ej"d&ƒd+„ƒZ1d,„Z2d-„Z3e!j"d&ƒd.„ƒZ4ej"dƒd/„ƒZ5ej"dƒd0„ƒZ6d1„Z7d2„Z8dS(3iÿÿÿÿNi(tconfig(tenginesi(t create_engine(texc(ttext(turl(tcompattregistercBs2eZd„Zed„ƒZd„Zd„ZRS(cCs i|_dS(N(tfns(tself((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyt__init__scCstƒjdƒ|ƒS(Nt*(Rtfor_db(tclstfn((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pytinitscs‡‡fd†}|S(Ncs|ˆjˆ<ˆS(N(R(R(tdbnameR (s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pytdecorates ((R RR((RR s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR scGs’t|tjƒr$tj|ƒ}n't|tjƒr?|}n |jj}|jƒ}||j krz|j |||ŒS|j d||ŒSdS(NR ( t isinstanceRt string_typestsa_urltmake_urltURLtdbRtget_backend_nameR(R tcfgtargRtbackend((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyt__call__#s   (t__name__t __module__R t classmethodRR R(((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyRs  cCsDx=tƒD]2}tjd||jjƒt||j|ƒq WdS(NsCREATE database %s, URI %r(t_configs_for_db_operationtlogtinfoRRt _create_db(tfollower_identR((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pytcreate_follower_db1scCs+x$tjjƒD]}t||ƒqWdS(N(RtConfigt all_configst_configure_follower(R$R((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pytconfigure_follower7scCs’|rt||ƒ}ni}t||ƒtj||ƒ}t|||ƒ|jƒjƒtjj ||||ƒ}|rŽt ||ƒn|S(N( t_follower_url_from_maint_update_db_optsRttesting_enginet_post_configure_enginetconnecttcloseRR&RR((tdb_urltoptionst file_configR$tdb_optstengR((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyt setup_config<s cCsDx=tƒD]2}tjd||jjƒt||j|ƒq WdS(NsDROP database %s, URI %r(R R!R"RRt_drop_db(R$R((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pytdrop_follower_dbKsccsÉtƒ}x$tjjƒD]}|jjƒqWxktjjƒD]Z}|jj}|jƒ}||j|j |j f}||kr@|V|j |ƒq@q@Wx$tjjƒD]}|jjƒq®WdS(N( tsetRR&R'RtdisposeRRtusernamethosttdatabasetadd(thostsRRRt host_conf((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR Qs    cCstd|jƒ‚dS(Ns"no DB creation routine for cfg: %s(tNotImplementedErrorR(RR4tident((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR#dscCstd|jƒ‚dS(Nsno DB drop routine for cfg: %s(R@R(RR4RA((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR6iscCsdS(N((R0R3((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR+nscCsdS(N((RRA((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR(sscCsdS(N((RtengineR$((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR-xscCstj|ƒ}||_|S(N(RRR<(RRA((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR*}s tmssqlcCst|dZs t_ts1t_ts2s%s_ts1s%s_ts2iis-Dropped %d out of %d stale databases detected( R!R"RR.tjoinRJR8tendswithR=t enumerateRr( RR€R4R^tto_reapt all_namestto_droptnametdroppedttotalR:((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR|Ns.       cCs%tj|ƒ}||_d|_|S(Ntxe(RRR:tpassword(RRA((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyt_oracle_follower_url_from_mainos  cCsƒ|jƒjddƒe}|jd|ƒ|jd|ƒ|jd|ƒ|jd|ƒ|jdƒ|jdƒWdQXdS( NRQRRscreate database %ss1ALTER DATABASE %s SET ALLOW_SNAPSHOT_ISOLATION ONs0ALTER DATABASE %s SET READ_COMMITTED_SNAPSHOT ONsuse %sscreate schema test_schemascreate schema test_schema_2(R.RTRJ(RR4RAR^((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyt_mssql_create_dbws   cCs2|jƒjddƒ}t||ƒWdQXdS(NRQRR(R.RTt_mssql_drop_ignore(RR4RAR^((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyt_mssql_drop_db†scCsWy)|jd|ƒtjd|ƒtSWn'tjk rR}tjd|ƒtSXdS(Nsdrop database %ss Reaped db: %sscouldn't drop db: %s(RJR!R"RXRRpRqRE(R^RAR`((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR˜Œs c Cstjd|ƒt|ƒ}|jƒjddƒÎ}tjddj|ƒƒ|jdƒ}d„|Dƒ}tƒ}x*|D]"}||kr|j|ƒqqWd}} x9t |d ƒD](\} } t || ƒr¿|d 7}q¿q¿Wtjd || ƒWdQXdS( Nsdb reaper connecting to %rRQRRsidentifiers 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)cSsh|]\}|jƒ’qS((R‡(RˆR((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pys «s iis-Dropped %d out of %d stale databases detected( R!R"RR.RTR‹RJR8R=RR˜( RR€R4R^RŽRRR‘R’R“R((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyR}žs"      (9RvtloggingRjR[tRRRRRRBRRtutilRt getLoggerRR!tNonetFOLLOWER_IDENTtobjectRR%R)R5R7R RR#R6R+R(R-R*R RFRHRORaRdRgRhRURlRcRnRoRrRsRtR†R|R–R—R™R˜R}(((s;lib/python2.7/site-packages/sqlalchemy/testing/provision.pyts\          %    !