feat(ingest/superset): map awsathena platform name to athena (datahub…

…-project#10005)
Salman-Apptware · Mar 11, 2024 · ae3443b · ae3443b
1 parent b6956f9
commit ae3443b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/superset.py b/metadata-ingestion/src/datahub/ingestion/source/superset.py
@@ -226,8 +226,14 @@ def get_platform_from_database_id(self, database_id):
         ).json()
         sqlalchemy_uri = database_response.get("result", {}).get("sqlalchemy_uri")
         if sqlalchemy_uri is None:
-            return database_response.get("result", {}).get("backend", "external")
-        return get_platform_from_sqlalchemy_uri(sqlalchemy_uri)
+            platform_name = database_response.get("result", {}).get(
+                "backend", "external"
+            )
+        else:
+            platform_name = get_platform_from_sqlalchemy_uri(sqlalchemy_uri)
+        if platform_name == "awsathena":
+            return "athena"
+        return platform_name
 
     @lru_cache(maxsize=None)
     def get_datasource_urn_from_id(self, datasource_id):