def lambda_handler()

in src/lambda_codebase/wait_until_complete.py [0:0]

• 44 lines of code
• 5 McCabe index (conditional complexity)

def lambda_handler(event, _):
    """Main Lambda Entry point
    """
    sts = STS()
    account_id = event.get('account_id')
    partition = get_partition(REGION_DEFAULT)
    cross_account_access_role = event.get('cross_account_access_role')

    role = sts.assume_cross_account_role(
        f'arn:{partition}:iam::{account_id}:role/{cross_account_access_role}',
        'master'
    )

    s3 = S3(REGION_DEFAULT, S3_BUCKET)

    for region in list(set([event['deployment_account_region']] + event['regions'])):

        cloudformation = CloudFormation(
            region=region,
            deployment_account_region=event['deployment_account_region'],
            role=role,
            wait=False,
            stack_name=None,
            s3=s3,
            s3_key_path=event['ou_name'],
            account_id=account_id
        )

        status = cloudformation.get_stack_status()

        if status in ('CREATE_IN_PROGRESS', 'UPDATE_IN_PROGRESS'):
            raise RetryError(f"CloudFormation Stack status: {status}")

        if status in (
                'CREATE_FAILED',
                'ROLLBACK_FAILED',
                'DELETE_FAILED',
                'UPDATE_ROLLBACK_FAILED',
                'ROLLBACK_IN_PROGRESS',
                'ROLLBACK_COMPLETE'
            ):
            raise Exception(
                f"Account Bootstrap Failed - Account: {account_id} "
                f"Region: {region} Status: {status}"
            )

        if event.get('is_deployment_account'):
            update_deployment_account_output_parameters(
                deployment_account_region=event['deployment_account_region'],
                region=region,
                deployment_account_role=role,
                cloudformation=cloudformation
            )

    return event