Submit
Path:
~
/
/
opt
/
alt
/
python37
/
lib64
/
python3.7
/
site-packages
/
sqlalchemy
/
testing
/
__pycache__
/
File Content:
engines.cpython-37.opt-1.pyc
B ��4]�( � @ s d dl mZ d dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm Z dd lm Z G d d � d e �Z e � Zd d � Ze dd� �Ze dd� �Ze dd� �Ze dd� �Zd&dd�ZG dd� de �Zd'dd�Zd(d d �Zd)d d �ZG d d!� d!e �ZG d"d#� d#e �Zeefd$d%�ZdS )*� )�absolute_importN� )�config)�uses_deprecated)� decorator� )�event)�poolc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z e � dd� �Ze � dd� �Zdd� ZdS ) �ConnectionKillerc C s t �� | _t �� | _t� | _d S )N)�weakref�WeakKeyDictionary� proxy_refs�testing_engines�set�conns)�self� r �M/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/engines.py�__init__ s zConnectionKiller.__init__c C s d| j |< d S )NT)r )r �enginer r r � add_engine s zConnectionKiller.add_enginec C s | j �||f� d S )N)r �add)r Z dbapi_conn� con_recordr r r �connect s zConnectionKiller.connectc C s d| j |< d S )NT)r )r � dbapi_conr Z con_proxyr r r �checkout! s zConnectionKiller.checkoutc C s | j �||f� d S )N)r �discard)r r r Z exceptionr r r � invalidate$ s zConnectionKiller.invalidatec C s@ y |� W n0 t k r: } zt�d| � W d 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 s2 x,t | j�D ] }|d k r |jr | �|j� q W d S )N)�listr �is_validr# �rollback)r �recr r r � rollback_all/ s z ConnectionKiller.rollback_allc C s2 x,t | j�D ] }|d k r |jr | �|j� q W d S )N)r$ r r% r# Z_close)r r' r r r � close_all4 s zConnectionKiller.close_allc C s4 x.t | j�D ] \}}|jd kr q | �|j� q W d S )N)r$ r � connectionr# r& )r �connr' r r r �_after_test_ctx9 s z ConnectionKiller._after_test_ctxc C s t jjr| �� n| �� d S )N)r �optionsZlow_connections�_stop_test_ctx_minimal�_stop_test_ctx_aggressive)r r r r �_stop_test_ctxH s z ConnectionKiller._stop_test_ctxc C s: | � � t� | _x$t| j�D ]}|tjk r |�� q W d S )N)r) r r r$ r r �db�dispose)r r' r r r r. N s z'ConnectionKiller._stop_test_ctx_minimalc C sZ | � � x(t| j�D ]\}}| �|j� d |_qW t� | _xt| j�D ] }|�� qFW d S )N) r) r$ r r# �closer* r r r2 )r r+ r' r r r r/ X s z*ConnectionKiller._stop_test_ctx_aggressivec C s x| j D ] }|jrqW d S )N)r r% )r r' r r r �assert_all_closedc s z"ConnectionKiller.assert_all_closedN)�__name__� __module__� __qualname__r r r r r r# r( r) r, r0 r r. r/ r4 r r r r r s r c C sd t �� t|d�r|�� tjjjsVddlm } |j ddd�� | � |� W d Q R X n | � |� d S )Nr3 r )� assertionszCan't sort tablesF)Zassert_) �testing_reaperr) �hasattrr3 r r1 �dialectZsupports_alter� r8 Zexpect_warningsZdrop_all)ZmetadataZbindr8 r r r �drop_all_tablesl s r= c O s z| ||� W d t �� X d S )N)r9 r4 )r! �args�kwr r r �assert_conns_closedz s r@ c O s z| ||� W dt �� X dS )zBDecorator that rolls back all open connections after fn execution.N)r9 r( )r! r> r? r r r �rollback_open_connections� s rA c O s t �� | ||� dS )z:Decorator that closes all connections before fn execution.N)r9 r) )r! r> r? r r r � close_first� s rB c O s z| ||� W dt �� X dS )z9Decorator that closes all connections after fn execution.N)r9 r) )r! r> r? r r r �close_open_connections� s rC c c s\ dd l m} xJ|jD ]@}| r&|| kr&qt||d �}|sJttd| �j|�}|�� V qW d S )Nr zsqlalchemy.databases.%s)Zsqlalchemy.databasesZ databases�__all__�getattr� __import__r; )Zexclude�d�name�modr r r � all_dialects� s rJ c @ s> e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd d � ZdS )�ReconnectFixturec C s || _ g | _d| _d S )NF)�dbapi� connections� is_stopped)r rL r r r r � s zReconnectFixture.__init__c C s t | j|�S )N)rE rL )r �keyr r r � __getattr__� s z ReconnectFixture.__getattr__c O sH | j j||�}| jr4| �|j� |�� }|�d� n| j�|� |S d S )Nzselect 1) rL r rN r# r3 �cursor�executerM �append)r r> �kwargsr+ Zcursr r r r � s zReconnectFixture.connectc C s@ y |� W n0 t k r: } zt�d| � W d d }~X Y nX d S )Nz.ReconnectFixture couldn't close connection: %s)r r r )r r! r" r r r r# � s zReconnectFixture._safeFc C s0 || _ x t| j�D ]}| �|j� qW g | _d S )N)rN r$ rM r# r3 )r �stop�cr r r �shutdown� s zReconnectFixture.shutdownc C s d| _ d S )NF)rN )r r r r �restart� s zReconnectFixture.restartN)F) r5 r6 r7 r rP r r# rW rX r r r r rK � s rK 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 )NF)r; rL rW rN r )�_disposer r r r2 � s z$reconnecting_engine.<locals>.dispose) r r1 �urlr; rL rK �testing_enginer2 rW Z test_shutdownrX Z test_restart)r[ r- rL r2 r )rZ r r �reconnecting_engine� s r] c C s ddl m} ddlm} |s"d}n |�dd�}| p8tjj} || �} |dkrrtjdksd| jtjjjkrltj }q�i }n.tjdk r�| jtjjjkr�tj � � }|� |� || f|�}d|_ t |jtj�r�d|j_d|j_|�rt�|jdtj� t�|jdtj� t�|jd tj� t�|� |S ) zBProduce an engine configured by --options with optional overrides.r )� create_engine)�make_urlT� use_reaperNr r r )� sqlalchemyr^ Zsqlalchemy.engine.urlr_ �popr r1 r[ Z drivernameZdb_opts�copy�updateZ _has_events� isinstancer Z QueuePoolZ_timeoutZ _max_overflowr Zlistenr9 r r r r )r[ r- r^ r_ r` Z default_optr r r r r\ � s2 r\ c sh ddl m} | stjj} g � � fdd�}� fdd�}�fdd�}|| d d |d ��� �_|�_|�_�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 )r^ c s � � | � d S )N)rS )Zsql�ar? )�bufferr r �executor' s z mock_engine.<locals>.executorc s dd� � D �}d S )Nc S s g | ]}t �d dt|���qS )z[\n\t]r< )�re�sub�str)�.0�sr r r � <listcomp>+ s z3mock_engine.<locals>.assert_sql.<locals>.<listcomp>r )ZstmtsZrecv)rg 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 ))r; N)rk �compile)rl rm )rG r r � <genexpr>0 s z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)r; �join�mockr )r )rG r � print_sql. s z mock_engine.<locals>.print_sqlz://rt )Zstrategyrh )ra r^ r r1 rH rt ro ru )Z dialect_namer^ rh ro ru r )rg r r � mock_engine s rv c @ s2 e Zd ZdZdd� Zd dd�Zdd� 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 )N)r r* rQ )r r r+ r> rT r r r r D s zDBAPIProxyCursor.__init__Nc K s* |r| j j||f|�S | j j|f|�S d S )N)rQ rR )r �stmtZ parametersr? r r r rR I s zDBAPIProxyCursor.executec K s | j j||f|�S )N)rQ � executemany)r rx Zparamsr? r r r ry O s z DBAPIProxyCursor.executemanyc C s t | j|�S )N)rE rQ )r rO r r r rP R s z DBAPIProxyCursor.__getattr__)N)r5 r6 r7 �__doc__r rR ry rP r r r r rw < s rw 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" |j �� | _| _|| _|| _d S )N)r Z_creatorr+ Z _sqla_unwrapr � cursor_cls)r r r| r r r r ^ s z DBAPIProxyConnection.__init__c O s | j | j| jf|�|�S )N)r| r r+ )r r> rT r r r rQ c s zDBAPIProxyConnection.cursorc C s | j �� d S )N)r+ r3 )r r r r r3 f s zDBAPIProxyConnection.closec C s t | j|�S )N)rE r+ )r rO r r r rP i s z DBAPIProxyConnection.__getattr__N)r5 r6 r7 rz r rQ r3 rP r r r r r{ V s r{ c s � �fdd�}t d|id�S )zIProduce an engine that provides proxy hooks for common methods. c s � t j��S )N)r r1 r )�conn_clsr| r r � mock_connu s z"proxying_engine.<locals>.mock_connZcreator)r- )r\ )r} r| r~ r )r} r| r �proxying_enginem s r )N)NN)NN)N) Z __future__r ri r r r< r r �utilr r r �objectr r9 r= r@ rA rB rC rJ rK r] r\ rv rw r{ r r r r r �<module> s. T . & (
Edit
Rename
Chmod
Delete
FILE
FOLDER
Name
Size
Permission
Action
__init__.cpython-37.opt-1.pyc
1966 bytes
0644
__init__.cpython-37.pyc
1966 bytes
0644
assertions.cpython-37.opt-1.pyc
16409 bytes
0644
assertions.cpython-37.pyc
17698 bytes
0644
assertsql.cpython-37.opt-1.pyc
11032 bytes
0644
assertsql.cpython-37.pyc
11260 bytes
0644
config.cpython-37.opt-1.pyc
2875 bytes
0644
config.cpython-37.pyc
2949 bytes
0644
engines.cpython-37.opt-1.pyc
11340 bytes
0644
engines.cpython-37.pyc
11466 bytes
0644
entities.cpython-37.opt-1.pyc
2594 bytes
0644
entities.cpython-37.pyc
2594 bytes
0644
exclusions.cpython-37.opt-1.pyc
14253 bytes
0644
exclusions.cpython-37.pyc
14404 bytes
0644
fixtures.cpython-37.opt-1.pyc
12420 bytes
0644
fixtures.cpython-37.pyc
12479 bytes
0644
mock.cpython-37.opt-1.pyc
656 bytes
0644
mock.cpython-37.pyc
656 bytes
0644
pickleable.cpython-37.opt-1.pyc
5210 bytes
0644
pickleable.cpython-37.pyc
5210 bytes
0644
profiling.cpython-37.opt-1.pyc
7254 bytes
0644
profiling.cpython-37.pyc
7254 bytes
0644
provision.cpython-37.opt-1.pyc
12156 bytes
0644
provision.cpython-37.pyc
12156 bytes
0644
replay_fixture.cpython-37.opt-1.pyc
6513 bytes
0644
replay_fixture.cpython-37.pyc
6513 bytes
0644
requirements.cpython-37.opt-1.pyc
36850 bytes
0644
requirements.cpython-37.pyc
36850 bytes
0644
schema.cpython-37.opt-1.pyc
2882 bytes
0644
schema.cpython-37.pyc
2882 bytes
0644
util.cpython-37.opt-1.pyc
8171 bytes
0644
util.cpython-37.pyc
8197 bytes
0644
warnings.cpython-37.opt-1.pyc
1095 bytes
0644
warnings.cpython-37.pyc
1095 bytes
0644
N4ST4R_ID | Naxtarrr
