?????????? ????????? - ??????????????? - /home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/testing/fixtures/__pycache__/sql.cpython-39.pyc
???????
a ��Jh > � @ s d dl mZ d dlZd dlZd dlZd dlZd dlZddlm Z ddl m Z ddl m Z ddl mZ dd l mZ dd lmZ dd lmZ d d l mZ d dl mZ d dlmZ d dlmZ d dlmZ G dd� de �ZG dd� d�ZG dd� d�Z G dd� de�Z G dd� d�Z d d d �Z dS ) � )� annotationsN� )�TestBase� )�config)�mock)�eq_)�ne_)�adict)� drop_all_tables_from_metadata� )�event)�util)�sort_tables_and_constraints)�visitors)� ClauseElementc @ s e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ ejddd�dd� �Zejdd d�d d � �Zed d � �Zedd� �Zedd� �Zedd� �Zdd� Zdd� Zdd� Zedd� �Zed d � �Z ed d � �Z ed d!� �Z ed"d#� �Z ed$d%� �Z d&d'� Z!d(d)� Z"ed*d+� �Z#dS ),� TablesTest�once�eachNT�class�ZautouseZscopec c s0 | j }|�� |�� |�� d V |�� d S �N)� __class__� _init_class�_setup_once_tables�_setup_once_inserts� _teardown_once_metadata_bind)�self�cls� r �T/home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/testing/fixtures/sql.py�_setup_tables_test_class7 s z#TablesTest._setup_tables_test_class�functionc c s" | � � | �� d V | �� d S r )�_setup_each_tables�_setup_each_inserts�_teardown_each_tables�r r r r �_setup_tables_test_instanceD s z&TablesTest._setup_tables_test_instancec C s | j S r )�_tables_metadatar& r r r �tables_test_metadataM s z TablesTest.tables_test_metadatac C sX | j dkr(| jdkrd| _| jdv s(J �t� | _t� | _t� | _| �� | _t � � | _ d S )Nr r )r N) �run_define_tables�run_create_tables� run_insertsr �other�tables� sequences� setup_bind�bind�saZMetaDatar( �r r r r r Q s zTablesTest._init_classc C sJ | j dkrF| �� | j�� �}| �|� W d � n1 s<0 Y d S �Nr �r, �_load_fixturesr1 �begin� insert_data)r �connr r r r _ s z TablesTest._setup_once_insertsc C sR | j dkrN| �| j� | jdkr.| j�| j� | j�| jj� | j�| jj � d S r4 � r* � define_tablesr( r+ Z create_allr1 r. �updater/ Z _sequencesr3 r r r r f s z TablesTest._setup_once_tablesc C sl | j dkrP| �| j� | jdkr.| j�| j� | j�| jj� | j�| jj � n| jdkrh| j�| j� d S �Nr r: r& r r r r# o s z TablesTest._setup_each_tablesc C sJ | j dkrF| �� | j�� �}| �|� W d � n1 s<0 Y d S r= r5 )r r9 r r r r$ y s z TablesTest._setup_each_insertsc C sp | j dkr8| j�� | jdkr,t| j| j� | j�� n| jdkrPt| j| j� ttj dd�}|rh|j }| j dk�rl| jdk�rl| j dk�rl| j� � ��}t dd� t| jj�� �D ��D ]�}zN|r�|�� � |�|�� � W d � n1 s�0 Y n|�|�� � W q� tjj�yH } z"td||f tjd� W Y d }~q�d }~0 0 q�W d � n1 �sb0 Y d S )Nr � savepointsFc S s g | ]\}}|d ur|�qS r r )�.0�t�fksr r r � <listcomp>� s �z4TablesTest._teardown_each_tables.<locals>.<listcomp>zError emptying table %s: %r)�file)r* r. �clearr+ r r( r1 �getattrr � requirements�enabled� run_deletesr7 �reversedr �valuesZ begin_nested�execute�deleter2 �excZ DBAPIError�print�sys�stderr)r r> r9 �table�exr r r r% sD ��� ��� . �z TablesTest._teardown_each_tablesc C sF | j rt| j| j� | jdkr*| �| j� d | j_| jd urBd | _d S r4 )r+ r r( r1 �run_dispose_bind� dispose_bind�run_setup_bindr3 r r r r � s z'TablesTest._teardown_once_metadata_bindc C s t jS r )r �dbr3 r r r r0 � s zTablesTest.setup_bindc C s* t |d�r|�� nt |d�r&|�� d S )N�dispose�close)�hasattrrW rX )r r1 r r r rT � s zTablesTest.dispose_bindc C s d S r r )r �metadatar r r r; � s zTablesTest.define_tablesc C s i S r r r3 r r r �fixtures� s zTablesTest.fixturesc C s d S r r )r � connectionr r r r8 � s zTablesTest.insert_datac C s | � | j||� d S r )Zassert_sql_countr1 )r �count�fnr r r � sql_count_� s zTablesTest.sql_count_c C s | � | j||� d S r )Z assert_sqlr1 )r Z callable_Z statementsr r r �sql_eq_� s zTablesTest.sql_eq_c s� i i � }| � � �� D ]F\�}t|�dk r,qt�t�r@| j� �|d � �< |dd� |�< qt| jj�� �D ]j\�}�du r�qn�� vr�qn| j � � �4}|� �� � � �fdd�|� D �� W d � qn1 s�0 Y qndS )z4Insert rows as represented by the fixtures() method.r r r Nc s g | ]}t t� � |���qS r )�dict�zip)r? Z column_values��headersrQ r r rB � s �z-TablesTest._load_fixtures.<locals>.<listcomp>) r[ �items�len� isinstance�strr. r r( rJ r1 r7 rK �insert)r �rows�datarA r9 r rc r r6 � s, � ��zTablesTest._load_fixtures)$�__name__� __module__� __qualname__rU r* r+ r, rH rS r1 r( r. r- r/ r �fixturer! r' �propertyr) � classmethodr r r r# r$ r% r r0 rT r; r[ r8 r_ r` r6 r r r r r sT ) r c @ s" e Zd Zejddd�dd� �ZdS )�NoCacheTr" r c c s" t jj}d t j_d V |t j_d S r )r rV Z_compiled_cache)r �_cacher r r �_disable_cache� s zNoCache._disable_cacheN)rl rm rn r ro rt r r r r rr � s rr c @ s8 e Zd Zejdd� �Zdd� Zejddd�dd � �Z d S ) � RemovesEventsc C s t � S r )�setr& r r r � _event_fns� s zRemovesEvents._event_fnsc K s, | j �|||f� tj|||fi |�� d S r )rw �addr �listen)r �target�namer^ �kwr r r � event_listen� s zRemovesEvents.event_listenTr" r c c s d V | j D ]}tj|� q d S r )rw r �remove)r �keyr r r �_remove_events s z RemovesEvents._remove_eventsN) rl rm rn r Zmemoized_propertyrw r} r ro r� r r r r ru � s ru c @ s: e Zd Zd ZZdZdZe�d�Z dd� Z e dd� �Z dS ) � ComputedReflectionFixtureTestNT)Zcomputed_columnsZtable_reflectionz[\[\]\(\)\s`'\"]*c C s | j �d|��� S )N� )�regexp�sub�lower)r �textr r r � normalize s z'ComputedReflectionFixtureTest.normalizec C sx ddl m} ddl m} ddlm} ddlm} ddlm} |d||d|d d �|d |�|d ||d ��|d|dd�� |d||d|d d �|d |�|d||d ���}|jjj r�|d||d|d d �|d |�|d||d��t j d�}|jj j �r(|� |d||ddd��� |jjj �r(|� |d||ddd��� |jjj �rt|� |d||dd d��� |jjj �rt|� |d||d d d��� d S ) Nr )�Integer)�testing)�Column)�Computed)�TableZcomputed_default_table�idT)Z primary_key�normalZ computed_colz normal + 42Z with_defaultZ42)Zserver_defaultZcomputed_column_tableZcomputed_no_flagz normal / 42)�schemaZcomputed_virtualz normal + 2F)Z persistedz normal / 2Zcomputed_storedz normal - 42z normal * 42)r� r� r� r� r� r� r� �requiresZschemasrG r Z test_schemaZcomputed_columns_virtualZ append_columnZcomputed_columns_stored) r rZ r� r� r� r� r� r@ �t2r r r r; sz � � � �� �� �� ��z+ComputedReflectionFixtureTest.define_tables) rl rm rn r, rH Z __backend__� __requires__�re�compiler� r� rq r; r r r r r� s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CacheKeyFixturec C s� |� � }|� � }|d u r6|j�d�s(J �|d u s�J �nNt|j|j� tt|j�t|j�� t|j|j�D ]\}}|j||d�shJ �qh||fS )N�nocache��compare_values) �_generate_cache_key� _annotations�getr r �hashrb � bindparams�compare)r �a�br� �a_key�b_key�a_param�b_paramr r r �_compare_equalX s z CacheKeyFixture._compare_equalc C s� |� }|� }t �tt|��d�D �]�\}}||krX| �|| || |�\}}|d u rVq n�|| �� }|| �� }|d u s�|d u r�|d u r�|| j�d�s�J �|d u r || j�d�s J �q |j|jk�rt |j |j �D ] \} } | j | |d�s� �qq�t |j|j� nt |j|j� t || t�r t || t�r g } g } t�|| �D ] } | jdk�rD| �| � �qDt�|| �D ] } | jdk�rp| �| � �qptt|j dd� d�tt�| �dd� d�� tt|j d d� d�tt�| �d d� d�� q d S ) Nr r� r� Z bindparamc S s | j S r �r �r� r r r �<lambda>� � z8CacheKeyFixture._run_cache_key_fixture.<locals>.<lambda>r� c S s | j S r r� r� r r r r� � r� c S s | j S r r� r� r r r r� � r� c S s | j S r r� r� r r r r� � r� )� itertools� combinations_with_replacement�rangerf r� r� r� r� r rb r� r� r rg r r ZiterateZ__visit_name__�appendr �sortedr Z unique_list)r ro r� �case_a�case_br� r� r� r� r� r� Zassert_a_paramsZassert_b_params�elemr r r �_run_cache_key_fixtureh sh �� � �� ����z&CacheKeyFixture._run_cache_key_fixturec C sD |� }|� }t �tt|��d�D ] \}}| �|| || |� q d S )Nr )r� r� r� rf r� )r ro r� r� r� r� r� r r r � _run_cache_key_equal_fixture� s � z,CacheKeyFixture._run_cache_key_equal_fixtureN)rl rm rn r� r� r� r r r r r� W s Kr� Fc sJ | j }|j�| j�G dd� d�� � ���fdd����fdd�}|| _d S )Nc @ s, e Zd ZdZdd� Zedd� �Zdd� ZdS ) z.insertmanyvalues_fixture.<locals>.RandomCursor��cursorc S s || _ d S r r� )r r� r r r �__init__� s z7insertmanyvalues_fixture.<locals>.RandomCursor.__init__c S s | j jS r )r� � descriptionr& r r r r� � s z:insertmanyvalues_fixture.<locals>.RandomCursor.descriptionc S s | j �� }t|�}t�|� |S r )r� �fetchall�list�random�shuffle)r rj r r r r� � s z7insertmanyvalues_fixture.<locals>.RandomCursor.fetchallN)rl rm rn � __slots__r� rp r� r� r r r r � RandomCursor� s r� c 3 sB �r � |�}�| |||||�D ] }�r6|j r6t�d� |V q d S )Nz)Batches were downgraded for sorted INSERT)Z is_downgradedr �warn)r\ r� Z statement� parametersZgeneric_setinputsizes�context�batch)r� � orig_dialect�randomize_rows�warn_on_downgradedr r �!_deliver_insertmanyvalues_batches� s � zCinsertmanyvalues_fixture.<locals>._deliver_insertmanyvalues_batchesc s@ t jj| d� d�� �| |�W d � S 1 s20 Y d S )Nr� )�new)r �patch�object)�dialectr� )r� � orig_connr r �_exec_insertmany_context� s �z:insertmanyvalues_fixture.<locals>._exec_insertmany_context)r� r� r� )r\ r� r� r� r� r )r� r� r� r� r� r� r �insertmanyvalues_fixture� s r� )FF) � __future__r r� r� r� rO Z sqlalchemyr2 �baser r� r r Z assertionsr r r r r r r� r Zsqlr Z sql.elementsr r rr ru r� r� r� r r r r �<module> s2 Q Pg �
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????