Dashboard
PHP:
8.2.30
OS:
Linux
User:
dh_nv8b2m
/
/
usr
/
local
/
lib
/
python3.10
/
dist-packages
/
sqlalchemy
/
testing
/
__pycache__
📤 Upload
📝 New File
📁 New Folder
Close
Editing: engines.cpython-310.pyc
o [�h�( � @ 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 �E/usr/local/lib/python3.10/dist-packages/sqlalchemy/testing/engines.py�__init__ s zConnectionKiller.__init__c C s d| j |< d S �NT)r )r �enginer r r � add_engine � zConnectionKiller.add_enginec C � | j �||f� d S r )r �add)r � dbapi_conn� con_recordr r r �connect � zConnectionKiller.connectc C s d| j |< d S r )r )r � dbapi_conr � con_proxyr r r �checkout! r zConnectionKiller.checkoutc C r r )r �discard)r r! r � exceptionr r r � invalidate$ r zConnectionKiller.invalidatec C �B z|� W d S t y } zt�d| � W Y d }~d S d }~ww )Nz5testing_reaper couldn't rollback/close connection: %s�� Exception�warnings�warn�r �fn�er r r �_safe' � ���zConnectionKiller._safec C �. t | j�D ]}|d ur|jr| �|j� qd S r )�listr �is_validr/ �rollback�r �recr r r � rollback_all/ � ��z ConnectionKiller.rollback_allc C r1 r )r2 r r3 r/ �_closer5 r r r � close_all4 r8 zConnectionKiller.close_allc C s0 t | j�D ]\}}|jd u rq| �|j� qd S r )r2 r � connectionr/ r4 �r �connr6 r r r �_after_test_ctx9 s �z ConnectionKiller._after_test_ctxc C s t jjr | �� d S | �� d S r )r �options�low_connections�_stop_test_ctx_minimal�_stop_test_ctx_aggressiver r r r �_stop_test_ctxH s z ConnectionKiller._stop_test_ctxc C s6 | � � t� | _t| j�D ] }|tjur|�� q d S r )r: r r r2 r r �db�disposer5 r r r rA N s ��z'ConnectionKiller._stop_test_ctx_minimalc C sR | � � t| j�D ] \}}| �|j� d |_q t� | _t| j�D ]}|�� q d S r ) r: r2 r r/ �closer; r r rE r< r r r rB X s �z*ConnectionKiller._stop_test_ctx_aggressivec C s | j D ]}|jr J �qd S r )r r3 r5 r r r �assert_all_closedc s ��z"ConnectionKiller.assert_all_closedN)�__name__� __module__� __qualname__r r r r# r& r/ r7 r: r> rC r rA rB rG r r r r r s r c C s| t �� t|d�r |�� tjjjs7ddlm } |j ddd�� | � |� W d � d S 1 s0w Y d S | � |� d S )NrF r )� assertionszCan't sort tablesF)�assert_) �testing_reaperr: �hasattrrF r rD �dialect�supports_alter� rK �expect_warnings�drop_all)�metadata�bindrK r r r �drop_all_tablesl s "�rV c O s( z| |i |�� W t �� d S t �� w r )rM rG �r- �args�kwr r r �assert_conns_closedz s rZ c O �( z| |i |�� W t �� dS t �� w )zBDecorator that rolls back all open connections after fn execution.N)rM r7 rW r r r �rollback_open_connections� s r\ c O s t �� | |i |�� dS )z:Decorator that closes all connections before fn execution.N�rM r: rW r r r � close_first� s r^ c O r[ )z9Decorator that closes all connections after fn execution.Nr] rW r r r �close_open_connections� s r_ c c sZ � dd l m} |jD ] }| r|| v rq t||d �}|s%ttd| �j|�}|�� V q d S )Nr zsqlalchemy.databases.%s)�sqlalchemy.databases� databases�__all__�getattr� __import__rO )�exclude�d�name�modr r r � all_dialects� s � � �ri 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 � t | j|�S r )rc rl �r �keyr r r � __getattr__� � z ReconnectFixture.__getattr__c O sN | j j|i |��}| jr | �|j� |�� }|�d� J d��| j�|� |S )Nzselect 1Fz%simulated connect failure didn't work) rl r rn r/ rF �cursor�executerm �append)r rX �kwargsr= �cursr r r r � s zReconnectFixture.connectc C r' )Nz.ReconnectFixture couldn't close connection: %sr( r, r r r r/ � r0 zReconnectFixture._safeFc C s, || _ t| j�D ]}| �|j� qg | _d S r )rn r2 rm r/ rF )r �stop�cr r r �shutdown� s zReconnectFixture.shutdownc C s d| _ d S rk )rn r r r r �restart� s zReconnectFixture.restartN)F) rH rI rJ r rr r r/ r{ r| r r r r rj � s rj c sj | pt jj} t jjj}|si }t|�|d< t| |���j� � �fdd�}�jjj�_ �jjj �_ |�_�S )N�modulec s �j j�� d�j j_� � d S rk )rO rl r{ rn r ��_disposer r r rE � s z$reconnecting_engine.<locals>.dispose) r rD �urlrO rl rj �testing_enginerE r{ � test_shutdownr| � test_restart)r� r? rl rE r r~ r �reconnecting_engine� s r� c C s ddl m} ddlm} |sd}n|�dd�}| p tjj} || �} |du r9tjdu s2| jtjjjkr6tj }ni }ntjdurP| jtjjjkrPtj � � }|� |� || fi |��}d|_ t |jtj�rjd|j_d|j_|r�t�|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� �sqlalchemy.engine.urlr� �popr rD r� � drivername�db_opts�copy�update� _has_events� isinstancer � QueuePool�_timeout� _max_overflowr �listenrM r r# r& r )r� r? r� r� r� � default_optr r r r r� � s2 r� c sv ddl m} | s tjj} g � � fdd�}� fdd�}�fdd�}|| d d |d ��t�d �r0J �� �_|�_|�_�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 r )rv )�sql�arY ��bufferr r �executor' r z mock_engine.<locals>.executorc s" dd� � D �}|| ksJ |��d S )Nc S s g | ] }t �d dt|���qS )z[\n\t]rQ )�re�sub�str��.0�sr r r � <listcomp>+ s z3mock_engine.<locals>.assert_sql.<locals>.<listcomp>r )�stmts�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� �rf r r � <genexpr>0 s � z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)rO �join�mockr )r r� r � print_sql. s z mock_engine.<locals>.print_sqlz://r� )�strategyr� ) r� r� r rD rg rN r� r� r� )� dialect_namer� r� r� r� r )r� r r � mock_engine s �r� 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|i |��| _d S r )r r; rt )r r r= rX rw r r r r D s zDBAPIProxyCursor.__init__Nc K s. |r | j j||fi |��S | j j|fi |��S r )rt ru )r �stmt� parametersrY r r r ru I s zDBAPIProxyCursor.executec K s | j j||fi |��S r )rt � executemany)r r� �paramsrY r r r r� O s z DBAPIProxyCursor.executemanyc C ro r )rc rt rp r r r rr R rs z DBAPIProxyCursor.__getattr__r )rH rI rJ �__doc__r ru r� rr 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" |j �� | _| _|| _|| _d S r )r �_creatorr= � _sqla_unwrapr � cursor_cls)r r r� r r r r ^ s z DBAPIProxyConnection.__init__c O s | j | j| jg|�R i |��S r )r� r r= )r rX rw r r r rt c s zDBAPIProxyConnection.cursorc C s | j �� d S r )r= rF r r r r rF f r zDBAPIProxyConnection.closec C ro r )rc r= rp r r r rr i rs z DBAPIProxyConnection.__getattr__N)rH rI rJ r� r rt rF rr 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 r )r rD r ��conn_clsr� r r � mock_connu rs z"proxying_engine.<locals>.mock_conn�creator)r? )r� )r� r� r� r r� r �proxying_enginem s r� r )NN) � __future__r r� r* r rQ r r �utilr r r �objectr rM rV rZ r\ r^ r_ ri rj r� r� r� r� r� r� r r r r �<module> s: T . &(�
Save
Cancel