From 0d064f4451ef344a6dbe8c86e40c19dd7c3f697c Mon Sep 17 00:00:00 2001 From: Ke Zhu Date: Fri, 24 Sep 2021 05:31:17 -0400 Subject: [PATCH] Be able to lookup driver by engine alias (#16800) --- superset/db_engine_specs/__init__.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/superset/db_engine_specs/__init__.py b/superset/db_engine_specs/__init__.py index c5b47f153c..9eeb87acfe 100644 --- 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