From e1ca8503d65f6423d91ec618c5f6939b09a3def5 Mon Sep 17 00:00:00 2001
From: Beto Dealmeida <roberto@dealmeida.net>
Date: Thu, 25 Apr 2024 15:49:53 -0400
Subject: [PATCH] fix: Dremio alias

---
 superset/db_engine_specs/__init__.py | 2 +-
 superset/db_engine_specs/dremio.py   | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/superset/db_engine_specs/__init__.py b/superset/db_engine_specs/__init__.py
index c40cf1554d6b1..2b840087c67c0 100644
--- a/superset/db_engine_specs/__init__.py
+++ b/superset/db_engine_specs/__init__.py
@@ -154,7 +154,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
index 746576d3f30da..32ffee81a6c5c 100644
--- 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}",