diff --git a/metadata-ingestion/src/datahub/ingestion/source/sql/sql_common.py b/metadata-ingestion/src/datahub/ingestion/source/sql/sql_common.py index fbbdf93b7e0b10..bed10be9b39fea 100644 --- 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"):