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

clickhouse_driver/dbapi/errors.py

@@ -1,40 +1,40 @@
-
-
-class Warning(Exception):
-    pass
-
-
-class Error(Exception):
-    pass
-
-
-class InterfaceError(Error):
-    pass
-
-
-class DatabaseError(Error):
-    pass
-
-
-class InternalError(DatabaseError):
-    pass
-
-
-class OperationalError(DatabaseError):
-    pass
-
-
-class ProgrammingError(DatabaseError):
-    pass
-
-
-class IntegrityError(DatabaseError):
-    pass
-
-
-class DataError(DatabaseError):
-    pass
-
-
-class NotSupportedError(DatabaseError):
-    pass
+
+
+class Warning(Exception):
+    pass
+
+
+class Error(Exception):
+    pass
+
+
+class InterfaceError(Error):
+    pass
+
+
+class DatabaseError(Error):
+    pass
+
+
+class InternalError(DatabaseError):
+    pass
+
+
+class OperationalError(DatabaseError):
+    pass
+
+
+class ProgrammingError(DatabaseError):
+    pass
+
+
+class IntegrityError(DatabaseError):
+    pass
+
+
+class DataError(DatabaseError):
+    pass
+
+
+class NotSupportedError(DatabaseError):
+    pass

clickhouse_driver/dbapi/extras.py

@@ -0,0 +1,73 @@
+import re
+from collections import namedtuple
+from functools import lru_cache
+
+from .cursor import Cursor
+
+
+class DictCursor(Cursor):
+    """
+    A cursor that generates results as :class:`dict`.
+
+    ``fetch*()`` methods will return dicts instead of tuples.
+    """
+
+    def fetchone(self):
+        rv = super(DictCursor, self).fetchone()
+        if rv is not None:
+            rv = dict(zip(self._columns, rv))
+        return rv
+
+    def fetchmany(self, size=None):
+        rv = super(DictCursor, self).fetchmany(size=size)
+        return [dict(zip(self._columns, x)) for x in rv]
+
+    def fetchall(self):
+        rv = super(DictCursor, self).fetchall()
+        return [dict(zip(self._columns, x)) for x in rv]
+
+
+class NamedTupleCursor(Cursor):
+    """
+    A cursor that generates results as named tuples created by
+    :func:`~collections.namedtuple`.
+
+    ``fetch*()`` methods will return named tuples instead of regular tuples, so
+    their elements can be accessed both as regular numeric items as well as
+    attributes.
+    """
+
+    # ascii except alnum and underscore
+    _re_clean = re.compile(
+        '[' + re.escape(' !"#$%&\'()*+,-./:;<=>?@[\\]^`{|}~') + ']')
+
+    @classmethod
+    @lru_cache(512)
+    def _make_nt(self, key):
+        fields = []
+        for s in key:
+            s = self._re_clean.sub('_', s)
+            # Python identifier cannot start with numbers, namedtuple fields
+            # cannot start with underscore.
+            if s[0] == '_' or '0' <= s[0] <= '9':
+                s = 'f' + s
+            fields.append(s)
+
+        return namedtuple('Record', fields)
+
+    def fetchone(self):
+        rv = super(NamedTupleCursor, self).fetchone()
+        if rv is not None:
+            nt = self._make_nt(self._columns)
+            rv = nt(*rv)
+        return rv
+
+    def fetchmany(self, size=None):
+        rv = super(NamedTupleCursor, self).fetchmany(size=size)
+        nt = self._make_nt(self._columns)
+        return [nt(*x) for x in rv]
+
+    def fetchall(self):
+        rv = super(NamedTupleCursor, self).fetchall()
+        nt = self._make_nt(self._columns)
+        return [nt(*x) for x in rv]

clickhouse_driver/defines.py

