Supporting new API for notification channels.

Author: davideschiera

examples/create_alert.py

@@ -26,6 +26,20 @@
 #
 sdclient = SdcClient(sdc_token)
 
+#
+# Find notification channels (you need IDs to create an alert).
+#
+notify_channels = [ {'type': 'SLACK', 'channel': 'sysdig-demo2-alerts'},
+                    {'type': 'EMAIL', 'emailRecipients': [u'[email protected]']},
+                    {'type': 'SNS', 'snsTopicARNs': [u'arn:aws:sns:us-east-1:273107874544:alarms-stg']}
+                    ]
+res = sdclient.get_notification_ids(notify_channels)
+if not res[0]:
+    print "Could not get IDs and hence not creating the alert"
+    sys.exit(-1)
+
+notification_channel_ids = res[1]
+
 #
 # Create the alert.
 #
@@ -37,7 +51,7 @@
                             ['host.mac', 'proc.name'], # Segmentation. We want to check this metric for every process on every machine.
                             'ANY', # in case there is more than one tomcat process, this alert will fire when a single one of them crosses the 80% threshold.
                             'proc.name = "tomcat"', # Filter. We want to receive a notification only if the name of the process meeting the condition is 'tomcat'.
-                            ['EMAIL'], # Notification target. We want an email to be sent. Alerts email recipients can be defined here: https://app.sysdigcloud.com/#/settings/notifications
+                            notification_channel_ids,
                             False) # This alert will be disabled when it's created.
 
 #

sdcclient/_client.py

@@ -102,8 +102,39 @@ def update_notification_resolution(self, notification, resolved):
             return [False, self.lasterr]
         return [True, res.json()]
 
+    def get_notification_ids(self, channels):
+        res = requests.get(self.url + '/api/notificationChannels', headers=self.hdrs)
+
+        if not self.__checkResponse(res):
+            return [False, self.lasterr]
+
+        # Should try and improve this M * N lookup
+        ids = []
+        for ch in res.json()["notificationChannels"]:
+            for c in channels:
+                if c['type'] == ch['type']:
+                    print c, ch
+                    if c['type'] == 'SNS':
+                        opt = ch['options']
+                        if set(opt['snsTopicARNs']) == set(c['snsTopicARNs']):
+                            ids.append(ch['id'])
+                    elif c['type'] == 'EMAIL':
+                        opt = ch['options']
+                        if set(c['emailRecipients']) == set(opt['emailRecipients']):
+                            ids.append(ch['id'])
+                    elif c['type'] == 'PAGER_DUTY':
+                        opt = ch['options']
+                        if opt['account'] == c['account'] and opt['serviceName'] == c['serviceName']:
+                            ids.append(ch['id'])
+                    elif c['type'] == 'SLACK':
+                        opt = ch['options']
+                        if opt['channel'] == c['channel']:
+                            ids.append(ch['id'])
+
+        return [True, ids]
+                        
     def create_alert(self, name, description, severity, for_atleast_s, condition, segmentby=[],
-                     segment_condition='ANY', user_filter='', notify='', enabled=True, annotations={}):
+                     segment_condition='ANY', user_filter='', notify=None, enabled=True, annotations={}):
         #
         # Get the list of alerts from the server
         #
@@ -143,8 +174,8 @@ def create_alert(self, name, description, severity, for_atleast_s, condition, se
         if annotations != None and annotations != {}:
             alert_json['alert']['annotations'] = annotations
 
-        if notify != None and notify != []:
-            alert_json['alert']['notify'] = notify
+        if notify != None:
+            alert_json['alert']['notificationChannelIds'] = notify
 
         #
         # Create the new alert