Be able to lookup driver by engine alias (apache#16800)

opus-42 · Nov 14, 2021 · f5dd0b9 · f5dd0b9
1 parent e913f90
commit f5dd0b9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/superset/db_engine_specs/__init__.py b/superset/db_engine_specs/__init__.py
@@ -147,6 +147,15 @@ def get_available_engine_specs() -> Dict[Type[BaseEngineSpec], Set[str]]:
 
     available_engines = {}
     for engine_spec in load_engine_specs():
-        available_engines[engine_spec] = drivers[engine_spec.engine]
+        driver = drivers[engine_spec.engine]
+
+        # lookup driver by engine aliases.
+        if not driver and engine_spec.engine_aliases:
+            for alias in engine_spec.engine_aliases:
+                driver = drivers[alias]
+                if driver:
+                    break
+
+        available_engines[engine_spec] = driver
 
     return available_engines