U
|
¸ý°dÒ ã @ s$ d dl mZ d dlmZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
|
Z
|
d
|
d l m Z d d l mZ ejrþd
|
dlmZ d
|
dlmZ d
|
dlmZ d
|
dlmZ d
|
dl mZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z G dd dZ ddddd d d Z dS ) é )Ú annotations)Ú
|
attrgetterN)ÚAny)ÚCallable)Úcast)ÚOptional)ÚType)ÚUnioné )Úurlé )Úutil)ÚEngine)Ú_CoreAnyExecuteParams)ÚCoreExecuteOptionsParameter)ÚDialect)ÚURL)Ú
|
Executable)Ú SchemaDropper)ÚSchemaGenerator)Ú HasSchemaAttr)Ú
|
SchemaItemc @ sÆ e Zd ZU dddddZeeedd Zded < eeee d
|
Z
|
ded < eeed d Z d ed< dd dddZ dddddZ dd dddZdd dd d d d Zd(d"d#d$dd%d&d'Zd!S ))ÚMockConnectionr zCallable[..., Any])ÚdialectÚexecutec C s || _ || _d S ©N)Ú_dialectÚ _execute_impl)Úselfr r © r úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/engine/mock.pyÚ__init__% s zMockConnection.__init__c C s | S r r ©Úsr r r Ú<lambda>) ó zMockConnection.<lambda>r Úenginer r c C s | j jS r )r Únamer" r r r r$ + r% Ústrr' r )ÚkwargsÚreturnc K s | S r r )r r) r r r Úconnect- s zMockConnection.connectr z Optional[str])Úobjr* c C s |j S r )Zschema)r r, r r r Úschema_for_object0 s z MockConnection.schema_for_object)Úkwr* c K s | S r r )r r. r r r Úexecution_options3 s z MockConnection.execution_optionsz+Type[Union[SchemaGenerator, SchemaDropper]]r ÚNone)ÚvisitorcallableÚelementr) r* c K s" d|d< || j | f| |¡ d S )NFZ
|
checkfirst)r Ztraverse_single)r r1 r2 r) r r r Ú_run_ddl_visitor6 s z MockConnection._run_ddl_visitorNr z Optional[_CoreAnyExecuteParams]z%Optional[CoreExecuteOptionsParameter])r, Ú
|
parametersr/ r* c C s | ||¡S r )r )r r, r4 r/ r r r r ? s zMockConnection.execute)NN)Ú__name__Ú
|
__module__Ú __qualname__r! r r Úpropertyr& Ú__annotations__r r r' r+ r- r/ r3 r r r r r r $ s
|
ür r r )r Úexecutorr. r* c K sP t | ¡}| ¡ }i }t |¡D ]}||kr | |¡||< q |f |}t||S )a= Create a "mock" engine used for echoing DDL.
|
|
This is a utility function used for debugging or storing the output of DDL
|
sequences as generated by :meth:`_schema.MetaData.create_all`
|
and related methods.
|
|
The function accepts a URL which is used only to determine the kind of
|
dialect to be used, as well as an "executor" callable function which
|
will receive a SQL expression object and parameters, which can then be
|
echoed or otherwise printed. The executor's return value is not handled,
|
nor does the engine allow regular string statements to be invoked, and
|
is therefore only useful for DDL that is sent to the database without
|
receiving any results.
|
|
E.g.::
|
|
from sqlalchemy import create_mock_engine
|
|
def dump(sql, *multiparams, **params):
|
print(sql.compile(dialect=engine.dialect))
|
|
engine = create_mock_engine('postgresql+psycopg2://', dump)
|
metadata.create_all(engine, checkfirst=False)
|
|
:param url: A string URL which typically needs to contain only the
|
database backend name.
|
|
:param executor: a callable which receives the arguments ``sql``,
|
``*multiparams`` and ``**params``. The ``sql`` parameter is typically
|
an instance of :class:`.ExecutableDDLElement`, which can then be compiled
|
into a string using :meth:`.ExecutableDDLElement.compile`.
|
|
.. versionadded:: 1.4 - the :func:`.create_mock_engine` function replaces
|
the previous "mock" engine strategy used with
|
:func:`_sa.create_engine`.
|
|
.. seealso::
|
|
:ref:`faq_ddl_as_string`
|
|
)Ú_urlZmake_urlÚ get_dialectr Zget_cls_kwargsÚpopr )r r: r. ÚuZ dialect_clsZ dialect_argsÚkr r r r Úcreate_mock_engineH s ,
|
|
r@ )!Ú
|
__future__r Úoperatorr Útypingr r r r r r Ú r r; r Ú TYPE_CHECKINGÚbaser Z
|
interfacesr r r r Zsql.baser Zsql.ddlr r Z
|
sql.schemar r r r@ r r r r Ú<module> s. $
|
