From 40ab26d4f14b2fbf24ba46a7bf8c75af1f366c52 Mon Sep 17 00:00:00 2001
From: Beto Dealmeida <roberto@dealmeida.net>
Date: Fri, 11 Feb 2022 17:40:20 -0800
Subject: [PATCH] fix: unambiguous trino driver (#18683)

---
 setup.py                          | 5 ++++-
 superset/db_engine_specs/trino.py | 1 +
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index ba0f5bad90..9899daf400 100644
--- a/setup.py
+++ b/setup.py
@@ -62,7 +62,10 @@ setup(
     packages=find_packages(),
     include_package_data=True,
     zip_safe=False,
-    entry_points={"console_scripts": ["superset=superset.cli.main:superset"]},
+    entry_points={
+        "console_scripts": ["superset=superset.cli.main:superset"],
+        "sqlalchemy.dialects": ["trinonative = sqlalchemy_trino.dialect:TrinoDialect"],
+    },
     install_requires=[
         "backoff>=1.8.0",
         "bleach>=3.0.2, <4.0.0",
diff --git a/superset/db_engine_specs/trino.py b/superset/db_engine_specs/trino.py
index 778845d132..4e5f153ad2 100644
--- a/superset/db_engine_specs/trino.py
+++ b/superset/db_engine_specs/trino.py
@@ -34,6 +34,7 @@ logger = logging.getLogger(__name__)
 
 class TrinoEngineSpec(BaseEngineSpec):
     engine = "trino"
+    engine_aliases = {"trinonative"}
     engine_name = "Trino"
 
     _time_grain_expressions = {