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

clickhouse_driver/context.py

@@ -1,36 +1,36 @@
-
-class Context(object):
-    def __init__(self):
-        self._server_info = None
-        self._settings = None
-        self._client_settings = None
-        super(Context, self).__init__()
-
-    @property
-    def server_info(self):
-        return self._server_info
-
-    @server_info.setter
-    def server_info(self, value):
-        self._server_info = value
-
-    @property
-    def settings(self):
-        return self._settings.copy()
-
-    @settings.setter
-    def settings(self, value):
-        self._settings = value.copy()
-
-    @property
-    def client_settings(self):
-        return self._client_settings.copy()
-
-    @client_settings.setter
-    def client_settings(self, value):
-        self._client_settings = value.copy()
-
-    def __repr__(self):
-        return '<Context(server_info=%s, client_settings=%s, settings=%s)>' % (
-            self._server_info, self._client_settings, self._settings
-        )
+
+class Context(object):
+    def __init__(self):
+        self._server_info = None
+        self._settings = None
+        self._client_settings = None
+        super(Context, self).__init__()
+
+    @property
+    def server_info(self):
+        return self._server_info
+
+    @server_info.setter
+    def server_info(self, value):
+        self._server_info = value
+
+    @property
+    def settings(self):
+        return self._settings.copy()
+
+    @settings.setter
+    def settings(self, value):
+        self._settings = value.copy()
+
+    @property
+    def client_settings(self):
+        return self._client_settings.copy()
+
+    @client_settings.setter
+    def client_settings(self, value):
+        self._client_settings = value.copy()
+
+    def __repr__(self):
+        return '<Context(server_info=%s, client_settings=%s, settings=%s)>' % (
+            self._server_info, self._client_settings, self._settings
+        )

clickhouse_driver/dbapi/__init__.py

@@ -1,62 +1,62 @@
-from .connection import Connection
-from .errors import (
-    Warning, Error, DataError, DatabaseError, ProgrammingError, IntegrityError,
-    InterfaceError, InternalError, NotSupportedError, OperationalError
-)
-from .. import defines
-
-apilevel = '2.0'
-
-threadsafety = 2
-
-paramstyle = 'pyformat'
-
-
-def connect(dsn=None, host=None,
-            user=defines.DEFAULT_USER, password=defines.DEFAULT_PASSWORD,
-            port=defines.DEFAULT_PORT, database=defines.DEFAULT_DATABASE,
-            **kwargs):
-    """
-    Create a new database connection.
-
-    The connection can be specified via DSN:
-
-        ``conn = connect("clickhouse://localhost/test?param1=value1&...")``
-
-    or using database and credentials arguments:
-
-        ``conn = connect(database="test", user="default", password="default",
-        host="localhost", **kwargs)``
-
-    The basic connection parameters are:
-
-    - *host*: host with running ClickHouse server.
-    - *port*: port ClickHouse server is bound to.
-    - *database*: database connect to.
-    - *user*: database user.
-    - *password*: user's password.
-
-    See defaults in :data:`~clickhouse_driver.connection.Connection`
-    constructor.
-
-    DSN or host is required.
-
-    Any other keyword parameter will be passed to the underlying Connection
-    class.
-
-    :return: a new connection.
-    """
-
-    if dsn is None and host is None:
-        raise ValueError('host or dsn is required')
-
-    return Connection(dsn=dsn, user=user, password=password, host=host,
-                      port=port, database=database, **kwargs)
-
-
-__all__ = [
-    'connect',
-    'Warning', 'Error', 'DataError', 'DatabaseError', 'ProgrammingError',
-    'IntegrityError', 'InterfaceError', 'InternalError', 'NotSupportedError',
-    'OperationalError'
-]
+from .connection import Connection
+from .errors import (
+    Warning, Error, DataError, DatabaseError, ProgrammingError, IntegrityError,
+    InterfaceError, InternalError, NotSupportedError, OperationalError
+)
+from .. import defines
+
+apilevel = '2.0'
+
+threadsafety = 2
+
+paramstyle = 'pyformat'
+
+
+def connect(dsn=None, host=None,
+            user=defines.DEFAULT_USER, password=defines.DEFAULT_PASSWORD,
+            port=defines.DEFAULT_PORT, database=defines.DEFAULT_DATABASE,
+            **kwargs):
+    """
+    Create a new database connection.
+
+    The connection can be specified via DSN:
+
+        ``conn = connect("clickhouse://localhost/test?param1=value1&...")``
+
+    or using database and credentials arguments:
+
+        ``conn = connect(database="test", user="default", password="default",
+        host="localhost", **kwargs)``
+
+    The basic connection parameters are:
+
+    - *host*: host with running ClickHouse server.
+    - *port*: port ClickHouse server is bound to.
+    - *database*: database connect to.
+    - *user*: database user.
+    - *password*: user's password.
+
+    See defaults in :data:`~clickhouse_driver.connection.Connection`
+    constructor.
+
+    DSN or host is required.
+
+    Any other keyword parameter will be passed to the underlying Connection
+    class.
+
+    :return: a new connection.
+    """
+
+    if dsn is None and host is None:
+        raise ValueError('host or dsn is required')
+
+    return Connection(dsn=dsn, user=user, password=password, host=host,
+                      port=port, database=database, **kwargs)
+
+
+__all__ = [
+    'connect',
+    'Warning', 'Error', 'DataError', 'DatabaseError', 'ProgrammingError',
+    'IntegrityError', 'InterfaceError', 'InternalError', 'NotSupportedError',
+    'OperationalError'
+]

