관리-도구
편집 파일: mock.cpython-38.pyc
U -?�fS � @ 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� d�Z 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 ddd�dd�Zeeedd� ��Zded < eeee d ���Z ded < eeed d� ��Z d ed< dd d�dd�Z ddd�dd�Z dd d�dd�Zdd 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 �F/opt/hc_python/lib64/python3.8/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 zUnion[str, URL]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. $