def on_delete()

in cloudformation/solution-assistant/src/lambda_fn.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def on_delete(event, __):
    # remove sagemaker endpoints
    solution_prefix = event["ResourceProperties"]["SolutionPrefix"]
    endpoint_names = [
        "{}-endpoint".format(solution_prefix),  # make sure it is the same as your endpoint name
    ]
    for endpoint_name in endpoint_names:
        delete_sagemaker_model(endpoint_name)
        delete_sagemaker_endpoint_config(endpoint_name)
        delete_sagemaker_endpoint(endpoint_name)

    # remove files in s3
    output_bucket = event["ResourceProperties"]["S3Bucket"]
    delete_s3_objects(output_bucket)

    # delete buckets
    delete_s3_bucket(output_bucket)

    # delete images from ecr repository
    ecr_repository = event["ResourceProperties"]["ECRRepository"]
    delete_ecr_images(ecr_repository)