?????????? ????????? - ??????????????? - /home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/testing/__pycache__/engines.cpython-39.pyc
???????
a ��Jhf4 � @ 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 �O/home/agenciai/.local/lib/python3.9/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� |dv sJ �| j| �|� d S )N)�class�global�functionZfixture)r& r r r' )r �engine�scoper r r � add_engine9 s zConnectionKiller.add_enginec 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_validr7 �rollback�r �recr r r � rollback_allG s z ConnectionKiller.rollback_allc C s^ t | j�D ],}|d ur |jr | j�|j� | �|j� q | jD ]}| �|j� q>| j� � d S r ) r8 r r9 r r( �dbapi_connectionr7 �_checkinr: �clear)r r< �conr r r � checkin_allL s z ConnectionKiller.checkin_allc C s | � � d S r )rB 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� rE �prepare_for_drop_tablesr- �url)r � connectionrE r r r rH ^ s z(ConnectionKiller.prepare_for_drop_tablesc C s� | j | }t|�D ]d}t| j�D ]4}|d ur |jr |jd ur |j|ju r | �|j� q t|d�rnt |� � � q|� � q|� � d S )N� sync_engine) r r8 r r9 �_poolr r7 r? �hasattrr �disposer@ )r r. Zengr< Z proxy_refr r r �_drop_testing_enginesg s � � z&ConnectionKiller._drop_testing_enginesc C s | � d� d S )Nr, )rO r r r r � after_testx s zConnectionKiller.after_testc C sh t js d S |jjrd S | �� ddlm} t j�� � }|� |j j |� W d � n1 sZ0 Y d S rD ) r rF � __class__Z"__leave_connections_for_teardown__rB rG rE �db�connectrH r- rI )r �testrE �connr r r �after_test_outside_fixtures{ s z,ConnectionKiller.after_test_outside_fixturesc C s | � � | �d� | �d� d S )Nr, r* )rB rO 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>J d| ��d S )NFz/%d connection recs not cleared after test suite)r �baseZ _strong_ref_connection_recordsr �lenr@ )r �lnr r r � stop_test_class_outside_fixtures� s ��z1ConnectionKiller.stop_test_class_outside_fixturesc C s" | � � | jD ]}| �|� qd S r )rB r rO )r r. r r r � final_cleanup� s z ConnectionKiller.final_cleanupc C s | j D ]}|jrdsJ �qd S �NF)r r9 r; r r r �assert_all_closed� s z"ConnectionKiller.assert_all_closedN)�__name__� __module__� __qualname__r r& r$ r% r/ r7 r= rB rC rH rO rP rV rW r[ r\ r^ r r r r r $ s" r c O s* z| |i |�� W t �� n t �� 0 d S r )�testing_reaperr^ �r5 �args�kwr r r �assert_conns_closed� s rf c O s* z| |i |�� W t �� n t �� 0 dS )zBDecorator that rolls back all open connections after fn execution.N)rb r= rc r r r �rollback_open_connections� s rg c O s t �� | |i |�� dS )z:Decorator that closes all connections before fn execution.N�rb rB rc r r r � close_first� s ri c O s* z| |i |�� W t �� n t �� 0 dS )z9Decorator that closes all connections after fn execution.Nrh rc r r r �close_open_connections� s rj c c sX dd l m} |jD ]@}| r$|| v r$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 �rs 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 ru r r r r � s zReconnectFixture.__init__c C s t | j|�S r )rl ru �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) ru rS rw r7 r" �cursor�executerv �append)r rd �kwargsrU Zcursr r r rS � 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: %sr0 r4 r r r r7 � s zReconnectFixture._safeFc C s, || _ t| j�D ]}| �|j� qg | _d S r )rw r8 rv r7 r" )r �stop�cr r r �shutdown� s zReconnectFixture.shutdownc C s d| _ d S r] )rw r r r r �restart s zReconnectFixture.restartN)F) r_ r` ra r rz rS r7 r� r� r r r r rt � s rt 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] )rn ru r� rw r �Z_disposer- r r rN s z$reconnecting_engine.<locals>.dispose) r rR rI rn ru rt �testing_enginerN r� Z test_shutdownr� Z test_restart)rI �optionsru rN r r� r �reconnecting_engine s r� Fz Optional[URL]zOptional[Dict[str, Any]]zLiteral[False]�boolr )rI r� �asyncio�transfer_staticpool�returnc C s d S r r �rI 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 J �ddl m} n ddlm} ddlm} |sDd}d} d} n$|�dd�}|�d d�} |�d d�} | prtjj } || �} tjd u s�| j tjj j kr�tj r�tj � � } ni } |d ur�| � |� || fi | ��} | �r| jd k�rt�| d �d d� �} t�| d�dd� �}|�rLddlm} tjd u�r\ttjj|��r\d}| j�tjj� n|�r\tjj| _| dk�r||�rvd| j_nd| _t| jtj��r�d| v�r�d| v�r�d| v�r�d| j_d| j_|�r�t� | | � | S )Nr )�create_async_engine)� create_engine)�make_urlTr, F� use_reaperr. �sqlite_savepointZsqliterS c S s d | _ d S r )Zisolation_level)r> Zconnection_recordr r r � do_connect\ s z"testing_engine.<locals>.do_connect�beginc S s | � d� d S )NZBEGIN)Zexec_driver_sql)rU r r r �do_begin` s z testing_engine.<locals>.do_begin)� StaticPoolr+ r � pool_timeoutZ max_overflow) Zsqlalchemy.ext.asyncior� � sqlalchemyr� Zsqlalchemy.engine.urlr� �popr rR rI Z drivernameZdb_opts�copy�updaterq r Z listens_forZsqlalchemy.poolr� � isinstancer Z_transfer_fromrK Z _has_eventsZ QueuePool�_timeoutZ _max_overflowrb r/ )rI r� r� r� Z share_poolZ_sqlite_savepointr� r� r� r. r� Z use_optionsr- r� r� r� r r r r� 0 sn ��� � ���� c sr ddl m} | stjj} g � � fdd�}� fdd�}�fdd�}|| 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 )�create_mock_enginec s � � | � d S r )r} )Zsql�are ��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]rG )�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 ))rn N)r� �compiler� �rp r r � <genexpr>� r� z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)rn �join�mockr )r- r� r � print_sql� s z mock_engine.<locals>.print_sqlz://r� ) r� r� r rR rq rM 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|i |��| _d S r )r- rJ r{ )r r- rU rd r~ r r r r � s zDBAPIProxyCursor.__init__Nc K s2 |r| j j||fi |��S | j j|fi |��S d S r )r{ r| )r �stmt� parametersre r r r r| � s zDBAPIProxyCursor.executec K s | j j||fi |��S r )r{ � executemany)r r� �paramsre 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 )rl r{ rx r r r rz � s z DBAPIProxyCursor.__getattr__)N) r_ r` ra �__doc__r r| r� r� rz 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 )rU r- � cursor_cls)r r- rU r� r r r r � s z DBAPIProxyConnection.__init__c O s | j | j| jg|�R i |��S r )r� r- rU )r rd r~ r r r r{ � s zDBAPIProxyConnection.cursorc C s | j �� d S r )rU r" r r r r r" � s zDBAPIProxyConnection.closec C s t | j|�S r )rl rU rx r r r rz � s z DBAPIProxyConnection.__getattr__N)r_ r` ra r� r r{ r" rz r r r r r� � s r� )N)NN)NNFF)NNTF)NNFFFF)N))� __future__r r r� �typingr r r r2 r rG r �utilr r r r r Z util.typingr � TYPE_CHECKINGr- r Z engine.urlr Z ext.asyncior r rb rf rg ri rj rs rt r� �overloadr� r� r� r� r r r r �<module> sn + � � � S &
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????