Файловый менеджер - Редактировать - /opt/imh-python/lib/python3.9/site-packages/sqlalchemy/testing/__pycache__/engines.cpython-39.pyc
Ðазад
a _�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 ��/root/rpmbuild/BUILDROOT/imh-python39-modules-3.9.7-92.el8.x86_64/opt/imh-python/lib/python3.9/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 r )r �add)r Z dbapi_conn� con_recordr r r �connect s zConnectionKiller.connectc C s d| j |< d S r )r )r � dbapi_conr Z con_proxyr r r �checkout! s zConnectionKiller.checkoutc C s | j �||f� d S r )r �discard)r r r � exceptionr r r � invalidate$ s zConnectionKiller.invalidatec C sB z |� W n2 t y< } zt�d| � W Y d }~n d }~0 0 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 ur |jr | �|j� q d S r )�listr �is_validr) �rollback�r �recr r r � rollback_all/ s z ConnectionKiller.rollback_allc C s. t | j�D ] }|d ur |jr | �|j� q d S r )r* r r+ r) �_closer- r r r � close_all4 s zConnectionKiller.close_allc C s0 t | j�D ] \}}|jd u r q | �|j� q d S r )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 r )r �optionsZlow_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|�� qd S r )r1 r r r* r r �db�disposer- r r r r7 N s z'ConnectionKiller._stop_test_ctx_minimalc C sR | � � t| j�D ]\}}| �|j� d |_qt� | _t| j�D ] }|�� q@d S r ) r1 r* r r) �closer2 r r r; r3 r r r r8 X s z*ConnectionKiller._stop_test_ctx_aggressivec C s | j D ]}|jrdsJ �qd S �NF)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/ r1 r5 r9 r r7 r8 r> r r r r r s r c C sx t �� t|d�r|�� tjjjsjddlm } |j ddd�� | � |� W d � qt1 s^0 Y n | � |� d S )Nr< r )� assertionszCan't sort tablesF)Zassert_) �testing_reaperr1 �hasattrr< r r: �dialectZsupports_alter� rB Zexpect_warningsZdrop_all)�metadata�bindrB r r r �drop_all_tablesl s *rI c O s* z| |i |�� W t �� n t �� 0 d S r )rC r> �r' �args�kwr r r �assert_conns_closedz s rM c O s* z| |i |�� W t �� n t �� 0 dS )zBDecorator that rolls back all open connections after fn execution.N)rC r/ rJ r r r �rollback_open_connections� s rN c O s t �� | |i |�� dS )z:Decorator that closes all connections before fn execution.N�rC r1 rJ r r r � close_first� s rP c O s* z| |i |�� W t �� n t �� 0 dS )z9Decorator that closes all connections after fn execution.NrO rJ r r r �close_open_connections� s rQ c c sX dd l m} |jD ]@}| r$|| v r$qt||d �}|sHttd| �j|�}|�� V qd S )Nr zsqlalchemy.databases.%s)Zsqlalchemy.databasesZ databases�__all__�getattr� __import__rE )�exclude�d�name�modr r r � all_dialects� s �rY 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 r[ r r r r � s zReconnectFixture.__init__c C s t | j|�S r )rS r[ �r �keyr r r � __getattr__� s z ReconnectFixture.__getattr__c O sX | j j|i |��}| jrD| �|j� |�� }|�d� dsTJ d��n| j�|� |S d S )Nzselect 1Fz%simulated connect failure didn't work) r[ r r] r) r< �cursor�executer\ �append)r rK �kwargsr4 Zcursr r r r � s zReconnectFixture.connectc C sB z |� W n2 t y< } zt�d| � W Y d }~n d }~0 0 d S )Nz.ReconnectFixture couldn't close connection: %sr" r& r r r r) � s �zReconnectFixture._safeFc C s, || _ t| j�D ]}| �|j� qg | _d S r )r] r* r\ r) r< )r �stop�cr r r �shutdown� s zReconnectFixture.shutdownc C s d| _ d S r= )r] r r r r �restart� s zReconnectFixture.restartN)F) r? r@ rA r r` r r) rg rh r r r r rZ � s rZ 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= )rE r[ rg r] r �Z_disposer r r r; � s z$reconnecting_engine.<locals>.dispose) r r: �urlrE r[ rZ �testing_enginer; rg Z test_shutdownrh Z test_restart)rk r6 r[ r; r rj r �reconnecting_engine� s rm c C s ddl m} ddlm} |s"d}n |�dd�}| p8tjj} || �} |du rrtjdu sd| jtjjjkrltj }q�i }n.tjdur�| jtjjjkr�tj � � }|� |� || fi |��}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! )� sqlalchemyro Zsqlalchemy.engine.urlrp �popr r: rk Z drivernameZdb_opts�copy�updateZ _has_events� isinstancer Z QueuePool�_timeoutZ _max_overflowr �listenrC r r r! r )rk r6 ro rp rq Z default_optr r r r rl � s2 rl c sv ddl m} | stjj} g � � fdd�}� fdd�}�fdd�}|| d d |d ��t�d �r`J �� �_|�_|�_�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 rn c s � � | � d S r )rc )�sql�arL ��bufferr r �executor' s z mock_engine.<locals>.executorc s" dd� � D �}|| ks J |��d S )Nc S s g | ]}t �d dt|���qS )z[\n\t]rF )�re�sub�str��.0�sr r r � <listcomp>+ � 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 ))rE N)r� �compiler� �rV r r � <genexpr>0 r� z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)rE �join�mockr )r r� r � print_sql. s z mock_engine.<locals>.print_sqlz://r� )Zstrategyr} ) rr ro r r: rW rD r� r� r� )Z dialect_namero r} r� r� r )r| r r � mock_engine s �r� c @ sB e Zd ZdZdd� Zddd�Zdd� Zd d � Zd d � Zd d� Z dS )�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 r2 ra )r r r4 rK rd r r r r D s zDBAPIProxyCursor.__init__Nc K s2 |r| j j||fi |��S | j j|fi |��S d S r )ra rb )r �stmt� parametersrL r r r rb I s zDBAPIProxyCursor.executec K s | j j||fi |��S r )ra � executemany)r r� �paramsrL r r r r� O s z DBAPIProxyCursor.executemanyc C s t | j|�S r )rS ra r^ r r r r` R s z DBAPIProxyCursor.__getattr__c C s | S r r r r r r � __enter__U s zDBAPIProxyCursor.__enter__c G s d S r r )r rK r r r �__exit__X s zDBAPIProxyCursor.__exit__)N) r? r@ rA �__doc__r rb 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" |j �� | _| _|| _|| _d S r )r Z_creatorr4 Z _sqla_unwrapr � cursor_cls)r r r� r r r r d s z DBAPIProxyConnection.__init__c O s | j | j| jg|�R i |��S r )r� r r4 )r rK rd r r r ra i s zDBAPIProxyConnection.cursorc C s | j �� d S r )r4 r< r r r r r< l s zDBAPIProxyConnection.closec C s t | j|�S r )rS r4 r^ r r r r` o s z DBAPIProxyConnection.__getattr__N)r? r@ rA r� r ra r< r` r r r r r� \ 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 r: r ��conn_clsr� r r � mock_conn{ s z"proxying_engine.<locals>.mock_connZcreator)r6 )rl )r� r� r� r r� r �proxying_engines s r� )N)NN)NN)N) � __future__r r~ r$ r rF r r �utilr r r �objectr rC rI rM rN rP rQ rY rZ rm rl r� r� r� r� r r r r �<module> s8 T . & ( �
| ver. 1.1 | |
.
| PHP 8.3.30 | Ð“ÐµÐ½ÐµÑ€Ð°Ñ†Ð¸Ñ Ñтраницы: 0 |
proxy
|
phpinfo
|
ÐаÑтройка