clickhouse_driver/dbapi/connection.py

@@ -1,96 +1,99 @@
-from ..client import Client
-from .. import defines
-from .cursor import Cursor
-from .errors import InterfaceError
-
-
-class Connection(object):
-    """
-    Creates new Connection for accessing ClickHouse database.
-
-    Connection is just wrapper for handling multiple cursors (clients) and
-    do not initiate actual connections to the ClickHouse server.
-
-    See parameters description in
-    :data:`~clickhouse_driver.connection.Connection`.
-    """
-    def __init__(self, dsn=None, host=None,
-                 user=defines.DEFAULT_USER, password=defines.DEFAULT_PASSWORD,
-                 port=defines.DEFAULT_PORT, database=defines.DEFAULT_DATABASE,
-                 **kwargs):
-        self.cursors = []
-
-        self.dsn = dsn
-        self.user = user
-        self.password = password
-        self.host = host
-        self.port = port
-        self.database = database
-        self.connection_kwargs = kwargs
-        self.is_closed = False
-        self._hosts = None
-        super(Connection, self).__init__()
-
-    def __repr__(self):
-        return '<connection object at 0x{0:x}; closed: {1:}>'.format(
-            id(self), self.is_closed
-        )
-
-    # Context manager integrations.
-    def __enter__(self):
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        self.close()
-
-    def _make_client(self):
-        """
-        :return: a new Client instance.
-        """
-        if self.dsn is not None:
-            return Client.from_url(self.dsn)
-
-        return Client(self.host, port=self.port,
-                      user=self.user, password=self.password,
-                      database=self.database, **self.connection_kwargs)
-
-    def close(self):
-        """
-        Close the connection now. The connection will be unusable from this
-        point forward; an :data:`~clickhouse_driver.dbapi.Error` (or subclass)
-        exception will be raised if any operation is attempted with the
-        connection. The same applies to all cursor objects trying to use the
-        connection.
-        """
-        for cursor in self.cursors:
-            cursor.close()
-
-        self.is_closed = True
-
-    def commit(self):
-        """
-        Do nothing since ClickHouse has no transactions.
-        """
-        pass
-
-    def rollback(self):
-        """
-        Do nothing since ClickHouse has no transactions.
-        """
-        pass
-
-    def cursor(self):
-        """
-        :return: a new Cursor Object using the connection.
-        """
-        if self.is_closed:
-            raise InterfaceError('connection already closed')
-
-        client = self._make_client()
-        if self._hosts is None:
-            self._hosts = client.connection.hosts
-        else:
-            client.connection.hosts = self._hosts
-        cursor = Cursor(client, self)
-        self.cursors.append(cursor)
-        return cursor
+from ..client import Client
+from .. import defines
+from .cursor import Cursor
+from .errors import InterfaceError
+
+
+class Connection(object):
+    """
+    Creates new Connection for accessing ClickHouse database.
+
+    Connection is just wrapper for handling multiple cursors (clients) and
+    do not initiate actual connections to the ClickHouse server.
+
+    See parameters description in
+    :data:`~clickhouse_driver.connection.Connection`.
+    """
+    def __init__(self, dsn=None, host=None,
+                 user=defines.DEFAULT_USER, password=defines.DEFAULT_PASSWORD,
+                 port=defines.DEFAULT_PORT, database=defines.DEFAULT_DATABASE,
+                 **kwargs):
+        self.cursors = []
+
+        self.dsn = dsn
+        self.user = user
+        self.password = password
+        self.host = host
+        self.port = port
+        self.database = database
+        self.connection_kwargs = kwargs
+        self.is_closed = False
+        self._hosts = None
+        super(Connection, self).__init__()
+
+    def __repr__(self):
+        return '<connection object at 0x{0:x}; closed: {1:}>'.format(
+            id(self), self.is_closed
+        )
+
+    # Context manager integrations.
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    def _make_client(self):
+        """
+        :return: a new Client instance.
+        """
+        if self.dsn is not None:
+            return Client.from_url(self.dsn)
+
+        return Client(self.host, port=self.port,
+                      user=self.user, password=self.password,
+                      database=self.database, **self.connection_kwargs)
+
+    def close(self):
+        """
+        Close the connection now. The connection will be unusable from this
+        point forward; an :data:`~clickhouse_driver.dbapi.Error` (or subclass)
+        exception will be raised if any operation is attempted with the
+        connection. The same applies to all cursor objects trying to use the
+        connection.
+        """
+        for cursor in self.cursors:
+            cursor.close()
+
+        self.is_closed = True
+
+    def commit(self):
+        """
+        Do nothing since ClickHouse has no transactions.
+        """
+        pass
+
+    def rollback(self):
+        """
+        Do nothing since ClickHouse has no transactions.
+        """
+        pass
+
+    def cursor(self, cursor_factory=None):
+        """
+        :param cursor_factory: Argument can be used to create non-standard
+                               cursors.
+        :return: a new cursor object using the connection.
+        """
+        if self.is_closed:
+            raise InterfaceError('connection already closed')
+
+        client = self._make_client()
+        if self._hosts is None:
+            self._hosts = client.connection.hosts
+        else:
+            client.connection.hosts = self._hosts
+        cursor_factory = cursor_factory or Cursor
+        cursor = cursor_factory(client, self)
+        self.cursors.append(cursor)
+        return cursor