mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
zeekay
073c1bf61e
The wire protocol was the last holdout and it was not cosmetic: the Hanzo Datastore server reads X-Datastore-User, and answers a request bearing X-ClickHouse-User with 401. Verified against the live server, both directions. So a client still sending the old headers is not merely off-brand, it cannot authenticate over HTTP. Everything else follows: types, dialects, error modules, docstrings and doc links now name Datastore, and doc URLs point at docs.hanzo.ai/datastore rather than the upstream vendor's site. Zero occurrences of the old name remain. Still cut from the pinned upstream version, so behaviour is unchanged.
100 lines
3.0 KiB
Python
100 lines
3.0 KiB
Python
from ..client import Client
|
|
from .. import defines
|
|
from .cursor import Cursor
|
|
from .errors import InterfaceError
|
|
|
|
|
|
class Connection(object):
|
|
"""
|
|
Creates new Connection for accessing Datastore database.
|
|
|
|
Connection is just wrapper for handling multiple cursors (clients) and
|
|
do not initiate actual connections to the Datastore server.
|
|
|
|
See parameters description in
|
|
:data:`~datastore_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:`~datastore_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 Datastore has no transactions.
|
|
"""
|
|
pass
|
|
|
|
def rollback(self):
|
|
"""
|
|
Do nothing since Datastore 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
|