fix: Dremio alias (apache#28222)

(cherry picked from commit 173d5d0)
preset-io · Apr 29, 2024 · 2d0eb38 · 2d0eb38
1 parent ccda3b4
commit 2d0eb38
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/superset/db_engine_specs/__init__.py b/superset/db_engine_specs/__init__.py
@@ -153,7 +153,7 @@ def get_available_engine_specs() -> dict[type[BaseEngineSpec], set[str]]:
         try:
             dialect = ep.load()
         except Exception as ex:  # pylint: disable=broad-except
-            logger.warning("Unable to load SQLAlchemy dialect %s: %s", dialect, ex)
+            logger.warning("Unable to load SQLAlchemy dialect %s: %s", ep.name, ex)
         else:
             backend = dialect.name
             if isinstance(backend, bytes):

diff --git a/superset/db_engine_specs/dremio.py b/superset/db_engine_specs/dremio.py
@@ -37,6 +37,13 @@
 class DremioEngineSpec(BaseEngineSpec):
     engine = "dremio"
     engine_name = "Dremio"
+    engine_aliases = {"dremio+flight"}
+    sqlalchemy_uri_placeholder = (
+        "dremio+flight://data.dremio.cloud:443/?"
+        "Token=<TOKEN>&"
+        "UseEncryption=true&"
+        "disableCertificateVerification=true"
+    )
 
     _time_grain_expressions = {
         None: "{col}",