관리-도구
편집 파일: provision.cpython-37.pyc
B ��4]�3 � @ sl d dl Z d dlZd dlZd dlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd l m Z dd l mZ e�e�ZdZG d d � d e�Zd d� Zdd� Zdd� Zdd� Zdd� Zejdd� �Zejdd� �Z ejdd � �Z ejd d � �Z ejd d � �Z ejd!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/d0� �Z&e �!d.�d1d2� �Z'e�!d&�d3d4� �Z(e �!d+�d5d6� �Z)e �!d&�d7d8� �Z*e �!d.�d9d:� �Z+e�!d;�d<d=� �Z,e �!d;�d>d?� �Z-d@dA� Z.e �!d;�dBdC� �Z/e �!d;�dDdE� �Z0dFdG� Z1dHdI� Z2e �!d;�dJdK� �Z3e�!d#�dLdM� �Z4e �!d#�dNdO� �Z5dPdQ� Z6dRdS� Z7dS )T� N� )�config)�engines� )� create_engine)�exc)�text)�url)�compatc @ s0 e Zd Zdd� Zedd� �Zdd� Zdd� Zd S ) �registerc C s i | _ d S )N)�fns)�self� r �O/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/provision.py�__init__ s zregister.__init__c C s t � �d�|�S )N�*)r �for_db)�cls�fnr r r �init s z register.initc s � �fdd�}|S )Nc s | �j � < �S )N)r )r )�dbnamer r r �decorate s z!register.for_db.<locals>.decorater )r r r r )r r r r s zregister.for_dbc G sp t |tj�rt�|�}nt |tj�r*|}n|jj}|�� }|| j krX| j | |f|�� S | j d |f|�� S d S )Nr ) � isinstancer � string_types�sa_url�make_url�URL�dbr �get_backend_namer )r �cfg�argr �backendr r r �__call__# s zregister.__call__N)�__name__� __module__� __qualname__r � classmethodr r r" r r r r r s r c C s4 x.t � D ]$}t�d| |jj� t||j| � qW d S )NzCREATE database %s, URI %r)�_configs_for_db_operation�log�infor r � _create_db)�follower_identr r r r �create_follower_db1 s r, c C s" x t j�� D ]}t|| � q W d S )N)r �Config� all_configs�_configure_follower)r+ r r r r �configure_follower7 s r0 c C sd |rt | |�} i }t| |� t�| |�}t| ||� |�� �� tj� ||||�}|r`t ||� |S )N) �_follower_url_from_main�_update_db_optsr Ztesting_engine�_post_configure_engine�connect�closer r- r r/ )�db_url�optionsZ file_configr+ �db_opts�engr r r r � setup_config< s r: c C s4 x.t � D ]$}t�d| |jj� t||j| � qW d S )NzDROP database %s, URI %r)r' r( r) r r �_drop_db)r+ r r r r �drop_follower_dbK s r< c c s� t � } x tj�� D ]}|j�� qW xLtj�� D ]>}|jj}|�� }||j|j |j f}|| kr0|V | � |� q0W x tj�� D ]}|j�� q~W d S )N) �setr r- r. r Zdisposer r �username�host�database�add)Zhostsr r r! Z host_confr r r r' Q s r' c C s t d|j ��d S )Nz"no DB creation routine for cfg: %s)�NotImplementedErrorr )r r9 �identr r r r* d s r* c C s t d|j ��d S )Nz no DB drop routine for cfg: %s)rB r )r r9 rC r r r r; i s r; c C s d S )Nr )r6 r8 r r r r2 n s r2 c C s d S )Nr )r rC r r r r/ s s r/ c C s d S )Nr )r �enginer+ r r r r3 x s r3 c C s t �| �} || _| S )N)r r r@ )r rC r r r r1 } s r1 �mssqlc C s d|d< d S )NFZlegacy_schema_aliasingr )r6 r8 r r r �_mssql_update_db_opts� s rF Zsqlitec C s0 t �| �} | jr| jdkr | S t �d| �S d S )Nz:memory:zsqlite:///%s.db)r r r@ )r rC r r r � _sqlite_follower_url_from_main� s rG c s( ddl m} |�|d�� fdd��}d S )Nr )�eventr4 c s" � s| � d� n| � d� � d S )Nz/ATTACH DATABASE "test_schema.db" AS test_schemaz2ATTACH DATABASE "%s_test_schema.db" AS test_schema)�execute)Zdbapi_connectionZconnection_record)r+ r r r4 � s z._sqlite_post_configure_engine.<locals>.connect)� sqlalchemyrH Z listens_for)r rD r+ rH r4 r )r+ r � _sqlite_post_configure_engine� s rK � postgresqlc C s� | j j}|�� jdd���}yt| ||� W n tk r> Y nX |sN|�d�}d}x�y|�d||f � W nj tj k r� } z>|d7 }|dkr�� dt |�kr�t � d ||j |� t�d � W d d }~X Y qT � Y qTX P qTW W d Q R X d S ) N� AUTOCOMMIT)�isolation_levelzselect current_database()r z CREATE DATABASE %s TEMPLATE %sr � zaccessed by other userszFWaiting to create %s, URI %r, template DB %s is in use sleeping for .5g �?)r7 Zpostgresql_templatedbr4 �execution_options� _pg_drop_db� ExceptionZscalarrI r ZOperationalError�strr( r) r �time�sleep)r r9 rC Z template_db�connZattempt�errr r r � _pg_create_db� s6 rX �mysqlc C sh |� � �V}yt| ||� W n tk r. Y nX |�d| � |�d| � |�d| � W d Q R X d S )Nz(CREATE DATABASE %s CHARACTER SET utf8mb4z4CREATE DATABASE %s_test_schema CHARACTER SET utf8mb4z6CREATE DATABASE %s_test_schema_2 CHARACTER SET utf8mb4)r4 �_mysql_drop_dbrR rI )r r9 rC rV r r r �_mysql_create_db� s r[ c C s d| | _ d| | _d S )Nz%s_test_schemaz%s_test_schema_2)� test_schema� test_schema_2)r rC r r r �_mysql_configure_follower� s r^ c C s d S )Nr )r r9 rC r r r �_sqlite_create_db� s r_ c C s@ |� � jdd��&}|jtd�|d� |�d| � W d Q R X d S )NrM )rN z�select pg_terminate_backend(pid) from pg_stat_activity where usename=current_user and pid != pg_backend_pid() and datname=:dname)ZdnamezDROP DATABASE %s)r4 rP rI r )r r9 rC rV r r r rQ � s rQ c C s& |rt �d| � nt �d| � d S )Nz%s_test_schema.dbz%s.db)�os�remove)r r9 rC r r r �_sqlite_drop_db� s rb c C sB |� � �0}|�d| � |�d| � |�d| � W d Q R X d S )Nz DROP DATABASE %s_test_schemaz DROP DATABASE %s_test_schema_2zDROP DATABASE %s)r4 rI )r r9 rC rV r r r rZ � s rZ �oraclec C s| |� � �j}|�d| � |�d| � |�d| � |�d|f � |�d| � |�d| � |�d| � W d 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_ts2)r4 rI )r r9 rC rV r r r �_oracle_create_db s rd c C s d| | _ d| | _d S )Nz%s_ts1z%s_ts2)r\ r] )r rC r r r �_oracle_configure_follower s re c C sT y | � d| � t�d|� dS tjk rN } zt�d|� dS d }~X Y nX d S )Nzdrop user %s cascadez Reaped db: %sTzcouldn't drop db: %sF)rI r( r) r � DatabaseError�warning)rV r rW r r r �_ora_drop_ignore s rh c C s> |� � �,}t||� t|d| � t|d| � W d Q R X d S )Nz%s_ts1z%s_ts2)r4 rh )r r9 rC rV r r r �_oracle_drop_db# s ri c C s d S )Nr )r6 r8 r r r �_oracle_update_db_opts0 s rj c C s� t �d� t�t�}t�t�}t| ��^}xV|D ]N}|�� }|�d�\}}t� |�}|� � |j f}|| � |� || � |� q.W W d Q R X xR|D ]J}|d } t || �d } || } | dkr�t| | � q�| dkr�t| | � q�W d S )NzReaping databases...� r rc rE )r( r) � collections� defaultdictr= �open�strip�splitr r r r? rA �list�_reap_oracle_dbs�_reap_mssql_dbs) Z idents_fileZurls�identsZfile_�lineZdb_namer6 Zurl_objZurl_keyr! r rC r r r �reap_dbs5 s& rv c C s t �d| � t| �}|�� ��}t �dd�|�� |�d�}dd� |D �}t� }xj|D ]b}|�d�sT|�d�rpqTqT||krT|�|� d | |kr�|�d | � d | |krT|�d | � qTW d }} x(t |d �D ]\} } t || �r�|d 7 }q�W t �d || � W d 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)�.0r> r r r � <setcomp>Z s z#_reap_oracle_dbs.<locals>.<setcomp>Z_ts1Z_ts2z%s_ts1z%s_ts2r r z-Dropped %d out of %d stale databases detected) r( r) r r4 �joinrI r= �endswithrA � enumeraterh ) r rt r9 rV �to_reap� all_names�to_drop�name�dropped�totalr> r r r rr N s. rr c C s t �| �} || _d| _| S )NZxe)r r r> �password)r rC r r r � _oracle_follower_url_from_maino s r� c C sl |� � jdd��R}|�d| � |�d| � |�d| � |�d| � |�d� |�d� W d Q R X d S ) NrM )rN zcreate database %sz1ALTER DATABASE %s SET ALLOW_SNAPSHOT_ISOLATION ONz0ALTER DATABASE %s SET READ_COMMITTED_SNAPSHOT ONzuse %szcreate schema test_schemazcreate schema test_schema_2)r4 rP rI )r r9 rC rV r r r �_mssql_create_dbw s r� c C s* |� � jdd��}t||� W d Q R X d S )NrM )rN )r4 rP �_mssql_drop_ignore)r r9 rC rV r r r �_mssql_drop_db� s r� c C sT y | � d| � t�d|� dS tjk rN } zt�d|� dS d }~X Y nX d S )Nzdrop database %sz Reaped db: %sTzcouldn't drop db: %sF)rI r( r) r rf rg )rV rC rW r r r r� � s r� c C s� t �d| � t| �}|�� jdd���}t �dd�|�� |�d�}dd� |D �}t� }x |D ]}||kr\|�|� q\W d }} x(t |d �D ]\} } t || �r�|d 7 }q�W t �d || � W d Q R X d S ) Nzdb reaper connecting to %rrM )rN zidentifiers in file: %sz, z�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 | ]\}|� � �qS r )rw )rx r r r r ry � s z"_reap_mssql_dbs.<locals>.<setcomp>r r z-Dropped %d out of %d stale databases detected) r( r) r r4 rP rz rI r= rA r| r� ) r rt r9 rV r} r~ r r� r� r� r r r r rs � s" rs )8rl �loggingr` rT � r r r r r rD r r �utilr � getLoggerr# r( ZFOLLOWER_IDENT�objectr r, r0 r: r< r' r r* r; r2 r/ r3 r1 r rF rG rK rX r[ r^ r_ rQ rb rZ rd re rh ri rj rv rr r� r� r� r� rs r r r r �<module> s\ % !