관리-도구
편집 파일: provision.cpython-38.pyc
U -?�fp � @ 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_opts�oraclec 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 �T/opt/hc_python/lib64/python3.8/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_ignore5 s r( c C s t |� t || j� d S �N)�_purge_recyclebinr! �r r r r r �'_ora_drop_all_schema_objects_pre_tables? 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_tablesE 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_dbV s r0 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_conn�scr r r �%_ora_stop_test_class_outside_fixturesc s r7 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 r8 r �ownerZ object_name�type_r r r r* y s �r* c C s8 ddl m} |�|d�dd� �}|�|d�dd� �}d S ) Nr1 )�event�checkoutc S s t �| � d S r) )r2 �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 rA � s z._oracle_post_configure_engine.<locals>.checkin)Z sqlalchemyr> Z listens_for)r �engineZfollower_identr> r? rA r r r � _oracle_post_configure_engine� s rD 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, z�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 | ]\}|� � �qS r )�lower)�.0�usernamer r r � <setcomp>� s z#_reap_oracle_dbs.<locals>.<setcomp>Z_ts1Z_ts2r r r1 � 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�totalrG r r r �_reap_oracle_dbs� s6 � �rP c C s t �| �} | j|dd�S )NZxe)rG �password)�sa_url�make_urlrK )r r r r r � _oracle_follower_url_from_main� s rT c C s dgdd�S )NzGLOBAL TEMPORARYz PRESERVE ROWS)�prefixesZoracle_on_commitr r+ r r r � _oracle_temp_table_keyword_args� s �rV c C s" |� � }|�d| � |�� d S )Nz#ALTER SESSION SET CURRENT_SCHEMA=%s)�cursor�execute�close)r rB Z schema_namerW r r r �(_oracle_set_default_schema_on_connection� s rZ 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_moderR rS Zget_driver_name)Zdb_urlZdb_opts�optionsr r r �_update_db_opts� s ��r\ )N)&� r r r rC r rR Ztesting.provisionr r r r r r r r r r r r r Zfor_dbr r# r( r, r/ r0 r7 r* rK r2 rD rP rT rV rZ r\ r r r r �<module> sV