def on_delete()

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

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

def on_delete(event, __):
    # remove sagemaker endpoint
    solution_prefix = event["ResourceProperties"]["SolutionPrefix"]
    endpoint_names = [
        "{}-summarization".format(solution_prefix),
        "{}-question-answering".format(solution_prefix),
        "{}-entity-recognition".format(solution_prefix),
        "{}-relationship-extraction".format(solution_prefix)
    ]
    for endpoint_name in endpoint_names:
        delete_sagemaker_model(endpoint_name)
        delete_sagemaker_endpoint_config(endpoint_name)
        delete_sagemaker_endpoint(endpoint_name)
    # remove sagemaker endpoint config
    # remove files in s3
    s3_bucket = event["ResourceProperties"]["S3BucketName"]
    delete_s3_objects(s3_bucket)