�
g\�gU � �. � 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 ej r<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 d� d� � Z d!d �Z dS )"� )� annotations)�
attrgetterN)�Any)�Callable)�cast)�Optional)�Type)�Union� )�url� )�util)�Engine)�_CoreAnyExecuteParams)�CoreExecuteOptionsParameter)�Dialect)�URL)�
Executable)�
SchemaDropper)�SchemaGenerator)�
HasSchemaAttr)�
SchemaItemc �� � e Zd ZU d$d�Z ee ed� � � � � Zded< ee e e d � � � � � � Z
ded< ee ed
� � � � � Z d ed < d%d�Z d&d�Z
d'd�Zd(d �Z d)d*d#�Zd S )+�MockConnection�dialectr �execute�Callable[..., Any]c �" � || _ || _ d S �N)�_dialect�
_execute_impl)�selfr r s �f/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/sqlalchemy/engine/mock.py�__init__zMockConnection.__init__% s � � ��
� $����� c � � | S r � ��ss r# �<lambda>zMockConnection.<lambda>) s � �!� r% r �enginer c � � | j j S r )r �namer( s r# r* zMockConnection.<lambda>+ s
� �Q�Z�_� r% �strr- �kwargsr �returnc � � | S r r' )r" r/ s r# �connectzMockConnection.connect- � � �� r% �objr �
Optional[str]c � � |j S r )�schema)r" r4 s r# �schema_for_objectz MockConnection.schema_for_object0 s
� ��z�r% �kwc � � | S r r' )r" r9 s r# �execution_optionsz MockConnection.execution_options3 r3 r% �visitorcallable�+Type[Union[SchemaGenerator, SchemaDropper]]�elementr �Nonec �T � d|d< || j | fi |��� |� � d S )NF�
checkfirst)r �traverse_single)r" r< r> r/ s r# �_run_ddl_visitorz MockConnection._run_ddl_visitor6 s> � � %��|� ���� �d�5�5�f�5�5�E�E�g�N�N�N�N�Nr% Nr �
parameters� Optional[_CoreAnyExecuteParams]r; �%Optional[CoreExecuteOptionsParameter]c �. � | � ||� � S r )r! )r" r4 rD r; s r# r zMockConnection.execute? s � � �!�!�#�z�2�2�2r% )r r r r )r/ r r0 r )r4 r r0 r5 )r9 r r0 r )r< r= r> r r/ r r0 r? )NN)r4 r rD rE r; rF r0 r )�__name__�
__module__� __qualname__r$ r r �propertyr+ �__annotations__r r r- r2 r8 r; rC r r' r% r# r r $ s& � � � � � � �%� %� %� %� �T�#�x�x� � � 4� 4�5�5�F�5�5�5�5��t�C���*�*�Z�*@�*@�!A�!A�B�B�G�B�B�B�B���S�(�(�#<�#<�=�=�>�>�D�>�>�>�>�� � � �� � � �� � � �O� O� O� O� 7;�CG� 3� 3� 3� 3� 3� 3� 3r% r r �Union[str, URL]�executorr r9 r0 c �� � t j | � � }|� � � }i }t j |� � D ] }||v r|� |� � ||< � |di |��}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`
r' )�_url�make_url� get_dialectr �get_cls_kwargs�popr )r rN r9 �u� dialect_cls� dialect_args�kr s r# �create_mock_enginerY H s� � �`
�
�c���A��-�-�/�/�K��L�
�
��
-�
-� (� (�� ��7�7� �f�f�Q�i�i�L��O�� �k�)�)�L�)�)�G� �'�8� ,� ,�,r% )r rM rN r r9 r r0 r )!�
__future__r �operatorr �typingr r r r r r
� r rP r �
TYPE_CHECKING�baser �
interfacesr r r r �sql.baser �sql.ddlr r �
sql.schemar r r rY r' r% r# �<module>rd s� �� #� "� "� "� "� "� � � � � � �
�
�
�
� � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � �
��
(� � � � � � �1�1�1�1�1�1�7�7�7�7�7�7�#�#�#�#�#�#�������%�%�%�%�%�%�'�'�'�'�'�'�)�)�)�)�)�)�*�*�*�*�*�*�'�'�'�'�'�'�!3� !3� !3� !3� !3� !3� !3� !3�H=-� =-� =-� =-� =-� =-r%