U
    d/e
                     @   sT   G d d de ZG dd de ZG dd de ZG dd de ZG dd	 d	e Zd
S )c                   @   sD   e Zd ZdZdZdZdZdZdZdZ	dd	d
dddgZ
edd ZdS )ClientPacketTypesz,
    Packet types that client transmits
                       HelloZQueryDataZCancelZPingZTablesStatusRequestc                 C   s   |dkrdS | j | S )Nr   Unknown packet
_types_strclspacket r   >/tmp/pip-unpacked-wheel-mds_o03w/clickhouse_driver/protocol.pyto_str   s    zClientPacketTypes.to_strN)__name__
__module____qualname____doc__HELLOZQUERYDATAZCANCELZPINGZTABLES_STATUS_REQUESTr   classmethodr   r   r   r   r   r      s         r   c                   @   s   e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdddddddddddddddgZed d! Zed"d# Zd$S )%ServerPacketTypesz-
    Packet types that server transmits.
    r   r   r   r   r   r            	   
               r   r	   	ExceptionProgressZPongZEndOfStreamZProfileInfoZTotalsZExtremesZTablesStatusResponseLogZTableColumnsZ	PartUUIDsZReadTaskRequestZProfileEventsc                 C   s   |dkrdS | j | S )Nr#   r
   r   r   r   r   r   r   Z   s    zServerPacketTypes.to_strc                 C   s   || j krdS dS )Nr   r   )TABLE_COLUMNSr   r   r   r   strings_in_message^   s    
z$ServerPacketTypes.strings_in_messageN)r   r   r   r   r   r   Z	EXCEPTIONZPROGRESSZPONGZEND_OF_STREAMZPROFILE_INFOZTOTALSZEXTREMESZTABLES_STATUS_RESPONSEZLOGr'   Z
PART_UUIDSZREAD_TASK_REQUESTZPROFILE_EVENTSr   r   r   r(   r   r   r   r   r   #   sH               
r   c                   @   s   e Zd ZdZdZdS )Compressionr   r   N)r   r   r   ZDISABLEDZENABLEDr   r   r   r   r)   e   s   r)   c                   @   s   e Zd ZdZdZdZdS )CompressionMethodr   r   r   N)r   r   r   LZ4ZLZ4HCZSTDr   r   r   r   r*   j   s   r*   c                   @   s   e Zd ZdZdZdS )CompressionMethodByte      N)r   r   r   r+   r,   r   r   r   r   r-   p   s   r-   N)objectr   r   r)   r*   r-   r   r   r   r   <module>   s   !B