from __future__ import print_function
import json
import os
import boto3
import uuid
import random
import math

sagemakerclient = boto3.client('sagemaker')

def handler(event, context):
  if "xgboostresult" in event["training"].keys():
      try:
        endpoint_name=event["input"]["xgb_endpoint_name"]
        sagemakerclient.delete_endpoint(
            EndpointName=endpoint_name
        )
        print(f"deleted endpoint {endpoint_name} successfully!")
      except Exception as e:
          print(e)
      return {
          'message': 'Deleted xgboost endpoint successfully!'
      }
  elif "llresult" in event["training"].keys():
      try:
        endpoint_name=event["input"]["ll_endpoint_name"]
        sagemakerclient.delete_endpoint(
            EndpointName=endpoint_name
        )
        print(f"deleted endpoint {endpoint_name} successfully!")
      except Exception as e:
          print(e)
      return {
          'message': 'Deleted Linear endpoint successfully!'
      }
  else:
      return {
          'message': 'something wrong has happened!'
      }