fix(ingest): extract redshift platform correctly from sqlalchemy uri (d…

…atahub-project#4421) * fix(ingest): extract redshift platform from sqlalchemy uri
maggiehays · Aug 1, 2022 · b2c8858 · b2c8858
1 parent 51e912c
commit b2c8858
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/sql_common.py b/metadata-ingestion/src/datahub/ingestion/source/sql/sql_common.py
@@ -102,8 +102,8 @@ def get_platform_from_sqlalchemy_uri(sqlalchemy_uri: str) -> str:
     if sqlalchemy_uri.startswith("mssql"):
         return "mssql"
     if (
-        sqlalchemy_uri.startswith("jdbc:postgres:")
-        and sqlalchemy_uri.index("redshift.amazonaws") > 0
+        sqlalchemy_uri.startswith(("jdbc:postgres:", "postgresql"))
+        and sqlalchemy_uri.find("redshift.amazonaws") > 0
     ) or sqlalchemy_uri.startswith("redshift"):
         return "redshift"
     if sqlalchemy_uri.startswith("snowflake"):