databricks-sql-connector 4.0.1__tar.gz → 4.0.3__tar.gz

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/CHANGELOG.md

@@ -1,7 +1,17 @@
 # Release History
 
+# 4.0.3 (2025-04-22)
+
+- Fix: Removed `packaging` dependency in favour of default libraries, for `urllib3` version checks (databricks/databricks-sql-python#547 by @jprakash-db)
+- 
+# 4.0.2 (2025-04-01)
+
+- Fix: Relaxed the pin for `python-dateutil` to be `^2.8.0` (databricks/databricks-sql-python#538 by @jprakash-db)
+
 # 4.0.1 (2025-03-19)
 
+**Note: this release was yanked from Pypi on 19 March 2025 due to compatibility issues with `dbt-databricks<1.5.3`**
+
 - Support for multiple timestamp formats parsing (databricks/databricks-sql-python#533 by @jprakash-db)
 - Rename `_user_agent_entry` in connect call to `user_agent_entry` to expose it as a public parameter. (databricks/databricks-sql-python#530 by @shivam2680)
 - Fix: compatibility with urllib3 versions less than 2.x. (databricks/databricks-sql-python#526 by @shivam2680)

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: databricks-sql-connector
-Version: 4.0.1
+Version: 4.0.3
 Summary: Databricks SQL Connector for Python
 License: Apache-2.0
 Author: Databricks
@@ -21,7 +21,7 @@ Requires-Dist: pandas (>=1.2.5,<2.3.0) ; python_version >= "3.8" and python_vers
 Requires-Dist: pandas (>=2.2.3,<2.3.0) ; python_version >= "3.13"
 Requires-Dist: pyarrow (>=14.0.1) ; (python_version >= "3.8" and python_version < "3.13") and (extra == "pyarrow")
 Requires-Dist: pyarrow (>=18.0.0) ; (python_version >= "3.13") and (extra == "pyarrow")
-Requires-Dist: python-dateutil (>=2.9.0,<3.0.0)
+Requires-Dist: python-dateutil (>=2.8.0,<3.0.0)
 Requires-Dist: requests (>=2.18.1,<3.0.0)
 Requires-Dist: thrift (>=0.16.0,<0.21.0)
 Requires-Dist: urllib3 (>=1.26)

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "databricks-sql-connector"
-version = "4.0.1"
+version = "4.0.3"
 description = "Databricks SQL Connector for Python"
 authors = ["Databricks <databricks-sql-connector-maintainers@databricks.com>"]
 license = "Apache-2.0"
@@ -24,7 +24,7 @@ pyarrow = [
     { version = ">=14.0.1", python = ">=3.8,<3.13", optional=true },
     { version = ">=18.0.0", python = ">=3.13", optional=true }
 ]
-python-dateutil = "^2.9.0"
+python-dateutil = "^2.8.0"
 
 [tool.poetry.extras]
 pyarrow = ["pyarrow"]

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/src/databricks/sql/__init__.py

@@ -68,7 +68,7 @@ DATETIME = DBAPITypeObject("timestamp")
 DATE = DBAPITypeObject("date")
 ROWID = DBAPITypeObject()
 
-__version__ = "4.0.1"
+__version__ = "4.0.3"
 USER_AGENT_NAME = "PyDatabricksSqlConnector"
 
 # These two functions are pyhive legacy

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/src/databricks/sql/auth/retry.py

@@ -2,6 +2,7 @@ import logging
 import random
 import time
 import typing
+from importlib.metadata import version
 from enum import Enum
 from typing import List, Optional, Tuple, Union
 
@@ -16,7 +17,6 @@ except ImportError:
     from urllib3 import HTTPResponse as BaseHTTPResponse
 from urllib3 import Retry
 from urllib3.util.retry import RequestHistory
-from packaging import version
 
 
 from databricks.sql.exc import (
@@ -312,7 +312,9 @@ class DatabricksRetryPolicy(Retry):
 
         current_attempt = self.stop_after_attempts_count - int(self.total or 0)
         proposed_backoff = (2**current_attempt) * self.delay_min
-        if version.parse(urllib3.__version__) >= version.parse("2.0.0"):
+
+        library_version = version("urllib3")
+        if int(library_version.split(".")[0]) >= 2:
             if self.backoff_jitter != 0.0:
                 proposed_backoff += random.random() * self.backoff_jitter
 

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/src/databricks/sql/client.py

@@ -896,6 +896,19 @@ class Cursor:
         self._check_not_closed()
         return self.thrift_backend.get_query_state(self.active_op_handle)
 
+    def is_query_pending(self):
+        """
+        Checks whether the async executing query is in pending state or not
+
+        :return:
+        """
+        operation_state = self.get_query_state()
+
+        return not operation_state or operation_state in [
+            ttypes.TOperationState.RUNNING_STATE,
+            ttypes.TOperationState.PENDING_STATE,
+        ]
+
     def get_async_execution_result(self):
         """
 
@@ -905,13 +918,7 @@ class Cursor:
         """
         self._check_not_closed()
 
-        def is_executing(operation_state) -> "bool":
-            return not operation_state or operation_state in [
-                ttypes.TOperationState.RUNNING_STATE,
-                ttypes.TOperationState.PENDING_STATE,
-            ]
-
-        while is_executing(self.get_query_state()):
+        while self.is_query_pending():
             # Poll after some default time
             time.sleep(self.ASYNC_DEFAULT_POLLING_INTERVAL)
 

{databricks_sql_connector-4.0.1 → databricks_sql_connector-4.0.3}/src/databricks/sql/thrift_backend.py

@@ -797,12 +797,15 @@ class ThriftBackend:
             t_result_set_metadata_resp.schema
         )
 
-        schema_bytes = (
-            t_result_set_metadata_resp.arrowSchema
-            or self._hive_schema_to_arrow_schema(t_result_set_metadata_resp.schema)
-            .serialize()
-            .to_pybytes()
-        )
+        if pyarrow:
+            schema_bytes = (
+                t_result_set_metadata_resp.arrowSchema
+                or self._hive_schema_to_arrow_schema(t_result_set_metadata_resp.schema)
+                .serialize()
+                .to_pybytes()
+            )
+        else:
+            schema_bytes = None
 
         queue = ResultSetQueueFactory.build_queue(
             row_set_type=resp.resultSetMetadata.resultFormat,