slack support for reporting error

Author: prashantramangupta

constant.py

@@ -8,4 +8,13 @@
                }
     }
 }
+SLACK_HOOK = {
+    'hostname' : '',
+    'port': 443,
+    'path': '',
+    'method': 'POST',
+    'headers': {
+        'Content-Type': 'application/json'
+    }
+}
 METRICS_NETWORK_ID  = 999

platform-usage-authorizer/lambda_handler.py

@@ -2,28 +2,34 @@
 import os
 
 from authorizer import Token
+from utils import Utils
 
+obj_util = Utils()
 
 def request_handler(event, context):
     if 'path' not in event:
         return get_response("400", {"status": "failed", "error": "Bad Request"})
     try:
-        path = event['path'].lower()
         data = None
+        payload_dict = None
+        path = event['path'].lower()
         if "/event" == path:
             payload_dict = event['headers']
             print("Processing [" + str(path) + "] with body [" + str(payload_dict) + "]")
             token_instance = Token()
             data = token_instance.validate_token(daemon_id=payload_dict['x-daemonid'],
                                                  token=payload_dict['x-token'])
-            print(data)
         else:
             return get_response(500, "Invalid URL path.")
-        response = get_lambda_authorizer_response_format(event=event, allow=data.get('validated', False))
-        return response
+        return get_lambda_authorizer_response_format(event=event, allow=data.get('validated', False))
     except Exception as e:
-        print(repr(e))
-        return get_lambda_authorizer_response_format(event=event, allow=False)
+        err_msg = {"status": "failed",
+                   "error": repr(e),
+                   "api": event['path'],
+                   "payload": payload_dict,
+                   "type": "authorize"}
+        obj_util.report_slack(1, str(err_msg))
+        return get_response(500, err_msg)
 
 
 # Generate response JSON that API gateway expects from the lambda function

platform-usage-authorizer/requirement.txt

@@ -1 +1,2 @@
 pymysql==0.9.2
+requests==2.20.1

platform-usage-process/lambda_handler.py

@@ -2,15 +2,19 @@
 
 from metrics import Metrics
 from register import Token
+from utils import Utils
+
+obj_util = Utils()
 
 
 def request_handler(event, context):
     if 'path' not in event:
         return get_response("400", {"status": "failed",
                                     "error": "Bad Request"})
     try:
-        path = event['path'].lower()
         data = None
+        payload_dict = None
+        path = event['path'].lower()
         payload_dict = payload_check(payload=event['body'], path=path)
         if "/register" == path:
             token_instance = Token()
@@ -25,7 +29,6 @@ def request_handler(event, context):
         else:
             return get_response(500, "Invalid URL path.")
 
-
         if data is None:
             response = get_response("400", {"status": "failed",
                                             "error": "Bad Request",
@@ -41,11 +44,13 @@ def request_handler(event, context):
 
         return response
     except Exception as e:
-        response = get_response(500, {"status": "failed",
-                                      "error": repr(e),
-                                      "api": event['path'],
-                                      "payload": payload_dict})
-        return response
+        err_msg = {"status": "failed",
+                   "error": repr(e),
+                   "api": event['path'],
+                   "payload": payload_dict,
+                   "type": "process"}
+        obj_util.report_slack(1, str(err_msg))
+        return get_response(500, err_msg)
 
 
 def payload_check(payload, path):

platform-usage-process/requirement.txt

@@ -1 +1,2 @@
 pymysql==0.9.2
+requests==2.20.1

utils.py

@@ -0,0 +1,53 @@
+import json
+import datetime
+import decimal
+import requests
+
+from constant import SLACK_HOOK
+IGNORED_LIST = ['row_id', 'row_created', 'row_updated']
+
+class Utils:
+    def __init__(self):
+        self.msg_type = {
+            0 : 'info:: ',
+            1 : 'err:: '
+        }
+
+    def report_slack(self, type, slack_msg):
+        url = SLACK_HOOK['hostname'] + SLACK_HOOK['path']
+        prefix = self.msg_type.get(type, "")
+        print(url)
+        payload = {"channel": "#contract-index-alerts",
+                   "username": "webhookbot",
+                   "text": prefix + slack_msg,
+                   "icon_emoji": ":ghost:"
+                   }
+
+        resp = requests.post(url=url, data=json.dumps(payload))
+        print(resp.status_code, resp.text)
+
+    def clean(self, value_list):
+        for value in value_list:
+            self.clean_row(value)
+
+    def clean_row(self, row):
+        for item in IGNORED_LIST:
+            del row[item]
+
+        for key in row:
+            if isinstance(row[key], decimal.Decimal) or isinstance(row[key], datetime.datetime):
+                row[key] = str(row[key])
+            elif isinstance(row[key], bytes):
+                if row[key] == b'\x01':
+                    row[key] = 1
+                elif row[key] == b'\x00':
+                    row[key] = 0
+                else:
+                    raise Exception("Unsupported bytes object. Key " + str(key) + " value " + str(row[key]))
+
+        return row
+
+    def remove_http_https_prefix(self, url):
+        url = url.replace("https://","")
+        url = url.replace("http://","")
+        return url