?????????? ????????? - ??????????????? - /home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/testing/__pycache__/assertsql.cpython-39.pyc
???????
a ��Jh�A � @ sT d dl mZ d dlZd dlZd dlZd dlZddlmZ ddlm Z ddl m Z ddl m Z G dd � d �ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG d d � d e�ZG d d � d �ZG d d!� d!e�d!g d"���ZG d#d$� d$�Z ej d%d&� �Z dS )'� )� annotationsN� )�event)�url)�DefaultDialect)�BaseDDLElementc @ s( e Zd ZdZdZdZdd� Zdd� ZdS )� AssertRuleFNTc C s d S �N� ��self�execute_observedr r �Q/home/agenciai/.local/lib/python3.9/site-packages/sqlalchemy/testing/assertsql.py�process_statement s z AssertRule.process_statementc C s ds J d��d S )NFz?All statements are complete, but pending assertion rules remainr �r r r r �no_more_statements s �z AssertRule.no_more_statements)�__name__� __module__� __qualname__� is_consumed� errormessage�consume_statementr r r r r r r s r c @ s e Zd ZdS )� SQLMatchRuleN�r r r r r r r r & s r c @ s e Zd Zddd�Zdd� ZdS )� CursorSQLNTc C s || _ || _|| _d S r )� statement�paramsr )r r r r r r r �__init__+ s zCursorSQL.__init__c C sp |j d }| j|jks,| jd urN| j|jkrNd| _d| j| j|j|jf | _n |j �d� d| _|j sld| _d S )Nr Tz5Testing for exact SQL %s parameters %s received %s %s)� statementsr r � parametersr r �popr )r r �stmtr r r r 0 s& � ���� zCursorSQL.process_statement)NT�r r r r r r r r r r * s r c @ sF e Zd Zddd�Zdd� Zdd � Zd d � Zd d � Zdd� Zdd� Z dS )� CompiledSQLN�defaultTc C s || _ || _|| _|| _d S r )r r �dialect�enable_returning)r r r r% r& r r r r G s zCompiledSQL.__init__c C s t �dd| j�}||kS �N�[\n\t]� )�re�subr �r r �received_statementr! r r r � _compare_sqlO s zCompiledSQL._compare_sqlc C sl | j dkrTt� }d|_| jrPd |_ |_|_d|_d|_d|_ d|_ |j sPJ �|S t j �| j ��� � S d S )Nr$ T)r% r Zsupports_default_metavaluer& Zinsert_returningZupdate_returningZdelete_returningZuse_insertmanyvaluesZsupports_multivalues_insertZupdate_returning_multifromZdelete_returning_multifromZ insert_executemany_returningr �URL�create� get_dialect)r r r% r r r �_compile_dialectS s � z CompiledSQL._compile_dialectc s� |j }| �|�}d}d�d|jv r.|jd }nd}t|jt�rP|jj||d�� n |jj|||jj|jj |d�� t � ddt � ��}|j }|s�� j�d�g}n� �fdd �|D �}||fS ) zxreconstruct the statement and params in terms of a target dialect, which for CompiledSQL is just DefaultDialect.N�schema_translate_map)r% r3 )� cache_keyr% � column_keys�for_executemanyr3 r( r) ��extracted_parametersc s g | ]}� j |�d ��qS )r7 )�construct_params)�.0�m��compiledr8 r r � <listcomp>� s ��z3CompiledSQL._received_statement.<locals>.<listcomp>)�contextr2 �execution_options� isinstance� clauseelementr �compiler= r5 r6 r* r+ �strr r9 ) r r r? Zcompare_dialectr4 Zmap_�_received_statementr �_received_parametersr r< r rE m s> � ���� �z CompiledSQL._received_statementc C s� |j }| �|�\}}| �|�}| �||�}|r�|d ur�t|�}t|�}|r�|r�t|�d��} tt|��D ]8\} } | D ] } | | vs�| | | | krt qhqt|| = qFqhd}q�qF|s�|r�d}|r�d| _d | _ n| � ||�||d� | _ d S )Nr FT)r- Zreceived_parameters) r? rE � _all_paramsr. �list�dictr � enumerater r �_failure_message) r r r? rE rF r Z equivalent� all_paramsZ all_received�param�idxZreceivedZ param_keyr r r r � sF � �����z CompiledSQL.process_statementc C s> | j r6t| j �r | � |�}n| j }t|t�s2|g}|S d S d S r )r �callablerA rH )r r? r r r r rG � s zCompiledSQL._all_paramsc C s" d| j �dd�t|��dd�f S �NzzTesting for compiled statement %r partial params %s, received %%(received_statement)r with params %%(received_parameters)r�%�%%)r �replace�repr�r r Zexpected_paramsr r r rK � s ���z CompiledSQL._failure_message)Nr$ T) r r r r r. r2 rE r rG rK r r r r r# F s � 53 r# c @ s& e Zd Zd dd�Zdd� Zdd � ZdS ) �RegexSQLNr$ Fc C s2 t �| � t�|�| _|| _|| _|| _|| _d S r ) r r r* rC �regex� orig_regexr r% r& )r rW r r% r& r r r r � s zRegexSQL.__init__c C s" d| j �dd�t|��dd�f S )Nz{Testing for compiled statement ~%r partial params %s, received %%(received_statement)r with params %%(received_parameters)rrQ rR )rX rS rT rU r r r rK � s ���zRegexSQL._failure_messagec C s t | j�|��S r )�boolrW �match)r r r- r r r r. s zRegexSQL._compare_sql)Nr$ F)r r r r rK r. r r r r rV � s � rV c sD e Zd Zdd� Zdd� Z� fdd�Zdd� Zd d � Zd d � Z� Z S ) � DialectSQLc C s |j jS r )r? r% r r r r r2 s zDialectSQL._compile_dialectc C s t �dd|�}||kS r' )r* r+ )r � real_stmt� received_stmtr! r r r �_compare_no_space s z DialectSQL._compare_no_spacec sF t � �|�\}}|jD ]}| �|j|�r q:qtd| ��||jjfS )NzICan't locate compiled statement %r in list of statements actually invoked)�superrE r r^ r �AssertionErrorr? Zcompiled_parameters)r r r] Zreceived_paramsr\ �� __class__r r rE s � ��z DialectSQL._received_statementc s� |j }t�dd| j�}|�dd�}|dkr:t�dd|�}n`d }|dkrLd }n@|d krZd }n2|�d �r�t�d �� |dkrzdnd�� �fdd�}t�d||�}|�dd�}|S )Nr( r) z::z!!Zpyformatz :([\w_]+)z%(\1)sZqmark�?�formatz%s�numeric� Znumeric_dollar�$�:c s �� t � �� �S r )�next)r; ��counterZnum_identifierr r �repl5 s z4DialectSQL._dialect_adjusted_statement.<locals>.repl)� paramstyler* r+ r rS � startswith� itertools�count)r r% rm r! rl r rj r �_dialect_adjusted_statement s" z&DialectSQL._dialect_adjusted_statementc C s | � |jj�}||kS r )rq r? r% r, r r r r. ? s �zDialectSQL._compare_sqlc C s* d| � |jj��dd�t|��dd�f S rP )rq r? r% rS rT rU r r r rK E s �����zDialectSQL._failure_message) r r r r2 r^ rE rq r. rK � __classcell__r r ra r r[ s r[ c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CountStatementsc C s || _ d| _d S �Nr �rp �_statement_count)r rp r r r r T s zCountStatements.__init__c C s | j d7 _ d S )Nrf )rv r r r r r X s z!CountStatements.process_statementc C s( | j | jkr$ds$J d| j | jf ��d S )NFz,desired statement count %d does not match %dru r r r r r [ s �z"CountStatements.no_more_statementsN)r r r r r r r r r r rs S s rs c @ s e Zd Zdd� Zdd� ZdS )�AllOfc G s t |�| _d S r )�set�rules�r ry r r r r d s zAllOf.__init__c C sj t | j�D ]H}d |_|�|� |jrB| j�|� | js<d| _ qfq |js d | _ qfq t | j�d j| _d S �NTr )rH ry r r r �discard�r r �ruler r r r g s zAllOf.process_statementNr" r r r r rw c s rw c s, e Zd Zdd� Zdd� Z� fdd�Z� ZS )�EachOfc G s t |�| _d S r )rH ry rz r r r r y s zEachOf.__init__c C sh | j sd| _d| _| j rX| j d }|�|� |jr@| j �d� n|jrN|j| _|jrqXq| j sdd| _d S )NTFr )ry r r r r r r} r r r r | s zEachOf.process_statementc s6 | j r"| j d js"| j d �� n| j r2t� �� d S rt )ry r r r_ r ra r r r � s zEachOf.no_more_statements)r r r r r r rr r r ra r r x s r c s e Zd Z� fdd�Z� ZS )� Conditionalc s" |rt � j|� n t � j|� d S r )r_ r )r � conditionry Z else_rulesra r r r � s zConditional.__init__)r r r r rr r r ra r r� � s r� c @ s e Zd Zdd� ZdS )�Orc C s< | j D ] }|�|� |jrd| _ q8qt| j �d j| _d S r{ )ry r r rH r r} r r r r � s zOr.process_statementN)r r r r r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�SQLExecuteObservedc C s6 || _ || _|r|| _n|r&|g| _ng | _g | _d S r )r? rB r r )r r? rB � multiparamsr r r r r � s zSQLExecuteObserved.__init__c C s t | j�S r )rD r r r r r �__repr__� s zSQLExecuteObserved.__repr__N)r r r r r� r r r r r� � s r� c @ s e Zd ZdS )�SQLCursorExecuteObservedNr r r r r r� � s r� )r r r? � executemanyc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SQLAsserterc C s g | _ d S r )� accumulatedr r r r r � s zSQLAsserter.__init__c C s | j | _| ` d S r )r� �_finalr r r r �_close� s zSQLAsserter._closec G sr t |� }t| j�}|rJ|�d�}|�|� |jr4qJq|jrdsJ |j��q|r`dsnJ d| ��n|jsn|�� d S )Nr Fz$Additional SQL statements remain: %s)r rH r� r r r r r )r ry r~ Zobservedr r r r �assert_� s zSQLAsserter.assert_N)r r r r r� r� r r r r r� � s r� c # s� t � � g �t�| d��fdd��}t�| d�� �fdd��}z.� V W t�| d|� t�| d|� � �� n&t�| d|� t�| d|� � �� 0 d S )NZbefore_executec s |||f� d d �< d S r r )�connrB r� r r@ )�origr r �connection_execute� s z)assert_engine.<locals>.connection_executeZafter_cursor_executec sj |sd S � j r*� j d j|u r*� j d }n&t|�d �d �d �}� j �|� |j�t||||�� d S )N���r rf r )r� r? r� �appendr r� )r� �cursorr r r? r� Zobs�Zasserterr� r r �cursor_execute� s �� ��z%assert_engine.<locals>.cursor_execute)r� r Z listens_for�remover� )�enginer� r� r r� r � assert_engine� s �r� ) � __future__r � collections� contextlibro r* r) r r� r Zengine.defaultr Zschemar r r r r# rV r[ rs rw r r� r� r� � namedtupler� r� �contextmanagerr� r r r r �<module> s< (L ��
| ver. 1.6 |
Github
|
.
| PHP 8.2.30 | ??????????? ?????????: 0 |
proxy
|
phpinfo
|
???????????