관리-도구
편집 파일: engines.cpython-38.pyc
U -?�f�4 � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlZd dl Z ddl m Z ddl m Z dd l mZ d d l mZ d d l mZ d d l mZ d dlmZ ejr�d dlmZ d dlmZ d dlmZ G dd� d�Ze� Z e dd� �Z e dd� �Z e dd� �Z e dd� �Z d6d d �Z!G d d � d �Z"d7d d!�Z#ej$d8d#d$d%d&d'd(�d)d*��Z%ej$d9d#d$d,d&d-d(�d.d*��Z%d:d/d*�Z%d;d0d1�Z&G d2d3� d3�Z'G d4d5� d5�Z(dS )<� )� annotationsN)�Any)�Dict)�Optional� )�config)� decorator)� gc_collect� )�event)�pool)� await_only)�Literal)�Engine)�URL)� AsyncEnginec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�ConnectionKillerc C s" t �� | _t�t�| _t� | _d S �N)�weakref�WeakKeyDictionary� proxy_refs� collections� defaultdict�set�testing_engines�dbapi_connections��self� r �J/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/engines.py�__init__% s zConnectionKiller.__init__c C sD t �|d| j� t �|d| j� t �|d| j� t �|d| j� d S )N�checkoutZcheckin�closeZclose_detached)r �listen� _add_conn� _remove_conn)r r r r r �add_pool* s zConnectionKiller.add_poolc C s | j �|� d| j|< d S )NT)r �addr )r Z dbapi_conZ con_recordZ con_proxyr r r r$ 2 s zConnectionKiller._add_connc G s | j �|� d S r )r �discard)r Z dbapi_conn�argr r r r% 6 s z ConnectionKiller._remove_connc C s, | � |j� |dkst�| j| �|� d S )N)�class�global�functionZfixture)r& r �AssertionErrorr r' )r �engine�scoper r r � add_engine9 s zConnectionKiller.add_enginec C s@ z |� W n0 t k r: } zt�d| � W 5 d }~X Y nX d S )Nz5testing_reaper couldn't rollback/close connection: %s�� Exception�warnings�warn�r �fn�er r r �_safe? s �zConnectionKiller._safec C s. t | j�D ] }|d k r |jr | �|j� q d S r )�listr �is_validr8 �rollback�r �recr r r � rollback_allG s z ConnectionKiller.rollback_allc C s^ t | j�D ],}|d k r |jr | j�|j� | �|j� q | jD ]}| �|j� q>| j� � d S r ) r9 r r: r r( �dbapi_connectionr8 �_checkinr; �clear)r r= �conr r r � checkin_allL s z ConnectionKiller.checkin_allc C s | � � d S r )rC r r r r � close_all[ s zConnectionKiller.close_allc C s* t js d S ddlm} |�|jj|� d S �Nr )� provision)r �bootstrapped_as_sqlalchemy� rF �prepare_for_drop_tablesr. �url)r � connectionrF r r r rI ^ s z(ConnectionKiller.prepare_for_drop_tablesc C s� | j | }t|�D ]d}t| j�D ]4}|d k r |jr |jd k r |j|jkr | �|j� q t|d�rnt |� � � q|� � q|� � d S )N� sync_engine) r r9 r r: �_poolr r8 r@ �hasattrr �disposerA )r r/ Zengr= Z proxy_refr r r �_drop_testing_enginesg s � � z&ConnectionKiller._drop_testing_enginesc C s | � d� d S )Nr, )rP r r r r � after_testx s zConnectionKiller.after_testc C sT t js d S |jjrd S | �� ddlm} t j�� �}|� |j j |� W 5 Q R X d S rE ) r rG � __class__Z"__leave_connections_for_teardown__rC rH rF �db�connectrI r. rJ )r �testrF �connr r r �after_test_outside_fixtures{ s z,ConnectionKiller.after_test_outside_fixturesc C s | � � | �d� | �d� d S )Nr, r* )rC rP r r r r � stop_test_class_inside_fixtures� s z0ConnectionKiller.stop_test_class_inside_fixturesc C sB t jjr>t� t jjr>tt jj�}t jj�� ds>td| ��d S )NFz/%d connection recs not cleared after test suite)r �baseZ _strong_ref_connection_recordsr �lenrA r- )r �lnr r r � stop_test_class_outside_fixtures� s ��z1ConnectionKiller.stop_test_class_outside_fixturesc C s" | � � | jD ]}| �|� qd S r )rC r rP )r r/ r r r � final_cleanup� s z ConnectionKiller.final_cleanupc C s | j D ]}|jrdst�qd S �NF)r r: r- r< r r r �assert_all_closed� s z"ConnectionKiller.assert_all_closedN)�__name__� __module__� __qualname__r r& r$ r% r0 r8 r> rC rD rI rP rQ rW rX r\ r] r_ r r r r r $ s" r c O s z| ||� W 5 t �� X d S r )�testing_reaperr_ �r6 �args�kwr r r �assert_conns_closed� s rg c O s z| ||� W 5 t �� X dS )zBDecorator that rolls back all open connections after fn execution.N)rc r> rd r r r �rollback_open_connections� s rh c O s t �� | ||� dS )z:Decorator that closes all connections before fn execution.N�rc rC rd r r r � close_first� s rj c O s z| ||� W 5 t �� X dS )z9Decorator that closes all connections after fn execution.Nri rd r r r �close_open_connections� s rk c c sX dd l m} |jD ]@}| r$|| kr$qt||d �}|sHttd| �j|�}|�� V qd S )Nr zsqlalchemy.dialects.%s)Zsqlalchemy.dialectsZdialects�__all__�getattr� __import__�dialect)�exclude�d�name�modr r r � all_dialects� s �rt c @ s> e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd d � ZdS )�ReconnectFixturec C s || _ g | _d| _d S r^ )�dbapi� connections� is_stopped)r rv r r r r � s zReconnectFixture.__init__c C s t | j|�S r )rm rv �r �keyr r r � __getattr__� s z ReconnectFixture.__getattr__c O sT | j j||�}| jr@| �|j� |�� }|�d� dsPtd��n| j� |� |S d S )Nzselect 1Fz%simulated connect failure didn't work) rv rT rx r8 r" �cursor�executer- rw �append)r re �kwargsrV Zcursr r r rT � s zReconnectFixture.connectc C s@ z |� W n0 t k r: } zt�d| � W 5 d }~X Y nX d S )Nz.ReconnectFixture couldn't close connection: %sr1 r5 r r r r8 � s zReconnectFixture._safeFc C s, || _ t| j�D ]}| �|j� qg | _d S r )rx r9 rw r8 r" )r �stop�cr r r �shutdown� s zReconnectFixture.shutdownc C s d| _ d S r^ )rx r r r r �restart s zReconnectFixture.restartN)F) r` ra rb r r{ rT r8 r� r� r r r r ru � s ru c sj | p t jj} t jjj}|s i }t|�|d< t| |���j� � �fdd�}�jjj�_ �jjj �_ |�_�S )N�modulec s �j j�� d�j j_� � d S r^ )ro rv r� rx r �Z_disposer. r r rO s z$reconnecting_engine.<locals>.dispose) r rS rJ ro rv ru �testing_enginerO r� Z test_shutdownr� Z test_restart)rJ �optionsrv rO r r� r �reconnecting_engine s r� Fz Optional[URL]zOptional[Dict[str, Any]]zLiteral[False]�boolr )rJ r� �asyncio�transfer_staticpool�returnc C s d S r r �rJ r� r� r� r r r r� s r� Tz Literal[True]r c C s d S r r r� r r r r� ' s c C s� |r|r t �ddlm} n ddlm} ddlm} |sDd}d} d} n$|�dd�}|�d d�} |�d d�} | prtj j } || �} |d kr�tj d ks�| j tj j j kr�tj }q�i }n.tj d k r�| j tj j j kr�tj � � } | �|� || f|�} | �r | jd k�r t�| d �d d� �} t�| d�dd� �}|�rdddlm} tj d k �rtttj j|��rtd}| j�tj j� n|�rttj j| _| dk�r�|�r�d| j_nd| _t| jtj��r�d|k�r�d|k�r�d|k�r�d| j_d| j_|�r�t � | | � | S )Nr )�create_async_engine)� create_engine)�make_urlTr, F� use_reaperr/ �sqlite_savepointZsqliterT c S s d | _ d S r )Zisolation_level)r? Zconnection_recordr r r � do_connectZ s z"testing_engine.<locals>.do_connect�beginc S s | � d� d S )NZBEGIN)Zexec_driver_sql)rV r r r �do_begin^ s z testing_engine.<locals>.do_begin)� StaticPoolr+ r � pool_timeoutZ max_overflow) r- Zsqlalchemy.ext.asyncior� � sqlalchemyr� Zsqlalchemy.engine.urlr� �popr rS rJ Z drivernameZdb_opts�copy�updaterr r Z listens_forZsqlalchemy.poolr� � isinstancer Z_transfer_fromrL Z _has_eventsZ QueuePool�_timeoutZ _max_overflowrc r0 )rJ r� r� r� Z share_poolZ_sqlite_savepointr� r� r� r/ r� Z default_optr. r� r� r� r r r r� 0 sh � ���� c sr ddl m} | stjj} g � � fdd�}� fdd�}�fdd�}|| d |��t�d �r\t�� �_|�_|�_ �S ) a' Provides a mocking engine based on the current testing.db. This is normally used to test DDL generation flow as emitted by an Engine. It should not be used in other cases, as assert_compile() and assert_sql_execution() are much better choices with fewer moving parts. r )�create_mock_enginec s � � | � d S r )r~ )�sql�arf ��bufferr r �executor� s z mock_engine.<locals>.executorc s" dd� � D �}|| ks t |��d S )Nc S s g | ]}t �d dt|���qS )z[\n\t]rH )�re�sub�str��.0�sr r r � <listcomp>� s z3mock_engine.<locals>.assert_sql.<locals>.<listcomp>)r- )Zstmts�recvr� r r � assert_sql� s z mock_engine.<locals>.assert_sqlc s �j � d�� fdd��jD ��S )N� c 3 s | ]}t |j� d ��V qdS ))ro N)r� �compiler� �rq r r � <genexpr>� s z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)ro �join�mockr )r. r� r � print_sql� s z mock_engine.<locals>.print_sqlz://r� ) r� r� r rS rr rN r- r� r� r� )Z dialect_namer� r� r� r� r )r� r. r � mock_engine� s r� c @ s: e Zd ZdZdd� Zd dd�Zdd� Zd d � Zd d � ZdS )�DBAPIProxyCursorzuProxy a DBAPI cursor. Tests can provide subclasses of this to intercept DBAPI-level cursor operations. c O s || _ || _|j||�| _d S r )r. rK r| )r r. rV re r r r r r � s zDBAPIProxyCursor.__init__Nc K s* |r| j j||f|�S | j j|f|�S d S r )r| r} )r �stmt� parametersrf r r r r} � s zDBAPIProxyCursor.executec K s | j j||f|�S r )r| � executemany)r r� �paramsrf r r r r� � s z DBAPIProxyCursor.executemanyc C s t | j�S r )�iterr| r r r r �__iter__� s zDBAPIProxyCursor.__iter__c C s t | j|�S r )rm r| ry r r r r{ � s z DBAPIProxyCursor.__getattr__)N) r` ra rb �__doc__r r} r� r� r{ r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �DBAPIProxyConnectionz}Proxy a DBAPI connection. Tests can provide subclasses of this to intercept DBAPI-level connection operations. c C s || _ || _|| _d S r )rV r. � cursor_cls)r r. rV r� r r r r � s z DBAPIProxyConnection.__init__c O s | j | j| jf|�|�S r )r� r. rV )r re r r r r r| � s zDBAPIProxyConnection.cursorc C s | j �� d S r )rV r" r r r r r" � s zDBAPIProxyConnection.closec C s t | j|�S r )rm rV ry r r r r{ � s z DBAPIProxyConnection.__getattr__N)r` ra rb r� r r| r" r{ r r r r r� � s r� )N)NN)NNFF)NNTF)NNFFFF)N))� __future__r r r� �typingr r r r3 r rH r �utilr r r r r Z util.typingr � TYPE_CHECKINGr. r Z engine.urlr Z ext.asyncior r rc rg rh rj rk rt ru r� �overloadr� r� r� r� r r r r �<module> sn + � � � Q &