Update mysql.py

senorcinco · web-flow · commit a732121b28e0 · 2025-08-27T13:56:33.000-05:00
Add support for aurora-mysql engine
diff --git a/awswrangler/mysql.py b/awswrangler/mysql.py
@@ -168,7 +168,7 @@ def connect(
     attrs: _db_utils.ConnectionAttributes = _db_utils.get_connection_attributes(
         connection=connection, secret_id=secret_id, catalog_id=catalog_id, dbname=dbname, boto3_session=boto3_session
     )
-    if attrs.kind != "mysql":
+    if attrs.kind not in ("mysql", "aurora-mysql"):
         raise exceptions.InvalidDatabaseType(f"Invalid connection type ({attrs.kind}. It must be a MySQL connection.)")
     return pymysql.connect(
         user=attrs.user,

Original file line number	Diff line number	Diff line change
`@@ -168,7 +168,7 @@ def connect(`
`168`	`168`	`attrs: _db_utils.ConnectionAttributes = _db_utils.get_connection_attributes(`
`169`	`169`	`connection=connection, secret_id=secret_id, catalog_id=catalog_id, dbname=dbname, boto3_session=boto3_session`
`170`	`170`	`)`
`171`		`- if attrs.kind != "mysql":`
	`171`	`+ if attrs.kind not in ("mysql", "aurora-mysql"):`
`172`	`172`	`raise exceptions.InvalidDatabaseType(f"Invalid connection type ({attrs.kind}. It must be a MySQL connection.)")`
`173`	`173`	`return pymysql.connect(`
`174`	`174`	`user=attrs.user,`