clickhouse-driver 0.2.1__cp39-cp39-win_amd64.whl → 0.2.10__cp39-cp39-win_amd64.whl

clickhouse_driver/protocol.py

@@ -1,105 +1,130 @@
-
-class ClientPacketTypes(object):
-    """
-    Packet types that client transmits
-    """
-    # Name, version, revision, default DB
-    HELLO = 0
-
-    # Query id, query settings, stage up to which the query must be executed,
-    # whether the compression must be used, query text
-    # (without data for INSERTs).
-    QUERY = 1
-
-    # A block of data (compressed or not).
-    DATA = 2
-
-    # Cancel the query execution.
-    CANCEL = 3
-
-    # Check that connection to the server is alive.
-    PING = 4
-
-    # Check status of tables on the server.
-    TABLES_STATUS_REQUEST = 5
-
-    _types_str = [
-        'Hello', 'Query', 'Data', 'Cancel', 'Ping', 'TablesStatusRequest'
-    ]
-
-    @classmethod
-    def to_str(cls, packet):
-        return 'Unknown packet' if packet > 5 else cls._types_str[packet]
-
-
-class ServerPacketTypes(object):
-    """
-    Packet types that server transmits.
-    """
-    # Name, version, revision.
-    HELLO = 0
-
-    # A block of data (compressed or not).
-    DATA = 1
-
-    # The exception during query execution.
-    EXCEPTION = 2
-
-    # Query execution progress: rows read, bytes read.
-    PROGRESS = 3
-
-    # Ping response
-    PONG = 4
-
-    # All packets were transmitted
-    END_OF_STREAM = 5
-
-    # Packet with profiling info.
-    PROFILE_INFO = 6
-
-    # A block with totals (compressed or not).
-    TOTALS = 7
-
-    # A block with minimums and maximums (compressed or not).
-    EXTREMES = 8
-
-    # A response to TablesStatus request.
-    TABLES_STATUS_RESPONSE = 9
-
-    # System logs of the query execution
-    LOG = 10
-
-    # Columns' description for default values calculation
-    TABLE_COLUMNS = 11
-
-    _types_str = [
-        'Hello', 'Data', 'Exception', 'Progress', 'Pong', 'EndOfStream',
-        'ProfileInfo', 'Totals', 'Extremes', 'TablesStatusResponse', 'Log',
-        'TableColumns'
-    ]
-
-    @classmethod
-    def to_str(cls, packet):
-        return 'Unknown packet' if packet > 11 else cls._types_str[packet]
-
-    @classmethod
-    def strings_in_message(cls, packet):
-        if packet == cls.TABLE_COLUMNS:
-            return 2
-        return 0
-
-
-class Compression(object):
-    DISABLED = 0
-    ENABLED = 1
-
-
-class CompressionMethod(object):
-    LZ4 = 1
-    LZ4HC = 2
-    ZSTD = 3
-
-
-class CompressionMethodByte(object):
-    LZ4 = 0x82
-    ZSTD = 0x90
+
+class ClientPacketTypes(object):
+    """
+    Packet types that client transmits
+    """
+    # Name, version, revision, default DB
+    HELLO = 0
+
+    # Query id, query settings, stage up to which the query must be executed,
+    # whether the compression must be used, query text
+    # (without data for INSERTs).
+    QUERY = 1
+
+    # A block of data (compressed or not).
+    DATA = 2
+
+    # Cancel the query execution.
+    CANCEL = 3
+
+    # Check that connection to the server is alive.
+    PING = 4
+
+    # Check status of tables on the server.
+    TABLES_STATUS_REQUEST = 5
+
+    _types_str = [
+        'Hello', 'Query', 'Data', 'Cancel', 'Ping', 'TablesStatusRequest'
+    ]
+
+    @classmethod
+    def to_str(cls, packet):
+        try:
+            return cls._types_str[packet]
+        except IndexError:
+            return 'Unknown packet'
+
+
+class ServerPacketTypes(object):
+    """
+    Packet types that server transmits.
+    """
+    # Name, version, revision.
+    HELLO = 0
+
+    # A block of data (compressed or not).
+    DATA = 1
+
+    # The exception during query execution.
+    EXCEPTION = 2
+
+    # Query execution progress: rows read, bytes read.
+    PROGRESS = 3
+
+    # Ping response
+    PONG = 4
+
+    # All packets were transmitted
+    END_OF_STREAM = 5
+
+    # Packet with profiling info.
+    PROFILE_INFO = 6
+
+    # A block with totals (compressed or not).
+    TOTALS = 7
+
+    # A block with minimums and maximums (compressed or not).
+    EXTREMES = 8
+
+    # A response to TablesStatus request.
+    TABLES_STATUS_RESPONSE = 9
+
+    # System logs of the query execution
+    LOG = 10
+
+    # Columns' description for default values calculation
+    TABLE_COLUMNS = 11
+
+    # List of unique parts ids.
+    PART_UUIDS = 12
+
+    # String (UUID) describes a request for which next task is needed
+    READ_TASK_REQUEST = 13
+
+    # Packet with profile events from server.
+    PROFILE_EVENTS = 14
+
+    MERGE_TREE_ALL_RANGES_ANNOUNCEMENT = 15
+
+    # Request from a MergeTree replica to a coordinator
+    MERGE_TREE_READ_TASK_REQUEST = 16
+
+    # Receive server's (session-wide) default timezone
+    TIMEZONE_UPDATE = 17
+
+    _types_str = [
+        'Hello', 'Data', 'Exception', 'Progress', 'Pong', 'EndOfStream',
+        'ProfileInfo', 'Totals', 'Extremes', 'TablesStatusResponse', 'Log',
+        'TableColumns', 'PartUUIDs', 'ReadTaskRequest', 'ProfileEvents',
+        'MergeTreeAllRangesAnnouncement', 'MergeTreeReadTaskRequest',
+        'TimezoneUpdate'
+    ]
+
+    @classmethod
+    def to_str(cls, packet):
+        try:
+            return cls._types_str[packet]
+        except IndexError:
+            return 'Unknown packet'
+
+    @classmethod
+    def strings_in_message(cls, packet):
+        if packet == cls.TABLE_COLUMNS:
+            return 2
+        return 0
+
+
+class Compression(object):
+    DISABLED = 0
+    ENABLED = 1
+
+
+class CompressionMethod(object):
+    LZ4 = 1
+    LZ4HC = 2
+    ZSTD = 3
+
+
+class CompressionMethodByte(object):
+    LZ4 = 0x82
+    ZSTD = 0x90

