Skip to content

Commit a875e86

Browse files
committed
data api redshift-serverless: move workgroup_name arg to avoid breaking existing code
1 parent 4fcda80 commit a875e86

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

awswrangler/data_api/redshift.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@ class RedshiftDataApi(connector.DataApiConnector):
1717
----------
1818
cluster_id: str
1919
Id for the target Redshift cluster - only required if `workgroup_name` not provided.
20-
workgroup_name: str
21-
Name for the target serverless Redshift workgroup - only required if `cluster_id` not provided.
2220
database: str
2321
Target database name.
22+
workgroup_name: str
23+
Name for the target serverless Redshift workgroup - only required if `cluster_id` not provided.
2424
secret_arn: str
2525
The ARN for the secret to be used for authentication - only required if `db_user` not provided.
2626
db_user: str
@@ -38,8 +38,8 @@ class RedshiftDataApi(connector.DataApiConnector):
3838
def __init__(
3939
self,
4040
cluster_id: str = "",
41-
workgroup_name: str = "",
4241
database: str = "",
42+
workgroup_name: str = "",
4343
secret_arn: str = "",
4444
db_user: str = "",
4545
sleep: float = 0.25,
@@ -48,8 +48,8 @@ def __init__(
4848
boto3_session: Optional[boto3.Session] = None,
4949
) -> None:
5050
self.cluster_id = cluster_id
51-
self.workgroup_name = workgroup_name
5251
self.database = database
52+
self.workgroup_name = workgroup_name
5353
self.secret_arn = secret_arn
5454
self.db_user = db_user
5555
self.client: boto3.client = _utils.client(service_name="redshift-data", session=boto3_session)
@@ -186,8 +186,8 @@ class RedshiftDataApiTimeoutException(Exception):
186186

187187
def connect(
188188
cluster_id: str = "",
189-
workgroup_name: str = "",
190189
database: str = "",
190+
workgroup_name: str = "",
191191
secret_arn: str = "",
192192
db_user: str = "",
193193
boto3_session: Optional[boto3.Session] = None,
@@ -199,10 +199,10 @@ def connect(
199199
----------
200200
cluster_id: str
201201
Id for the target Redshift cluster - only required if `workgroup_name` not provided.
202-
workgroup_name: str
203-
Name for the target serverless Redshift workgroup - only required if `cluster_id` not provided.
204202
database: str
205203
Target database name.
204+
workgroup_name: str
205+
Name for the target serverless Redshift workgroup - only required if `cluster_id` not provided.
206206
secret_arn: str
207207
The ARN for the secret to be used for authentication - only required if `db_user` not provided.
208208
db_user: str
@@ -218,8 +218,8 @@ def connect(
218218
"""
219219
return RedshiftDataApi(
220220
cluster_id=cluster_id,
221-
workgroup_name=workgroup_name,
222221
database=database,
222+
workgroup_name=workgroup_name,
223223
secret_arn=secret_arn,
224224
db_user=db_user,
225225
boto3_session=boto3_session,

0 commit comments

Comments
 (0)