From 4a1f0f575f1032fe43f7bb6db6e65b8904ad20fa Mon Sep 17 00:00:00 2001 From: hantmac Date: Fri, 12 May 2023 17:56:14 +0800 Subject: [PATCH] fix: fix http handler session resp --- databend_py/__init__.py | 2 +- databend_py/connection.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/databend_py/__init__.py b/databend_py/__init__.py index 92fab7a..9d9777e 100644 --- a/databend_py/__init__.py +++ b/databend_py/__init__.py @@ -2,7 +2,7 @@ from .connection import Connection from .datetypes import DatabendDataType -VERSION = (0, 3, 9) +VERSION = (0, 4, 1) __version__ = '.'.join(str(x) for x in VERSION) __all__ = ['Client', 'Connection', 'DatabendDataType'] diff --git a/databend_py/connection.py b/databend_py/connection.py index b04fb2e..25c7b91 100644 --- a/databend_py/connection.py +++ b/databend_py/connection.py @@ -110,9 +110,11 @@ def query(self, statement): log.logger.debug(f"http sql: {statement}") query_sql = {'sql': statement, "string_fields": True} if self.client_session is not None and len(self.client_session) != 0: + if "database" not in self.client_session: + self.client_session = {"database": self.database} query_sql['session'] = self.client_session else: - self.client_session = {"db": self.database} + self.client_session = {"database": self.database} query_sql['session'] = self.client_session log.logger.debug(f"http headers {self.make_headers()}") response = requests.post(url,