o
    EDi3                     @   sB   d dl mZ d dlmZ d dlmZ d dlmZ G dd dZdS )    )Union)Cursor)create_client)QueryResultc                   @   s   e Zd ZdZ								ddededed	ed
edededeeef fddZdd Z	dd Z
dd ZdefddZdedefddZdd ZdS ) 
Connectionz6
    See :ref:`https://peps.python.org/pep-0249/`
    N r   Fdsnusernamepasswordhostdatabase	interfaceportsecurec	           
      K   s6   t |||||||||	d	| _| jd | jj| _d S )N)	r   r	   r
   r   r   r   r   r   generic_args
sqlalchemy)r   client_add_integration_tag	server_tztimezone)
selfr   r	   r
   r   r   r   r   r   kwargs r   \/var/www/Datamplify/venv/lib/python3.10/site-packages/clickhouse_connect/dbapi/connection.py__init__   s   

zConnection.__init__c                 C   s   | j   d S N)r   closer   r   r   r   r   $   s   zConnection.closec                 C      d S r   r   r   r   r   r   commit'      zConnection.commitc                 C   r   r   r   r   r   r   r   rollback*   r    zConnection.rollbackcmdc                 C      | j |S r   )r   command)r   r"   r   r   r   r$   -      zConnection.commandqueryreturnc                 C   r#   r   )r   r&   )r   r&   r   r   r   	raw_query0   r%   zConnection.raw_queryc                 C   s
   t | jS r   )r   r   r   r   r   r   cursor3   s   
zConnection.cursor)Nr   r   NNNr   F)__name__
__module____qualname____doc__strintr   boolr   r   r   r!   r$   r   r(   r)   r   r   r   r   r      sB    

r   N)	typingr   clickhouse_connect.dbapi.cursorr   clickhouse_connect.driverr   clickhouse_connect.driver.queryr   r   r   r   r   r   <module>   s
    