fix(iam, set_iam_credentials): utilize force_lowercase connection opt

aws · Aug 16, 2021 · 3348262 · 3348262
1 parent 727b8fc
commit 3348262
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/redshift_connector/iam_helper.py b/redshift_connector/iam_helper.py
@@ -184,6 +184,10 @@ def set_iam_properties(info: RedshiftProperty) -> RedshiftProperty:
                     ClientProtocolVersion.list()
                 )
             )
+
+        if info.db_groups and info.force_lowercase:
+            info.put("db_groups", [group.lower() for group in info.db_groups])
+
         if info.iam is True:
             if info.cluster_identifier is None:
                 raise InterfaceError(
@@ -308,7 +312,10 @@ def set_iam_credentials(info: RedshiftProperty) -> None:
                         info.put("db_user", saml_db_user)
 
                 if (len(info.db_groups) == 0) and (len(db_groups) > 0):
-                    info.put("db_groups", db_groups)
+                    if force_lowercase:
+                        info.db_groups = [group.lower() for group in db_groups]
+                    else:
+                        info.db_groups = db_groups
 
         IamHelper.set_cluster_credentials(provider, info)