minor update for retry

aws-samples · Jul 15, 2024 · 7fec328 · 7fec328
1 parent 8faca07
commit 7fec328
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/aws_sra_examples/solutions/firewall_manager/firewall_manager_org/lambda/src/app.py b/aws_sra_examples/solutions/firewall_manager/firewall_manager_org/lambda/src/app.py
@@ -109,6 +109,10 @@ def associate_admin_account(delegated_admin_account_id: str) -> None:
                 except botocore.exceptions.ClientError as error:
                     LOGGER.info(f"Attempt {i_retry} - error associating admin account: {error.response['Error']['Message']}")
                     associated = False
+                if associated is True:
+                    break
+                else:
+                    i_retry += 1
             if associated is False:
                 LOGGER.error("Unable to associate admin account.")
                 raise ValueError("Unable to associate admin account.")