Skip to content

Commit e939741

Browse files
authored
feat: add support for aurora-mysql and aurora-postgresql engines (#3188)
* Update postgresql.py Add support for aurora-postgresql engine * Update mysql.py Add support for aurora-mysql engine
1 parent f9c9215 commit e939741

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

awswrangler/mysql.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def connect(
168168
attrs: _db_utils.ConnectionAttributes = _db_utils.get_connection_attributes(
169169
connection=connection, secret_id=secret_id, catalog_id=catalog_id, dbname=dbname, boto3_session=boto3_session
170170
)
171-
if attrs.kind != "mysql":
171+
if attrs.kind not in ("mysql", "aurora-mysql"):
172172
raise exceptions.InvalidDatabaseType(f"Invalid connection type ({attrs.kind}. It must be a MySQL connection.)")
173173
return pymysql.connect(
174174
user=attrs.user,

awswrangler/postgresql.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ def connect(
225225
attrs: _db_utils.ConnectionAttributes = _db_utils.get_connection_attributes(
226226
connection=connection, secret_id=secret_id, catalog_id=catalog_id, dbname=dbname, boto3_session=boto3_session
227227
)
228-
if attrs.kind not in ("postgresql", "postgres"):
228+
if attrs.kind not in ("postgresql", "postgres", "aurora-postgresql"):
229229
raise exceptions.InvalidDatabaseType(
230230
f"Invalid connection type ({attrs.kind}. It must be a postgresql connection.)"
231231
)

0 commit comments

Comments
 (0)