U
|
¸ý°dC ã
|
@ sX d dl mZ d dlZd dlZd dlZd dlZddlmZ ddlm Z ddl
|
m Z ddl m Z G dd d ZG d
|
d d eZG d d d eZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG d d d eZG d d d ZG d d! d!e d!d"d#d$d%g¡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
|
úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/assertsql.pyÚprocess_statement s z AssertRule.process_statementc C s ds t dd S )NFz?All statements are complete, but pending assertion rules remain)ÚAssertionError©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 ZdddZdd ZdS )Ú CursorSQLNTc C s || _ || _|| _d S r )Ú statementÚparamsr )r r r r r
|
r
|
r Ú__init__, s zCursorSQL.__init__c C sj |j d }| j|jks,| jd k rH| j|jkrHd| j| j|j|jf | _n |j d¡ d| _|j sfd| _d S )Nr z5Testing for exact SQL %s parameters %s received %s %sT)Ú
|
statementsr r Ú
|
parametersr Úpopr r )r r Ústmtr
|
r
|
r r 1 s$
|
ÿ
|
ÿüÿÿ
|
zCursorSQL.process_statement)NT©r r r r r r
|
r
|
r
|
r r + s
|
r c @ sF e Zd ZdddZdd 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 sPt |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 r ÚURLÚcreateÚ get_dialect)r r r& r
|
r
|
r Ú_compile_dialectS s
|
þ
|
z CompiledSQL._compile_dialectc s¶ |j }| |¡}d}dd|jkr.|jd }nd}t|jtrP|jj||d n |jj|||jj|jj |d t
|
ddt ¡}|j }|s jdg}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& r4 )Ú cache_keyr& Ú column_keysÚfor_executemanyr4 r) r* ©Úextracted_parametersc s g | ]} j |d qS )r8 )Úconstruct_params)Ú.0Úm©Úcompiledr9 r
|
r Ú
|
<listcomp> s
|
ý ÿz3CompiledSQL._received_statement.<locals>.<listcomp>)Úcontextr3 Úexecution_optionsÚ
|
isinstanceÚ clauseelementr Úcompiler> r6 r7 r+ r, Ústrr r: ) r r r@ Zcompare_dialectr5 Zmap_Ú_received_statementr Ú_received_parametersr
|
r= r rF m s>
|
ÿ
|
þûÿÿ üz CompiledSQL._received_statementc C sä |j }| |¡\}}| |¡}| ||¡}|r¶|d k r¶t|}t|}|rª|rªt| d¡} tt|D ]8\}
|
} | D ] } | | ks| | | | krt qhqt||
|
= qFqhd}qªqF|s²|r¶d}|rÈd| _d | _ n|
|
||¡||d | _ d S )Nr FT)r. Zreceived_parameters) r@ rF Ú _all_paramsr/ ÚlistÚdictr! Ú enumerater r Ú_failure_message) r r r@ rF rG r Z
|
equivalentÚ
|
all_paramsZ all_receivedÚparamÚidxZreceivedZ param_keyr
|
r
|
r r £ sH ÿ
|
ÿþ ÿþþz CompiledSQL.process_statementc C s> | j r6t| j r | |¡}n| j }t|ts2|g}|S d S d S r )r ÚcallablerB rI )r r@ r r
|
r
|
r rH Ö 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 rL â s þýÿz CompiledSQL._failure_message)Nr% T)
|
r r r r r/ r3 rF r rH rL r
|
r
|
r
|
r r$ F s ÿ
|
63 r$ c @ s& e Zd Zd
|
ddZdd Zdd ZdS ) ÚRegexSQLNr% Fc C s2 t | ¡ t |¡| _|| _|| _|| _|| _d S r ) r r r+ rD ÚregexÚ
|
orig_regexr r& r' )r rX 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)rrR rS )rY rT rU rV r
|
r
|
r rL ù s þýÿzRegexSQL._failure_messagec C s t | j |¡S r )ÚboolrX Úmatch)r r r. r
|
r
|
r r/ s zRegexSQL._compare_sql)Nr% F)r r r r rL r/ r
|
r
|
r
|
r rW î s ÿ
|
|
rW c sD e Zd Zdd Zdd Z fddZdd Zd d
|
Zd d Z Z S ) Ú
|
DialectSQLc C s |j jS r )r@ r& r r
|
r
|
r r3 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)ÚsuperrF r r_ r r r@ Zcompiled_parameters)r r r^ Zreceived_paramsr] ©Ú __class__r
|
r rF s ÿ
|
ÿÿz DialectSQL._received_statementc sª |j }t dd| j¡}| dd¡}|dkr:t dd|¡}n`d }|dkrLd }n@|d
|
krZd }n2| d ¡rt 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 Úrepl6 s z4DialectSQL._dialect_adjusted_statement.<locals>.repl)Ú
|
paramstyler+ r, r rT Ú
|
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 rQ )rq r@ r& rT rU rV r
|
r
|
r rL F s ÿ þüýÿzDialectSQL._failure_message)
|
r r r r3 r_ rF rq r/ rL Ú __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 U s zCountStatements.__init__c C s | j d7 _ d S )Nrf )ru r r
|
r
|
r r Y s z!CountStatements.process_statementc C s( | j | jkr$ds$td| j | jf d S )NFz,desired statement count %d does not match %d)rp ru r r r
|
r
|
r r \ s
|
þz"CountStatements.no_more_statementsN)r r r r r r r
|
r
|
r
|
r rs T s rs c @ s e Zd Zdd Zdd ZdS )ÚAllOfc G s t || _d S r )ÚsetÚrules©r rx r
|
r
|
r r e s zAllOf.__init__c C sj t | jD ]H}d |_| |¡ |jrB| j |¡ | js<d| _ qfq
|
|js
|
d | _ qfq
|
t | jd j| _d S ©NTr )rI rx r r r Údiscard©r r Úruler
|
r
|
r r h s
|
zAllOf.process_statementNr# r
|
r
|
r
|
r rv d s rv c s, e Zd Zdd Zdd Z fddZ ZS )ÚEachOfc G s t || _d S r )rI rx ry r
|
r
|
r r z 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 )rx 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 )rx 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~ y s r~ c s e Zd Z fddZ ZS )Ú Conditionalc s" |rt j| n t j| d S r )r` r )r Ú conditionrx 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 rz )rx r r rI 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@ rC r r )r r@ rC Ú multiparamsr r
|
r
|
r r ª s
|
zSQLExecuteObserved.__init__c C s
|
t | jS r )rE 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|jrdst|jq|r`dsntd| n|jsn| ¡ d S )Nr Fz$Additional SQL statements remain:
|
%s) r~ rI r r! r r r r r )r rx r} Zobservedr r
|
r
|
r Úassert_Ë s
|
|
|
zSQLAsserter.assert_N)r r r r r r r
|
r
|
r
|
r r à s r c # sr t g t | d¡fdd}t | d¡ fdd}z
|
V W 5 t | d|¡ t | d|¡ ¡ X d S )NZbefore_executec s |||f d d
< d S r r
|
)ÚconnrC r r rA )Úorigr
|
r Úconnection_executeâ s z)assert_engine.<locals>.connection_executeZafter_cursor_executec sj |sd S j r* j d j|kr* 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$ rW r\ rs rv r~ r r r Ú
|
namedtupler
r Úcontextmanagerr r
|
r
|
r
|
r Ú<module>
|
s< )L
|
|
þÿ
|