@@ -1,42 +1,58 @@
-
-DEFAULT_DATABASE = 'default'
-DEFAULT_USER = 'default'
-DEFAULT_PASSWORD = ''
-
-DEFAULT_PORT = 9000
-DEFAULT_SECURE_PORT = 9440
-
-DBMS_MIN_REVISION_WITH_TEMPORARY_TABLES = 50264
-DBMS_MIN_REVISION_WITH_TOTAL_ROWS_IN_PROGRESS = 51554
-DBMS_MIN_REVISION_WITH_BLOCK_INFO = 51903
-# Legacy above.
-DBMS_MIN_REVISION_WITH_CLIENT_INFO = 54032
-DBMS_MIN_REVISION_WITH_SERVER_TIMEZONE = 54058
-DBMS_MIN_REVISION_WITH_QUOTA_KEY_IN_CLIENT_INFO = 54060
-DBMS_MIN_REVISION_WITH_SERVER_DISPLAY_NAME = 54372
-DBMS_MIN_REVISION_WITH_VERSION_PATCH = 54401
-DBMS_MIN_REVISION_WITH_SERVER_LOGS = 54406
-DBMS_MIN_REVISION_WITH_COLUMN_DEFAULTS_METADATA = 54410
-DBMS_MIN_REVISION_WITH_CLIENT_WRITE_INFO = 54420
-DBMS_MIN_REVISION_WITH_SETTINGS_SERIALIZED_AS_STRINGS = 54429
-DBMS_MIN_REVISION_WITH_INTERSERVER_SECRET = 54441
-
-# Timeouts
-DBMS_DEFAULT_CONNECT_TIMEOUT_SEC = 10
-DBMS_DEFAULT_TIMEOUT_SEC = 300
-
-DBMS_DEFAULT_SYNC_REQUEST_TIMEOUT_SEC = 5
-
-DEFAULT_COMPRESS_BLOCK_SIZE = 1048576
-DEFAULT_INSERT_BLOCK_SIZE = 1048576
-
-DBMS_NAME = 'ClickHouse'
-CLIENT_NAME = 'python-driver'
-CLIENT_VERSION_MAJOR = 20
-CLIENT_VERSION_MINOR = 10
-CLIENT_VERSION_PATCH = 2
-CLIENT_REVISION = 54441
-
-BUFFER_SIZE = 1048576
-
-STRINGS_ENCODING = 'utf-8'
+
+DEFAULT_DATABASE = ''
+DEFAULT_USER = 'default'
+DEFAULT_PASSWORD = ''
+
+DEFAULT_PORT = 9000
+DEFAULT_SECURE_PORT = 9440
+
+DBMS_MIN_REVISION_WITH_TEMPORARY_TABLES = 50264
+DBMS_MIN_REVISION_WITH_TOTAL_ROWS_IN_PROGRESS = 51554
+DBMS_MIN_REVISION_WITH_BLOCK_INFO = 51903
+# Legacy above.
+DBMS_MIN_REVISION_WITH_CLIENT_INFO = 54032
+DBMS_MIN_REVISION_WITH_SERVER_TIMEZONE = 54058
+DBMS_MIN_REVISION_WITH_QUOTA_KEY_IN_CLIENT_INFO = 54060
+DBMS_MIN_REVISION_WITH_SERVER_DISPLAY_NAME = 54372
+DBMS_MIN_REVISION_WITH_VERSION_PATCH = 54401
+DBMS_MIN_REVISION_WITH_SERVER_LOGS = 54406
+DBMS_MIN_REVISION_WITH_COLUMN_DEFAULTS_METADATA = 54410
+DBMS_MIN_REVISION_WITH_CLIENT_WRITE_INFO = 54420
+DBMS_MIN_REVISION_WITH_SETTINGS_SERIALIZED_AS_STRINGS = 54429
+DBMS_MIN_REVISION_WITH_INTERSERVER_SECRET = 54441
+DBMS_MIN_REVISION_WITH_OPENTELEMETRY = 54442
+DBMS_MIN_PROTOCOL_VERSION_WITH_DISTRIBUTED_DEPTH = 54448
+DBMS_MIN_PROTOCOL_VERSION_WITH_INITIAL_QUERY_START_TIME = 54449
+DBMS_MIN_PROTOCOL_VERSION_WITH_INCREMENTAL_PROFILE_EVENTS = 54451
+DBMS_MIN_REVISION_WITH_PARALLEL_REPLICAS = 54453
+DBMS_MIN_REVISION_WITH_CUSTOM_SERIALIZATION = 54454
+DBMS_MIN_PROTOCOL_VERSION_WITH_PROFILE_EVENTS_IN_INSERT = 54456
+DBMS_MIN_PROTOCOL_VERSION_WITH_ADDENDUM = 54458
+DBMS_MIN_PROTOCOL_VERSION_WITH_QUOTA_KEY = 54458
+DBMS_MIN_PROTOCOL_VERSION_WITH_PARAMETERS = 54459
+DBMS_MIN_PROTOCOL_VERSION_WITH_SERVER_QUERY_TIME_IN_PROGRESS = 54460
+DBMS_MIN_PROTOCOL_VERSION_WITH_PASSWORD_COMPLEXITY_RULES = 54461
+DBMS_MIN_REVISION_WITH_INTERSERVER_SECRET_V2 = 54462
+DBMS_MIN_PROTOCOL_VERSION_WITH_TOTAL_BYTES_IN_PROGRESS = 54463
+DBMS_MIN_PROTOCOL_VERSION_WITH_TIMEZONE_UPDATES = 54464
+DBMS_MIN_REVISION_WITH_SYSTEM_KEYWORDS_TABLE = 54468
+
+# Timeouts
+DBMS_DEFAULT_CONNECT_TIMEOUT_SEC = 10
+DBMS_DEFAULT_TIMEOUT_SEC = 300
+
+DBMS_DEFAULT_SYNC_REQUEST_TIMEOUT_SEC = 5
+
+DEFAULT_COMPRESS_BLOCK_SIZE = 1048576
+DEFAULT_INSERT_BLOCK_SIZE = 1048576
+
+DBMS_NAME = 'ClickHouse'
+CLIENT_NAME = 'python-driver'
+CLIENT_VERSION_MAJOR = 20
+CLIENT_VERSION_MINOR = 10
+CLIENT_VERSION_PATCH = 2
+CLIENT_REVISION = DBMS_MIN_REVISION_WITH_SYSTEM_KEYWORDS_TABLE
+
+BUFFER_SIZE = 1048576
+
+STRINGS_ENCODING = 'utf-8'