clickhouse_driver/queryprocessingstage.py

@@ -1,8 +1,8 @@
-
-class QueryProcessingStage(object):
-    """
-    Determines till which state SELECT query should be executed.
-    """
-    FETCH_COLUMNS = 0
-    WITH_MERGEABLE_STATE = 1
-    COMPLETE = 2
+
+class QueryProcessingStage(object):
+    """
+    Determines till which state SELECT query should be executed.
+    """
+    FETCH_COLUMNS = 0
+    WITH_MERGEABLE_STATE = 1
+    COMPLETE = 2

clickhouse_driver/reader.py

@@ -1,69 +1,69 @@
-from struct import Struct
-
-from .varint import read_varint
-
-
-def read_binary_str(buf):
-    length = read_varint(buf)
-    return read_binary_str_fixed_len(buf, length)
-
-
-def read_binary_bytes(buf):
-    length = read_varint(buf)
-    return read_binary_bytes_fixed_len(buf, length)
-
-
-def read_binary_str_fixed_len(buf, length):
-    return read_binary_bytes_fixed_len(buf, length).decode('utf-8')
-
-
-def read_binary_bytes_fixed_len(buf, length):
-    return buf.read(length)
-
-
-def read_binary_int(buf, fmt):
-    """
-    Reads int from buffer with provided format.
-    """
-    # Little endian.
-    s = Struct('<' + fmt)
-    return s.unpack(buf.read(s.size))[0]
-
-
-def read_binary_int8(buf):
-    return read_binary_int(buf, 'b')
-
-
-def read_binary_int16(buf):
-    return read_binary_int(buf, 'h')
-
-
-def read_binary_int32(buf):
-    return read_binary_int(buf, 'i')
-
-
-def read_binary_int64(buf):
-    return read_binary_int(buf, 'q')
-
-
-def read_binary_uint8(buf):
-    return read_binary_int(buf, 'B')
-
-
-def read_binary_uint16(buf):
-    return read_binary_int(buf, 'H')
-
-
-def read_binary_uint32(buf):
-    return read_binary_int(buf, 'I')
-
-
-def read_binary_uint64(buf):
-    return read_binary_int(buf, 'Q')
-
-
-def read_binary_uint128(buf):
-    hi = read_binary_int(buf, 'Q')
-    lo = read_binary_int(buf, 'Q')
-
-    return (hi << 64) + lo
+from struct import Struct
+
+from .varint import read_varint
+
+
+def read_binary_str(buf):
+    length = read_varint(buf)
+    return read_binary_str_fixed_len(buf, length)
+
+
+def read_binary_bytes(buf):
+    length = read_varint(buf)
+    return read_binary_bytes_fixed_len(buf, length)
+
+
+def read_binary_str_fixed_len(buf, length):
+    return read_binary_bytes_fixed_len(buf, length).decode('utf-8')
+
+
+def read_binary_bytes_fixed_len(buf, length):
+    return buf.read(length)
+
+
+def read_binary_int(buf, fmt):
+    """
+    Reads int from buffer with provided format.
+    """
+    # Little endian.
+    s = Struct('<' + fmt)
+    return s.unpack(buf.read(s.size))[0]
+
+
+def read_binary_int8(buf):
+    return read_binary_int(buf, 'b')
+
+
+def read_binary_int16(buf):
+    return read_binary_int(buf, 'h')
+
+
+def read_binary_int32(buf):
+    return read_binary_int(buf, 'i')
+
+
+def read_binary_int64(buf):
+    return read_binary_int(buf, 'q')
+
+
+def read_binary_uint8(buf):
+    return read_binary_int(buf, 'B')
+
+
+def read_binary_uint16(buf):
+    return read_binary_int(buf, 'H')
+
+
+def read_binary_uint32(buf):
+    return read_binary_int(buf, 'I')
+
+
+def read_binary_uint64(buf):
+    return read_binary_int(buf, 'Q')
+
+
+def read_binary_uint128(buf):
+    hi = read_binary_int(buf, 'Q')
+    lo = read_binary_int(buf, 'Q')
+
+    return (hi << 64) + lo

