U
|
¸ý°dS ã @ sÀ d dl mZ d dl mZ d dl mZ d dlmZ d dlmZ d dlm Z d dlm
|
Z
|
d dlm Z d d lm Z d d
|
lm Z d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ e d¡dd Ze d¡dd Zdd Ze d¡dd Ze
|
d¡dd Ze d¡d d Ze d¡d d Z d0d"d#Z e Z e 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.d/ Z%d!S )1é )Ú create_engine)Úexc)Úinspect)Úurl)Úconfigure_follower)Ú create_db)Ú#drop_all_schema_objects_post_tables)Ú"drop_all_schema_objects_pre_tables)Údrop_db)Úfollower_url_from_main)Úlog)Úpost_configure_engine)Ú run_reap_dbs)Ú set_default_schema_on_connection)Ú stop_test_class_outside_fixtures)Útemp_table_keyword_args)Úupdate_db_optsZoraclec C s¦ | ¡ }| d| ¡ | d| ¡ | d| ¡ | d|f ¡ | d| ¡ | d| ¡ | d| ¡ | d| ¡ | d | ¡ | d
|
| ¡ W 5 Q R X d S ) Nz create user %s identified by xez#create user %s_ts1 identified by xez#create user %s_ts2 identified by xezgrant dba to %sz grant unlimited tablespace to %sz$grant unlimited tablespace to %s_ts1z$grant unlimited tablespace to %s_ts2zgrant create table to %sz grant create table to %s_ts1z grant create table to %s_ts2)ÚbeginÚexec_driver_sql©ÚcfgÚengÚidentÚconn© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/oracle/provision.pyÚ_oracle_create_db s
|
r c C s d| | _ d| | _d S ©Nú%s_ts1ú%s_ts2)Ú test_schemaZ test_schema_2)Úconfigr r r r Ú_oracle_configure_follower) s
|
r" c
|
C s\ z | d| ¡ t d|¡ W dS tjk
|
rV } zt d|¡ W Y ¢dS d }~X Y nX d S )Nzdrop user %s cascadez Reaped db: %sTzcouldn't drop db: %sF)r r Úinfor Ú DatabaseErrorÚwarning)r ZdbnameÚerrr r r Ú_ora_drop_ignore/ s r' c C s t | t || j d S ©N)Ú_purge_recyclebinr ©r r r r r Ú'_ora_drop_all_schema_objects_pre_tables9 s r+ c C s | ¡ }|j |d d d ¡D ]}| d|d ¡ q |j || jd d ¡D ] }| d| j d|d ¡ qJt| ¡ D ]}| d| ¡ qxW 5 Q R X d S )Nz drop synonym Z synonym_nameÚ.z drop table )r ÚdialectZ _get_synonymsr r r Zget_temp_table_names)r r r ZsynZ tmp_tabler r r Ú(_ora_drop_all_schema_objects_post_tables? s
|
ÿÿr. c C s> | ¡ ,}t|| t|d| t|d| W 5 Q R X d S r )r r' r r r r Ú_oracle_drop_dbQ s
|
|
r/ c
|
C s z t | W n0 tjk
|
r< } zt d|¡ W 5 d }~X Y nX tD ]6}z
|
|j}W n |jjj k
|
rj Y qBX d|_||_qBt
|
¡ d S )Nz#purge recyclebin command failed: %sé ) r) r r$ r r% Ú
|
_all_connsZ stmtcachesizer- ZdbapiZInterfaceErrorÚclear)r! ÚdbÚclsr& Zcx_oracle_connZscr r r Ú%_ora_stop_test_class_outside_fixtures^ s
|
r5 Nc C sr | ¡ `}|d kr | d¡ nF| dd|j |¡i¡ ¡ D ](\}}}| d| d| d| d¡ q:W 5 Q R X d S )Nzpurge recyclebinzWselect owner, object_name,type from dba_recyclebin where owner=:schema and type='TABLE'Úschemazpurge ú z."ú")r r r- Zdenormalize_nameÚall)r r6 r ÚownerZ object_nameÚtype_r r r r) u s
|
ýr) c C s8 ddl m} | |d¡dd }| |d¡dd }d S ) Nr0 )ÚeventÚcheckoutc S s t | ¡ d S r( )r1 Úadd)Z dbapi_conZ
|
con_recordZ con_proxyr r r r= s z/_oracle_post_configure_engine.<locals>.checkoutÚcheckinc S s d|j kr| ¡ d S )NZ cx_oracle_xid)r# Z
|
invalidate)Údbapi_connectionZconnection_recordr r r r? s
|
z._oracle_post_configure_engine.<locals>.checkin)Z
|
sqlalchemyr< Z listens_for)r ÚengineZfollower_identr< r= r? r r r Ú _oracle_post_configure_engine s
|
|
|
|
rB c C s t d| ¡ t| }| ¡ Ú}t dd |¡¡ | d¡}dd |D }t }|D ]b}| d¡sR| d¡rnqRqR||krR| |¡ d | |kr| d | ¡ d
|
| |krR| d
|
| ¡ qRd }} t |d D ]\} }
|
t
|
||
|
rÈ|d 7 }qÈt d || ¡ W 5 Q R X d S )Nzdb reaper connecting to %rzidentifiers in file: %sz, zselect 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 | ]\}| ¡ qS r )Úlower)Ú.0Úusernamer r r Ú <setcomp>¦ s z#_reap_oracle_dbs.<locals>.<setcomp>Z_ts1Z_ts2r r r0 é z-Dropped %d out of %d stale databases detected) r r# r r Újoinr ÚsetÚendswithr> Ú enumerater' ) r Zidentsr r Zto_reapZ all_namesZto_dropÚnameZdroppedÚtotalrE r r r Ú_reap_oracle_dbs s6
|
ÿ
|
|
|
ÿrN c C s t | ¡} | j|ddS )NZxe)rE Úpassword)Úsa_urlÚmake_urlrI )r r r r r Ú _oracle_follower_url_from_main» s
|
rR c C s dgddS )NzGLOBAL TEMPORARYz PRESERVE ROWS)ÚprefixesZoracle_on_commitr r* r r r Ú _oracle_temp_table_keyword_argsÁ s þrT c C s" | ¡ }| d| ¡ | ¡ d S )Nz#ALTER SESSION SET CURRENT_SCHEMA=%s)ÚcursorÚexecuteÚclose)r r@ Z schema_namerU r r r Ú(_oracle_set_default_schema_on_connectionÉ s rX c C s$ |j r t | ¡ ¡ dkr d|d< dS )zCSet database options (db_opts) for a test database that we created.ZoracledbTZ
|
thick_modeN)Zoracledb_thick_moderP rQ Zget_driver_name)Zdb_urlZdb_optsÚoptionsr r r Ú_update_db_optsÒ s
|
ÿþrZ )N)&Ú r r r rA r rP Ztesting.provisionr r r r r
|
r r r r r r r r Zfor_dbr r" r' r+ r. r/ r5 r) rI r1 rB rN rR rT rX rZ r r r r Ú<module> sV
|
|
|
|
|
|
|
|
|
|
!
|
|
|
|
