From db959a6463843802af152bb60a1ace80fa22add5 Mon Sep 17 00:00:00 2001
From: Vitor Avila <96086495+Vitor-Avila@users.noreply.github.com>
Date: Fri, 4 Apr 2025 15:09:15 -0300
Subject: [PATCH] chore(Databricks): Display older Databricks driver as legacy
(#33001)
---
superset/db_engine_specs/databricks.py | 4 ++--
tests/unit_tests/db_engine_specs/test_databricks.py | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/superset/db_engine_specs/databricks.py b/superset/db_engine_specs/databricks.py
index 7574989abf..fe7f908400 100644
--- a/superset/db_engine_specs/databricks.py
+++ b/superset/db_engine_specs/databricks.py
@@ -353,7 +353,7 @@ class DatabricksDynamicBaseEngineSpec(BasicParametersMixin, DatabricksBaseEngine
class DatabricksNativeEngineSpec(DatabricksDynamicBaseEngineSpec):
engine = "databricks"
- engine_name = "Databricks"
+ engine_name = "Databricks (legacy)"
drivers = {"connector": "Native all-purpose driver"}
default_driver = "connector"
@@ -485,7 +485,7 @@ class DatabricksNativeEngineSpec(DatabricksDynamicBaseEngineSpec):
class DatabricksPythonConnectorEngineSpec(DatabricksDynamicBaseEngineSpec):
engine = "databricks"
- engine_name = "Databricks Python Connector"
+ engine_name = "Databricks"
default_driver = "databricks-sql-python"
drivers = {"databricks-sql-python": "Databricks SQL Python"}
diff --git a/tests/unit_tests/db_engine_specs/test_databricks.py b/tests/unit_tests/db_engine_specs/test_databricks.py
index 95403780e1..a585002009 100644
--- a/tests/unit_tests/db_engine_specs/test_databricks.py
+++ b/tests/unit_tests/db_engine_specs/test_databricks.py
@@ -187,7 +187,7 @@ def test_extract_errors() -> None:
error_type=SupersetErrorType.GENERIC_DB_ENGINE_ERROR,
level=ErrorLevel.ERROR,
extra={
- "engine_name": "Databricks",
+ "engine_name": "Databricks (legacy)",
"issue_codes": [
{
"code": 1002,
@@ -214,7 +214,7 @@ def test_extract_errors_with_context() -> None:
error_type=SupersetErrorType.GENERIC_DB_ENGINE_ERROR,
level=ErrorLevel.ERROR,
extra={
- "engine_name": "Databricks",
+ "engine_name": "Databricks (legacy)",
"issue_codes": [
{
"code": 1002,
--
GitLab