o
    EDi                     @   s6   d dl mZ d dlmZ d dlmZ G dd dZdS )    )Optional)get_from_name)QueryResultc                   @   s`   e Zd Zi Zddee fddZedefddZ	defdd	Z
defd
dZdefddZdS )QuerySummaryNsummaryc                 C   s   |d ur	|| _ d S d S N)r   )selfr    r	   Z/var/www/Datamplify/venv/lib/python3.10/site-packages/clickhouse_connect/driver/summary.py__init__   s   
zQuerySummary.__init__returnc                 C      t | jddS )Nwritten_rowsr   intr   getr   r	   r	   r
   r      s   zQuerySummary.written_rowsc                 C   r   )Nwritten_bytesr   r   r   r	   r	   r
   r      s   zQuerySummary.written_bytesc                 C   s   | j ddS )Nquery_id )r   r   r   r	   r	   r
   r      s   zQuerySummary.query_idc                 C   s   g }g }g }t d}t d}| j D ]$\}}|| | r-|t| || q|| || qt|gt|t|dS )NStringInt64)column_namescolumn_types)r   r   itemsappend	isnumericr   r   tuple)r   datar   r   str_typeint_typekeyvaluer	   r	   r
   as_query_result   s   

zQuerySummary.as_query_resultr   )__name__
__module____qualname__r   r   dictr   propertyr   r   r   strr   r   r#   r	   r	   r	   r
   r      s    r   N)typingr   %clickhouse_connect.datatypes.registryr   clickhouse_connect.driver.queryr   r   r	   r	   r	   r
   <module>   s    