clickhouse_driver/readhelpers.py

@@ -1,26 +1,26 @@
-from .errors import ServerException
-from .reader import read_binary_str, read_binary_uint8, read_binary_int32
-
-
-def read_exception(buf, additional_message=None):
-    code = read_binary_int32(buf)
-    name = read_binary_str(buf)
-    message = read_binary_str(buf)
-    stack_trace = read_binary_str(buf)
-    has_nested = bool(read_binary_uint8(buf))
-
-    new_message = ''
-
-    if additional_message:
-        new_message += additional_message + '. '
-
-    if name != 'DB::Exception':
-        new_message += name + ". "
-
-    new_message += message + ". Stack trace:\n\n" + stack_trace
-
-    nested = None
-    if has_nested:
-        nested = read_exception(buf)
-
-    return ServerException(new_message, code, nested=nested)
+from .errors import ServerException
+from .reader import read_binary_str, read_binary_uint8, read_binary_int32
+
+
+def read_exception(buf, additional_message=None):
+    code = read_binary_int32(buf)
+    name = read_binary_str(buf)
+    message = read_binary_str(buf)
+    stack_trace = read_binary_str(buf)
+    has_nested = bool(read_binary_uint8(buf))
+
+    new_message = ''
+
+    if additional_message:
+        new_message += additional_message + '. '
+
+    if name != 'DB::Exception':
+        new_message += name + ". "
+
+    new_message += message + ". Stack trace:\n\n" + stack_trace
+
+    nested = None
+    if has_nested:
+        nested = read_exception(buf)
+
+    return ServerException(new_message, code, nested=nested)