pt/superset
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2024-09-12 00:29:39 -04:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ke Zhu 2021-09-24 05:31:17 -04:00 committed by GitHub
parent c933250dad
commit 0d064f4451
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
superset/db_engine_specs/__init__.py
View File

@ -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