How to Setup PagerDuty Alerts From Snowflake Using External Functions (Part 2)

				
					import json
import boto3
import uuid
import requests
 
def lambda_handler(event, context):
 
   # 200 is the HTTP status code for "ok".
   status_code = 200
 
   # The return value will contain an array of arrays (one inner array per input row).
   array_of_rows_to_return = [ ]
  
   url = "https://api.pagerduty.com/incidents"
 
   headers = {"Content-Type": "application/json; charset=utf-8" , "Authorization": "Token token=<enter token from step 2>", "Accept": "application/vnd.pagerduty+json;version=2", "From": "dganiger@phdata.io"}
  
   uuid_value=uuid.uuid4()
 
   data='''{
     "incident": {
       "type": "incident",
       "title": "incident_title",
       "service": {
         "id": "PBQKQ1F",
         "type": "service_reference"
       },
       "urgency": "high",
       "incident_key": "uuid_value",
       "body": {
         "type": "incident_body",
         "details": "incident_details"
       },
       "escalation_policy": {
         "id": "P6P2F89",
         "type": "escalation_policy_reference"
       }
     }
   }'''
  
  
 
   try:
       # From the input parameter named "event", get the body, which contains
       # the input rows.
       event_body = event["body"]
       print(event_body)
       #print(json.dumps(event_body))
      
       # Convert the input from a JSON string into a JSON object.
       payload = json.loads(event_body)
       # This is basically an array of arrays. The inner array contains the
       # row number, and a value for each parameter passed to the function.
  
       rows = payload["data"]
       print(rows)
      
       # For each input row in the JSON object...
       for row in rows:
           # Read the input row number (the output row number will be the same).
           row_number = row[0]
          
           #read the notification message subject
           status = row[1]
           subject = row[2]
           message = row[3]
           print(message)
          
           #Create SNS client and send notification
           client = boto3.client('sns')
           response = client.publish (
               #update this with your newly created SNS topic ARN
               TargetArn= "arn:aws:sns:us-east-1:858366249668:snowflake-alert-sns" ,
               Subject = status + " : " + subject ,
               Message = json.dumps({'default':message}),
               MessageStructure = 'json'
               )
          
           if ( status == 'FAILURE' ):
             data = data.replace("uuid_value", str(uuid_value)).replace("incident_title", subject).replace("incident_details", message)
             print(data)
             try:
               response = requests.post(url, headers=headers, json=json.loads(data))
             except Exception as err:
               print(err)
             print(response)
  
           #Update output value with message as below. This will serve as an acknowledgement that email has been sent.
           output_value = ["Email Notification sent"]
 
           # Put the returned row number and the returned value into an array.
           row_to_return = [row_number, output_value]
 
           # ... and add that array to the main array.
           array_of_rows_to_return.append(row_to_return)
          
       json_compatible_string_to_return = json.dumps({"data" : array_of_rows_to_return})
 
   except Exception as err:
       # 400 implies some type of error.
       status_code = 400
       # Tell caller what this function could not handle.
       json_compatible_string_to_return = event_body
       print(err)
 
   # Return the return value and HTTP status code.
   return {
       'statusCode': status_code,
       'body': json_compatible_string_to_return
   }