Resolve notifications API

Author: gianlucaborello

.travis.yml

@@ -15,6 +15,7 @@ script:
 - examples/get_data_simple.py XXX
 - examples/list_alerts.py XXX
 - examples/list_alert_notifications.py XXX
+- examples/resolve_alert_notifications.py XXX
 - examples/list_dashboards.py XXX
 - examples/list_hosts.py XXX
 - examples/list_metrics.py XXX
@@ -36,6 +37,7 @@ script:
 - examples/get_data_simple.py XXX
 - examples/list_alerts.py XXX
 - examples/list_alert_notifications.py XXX
+- examples/resolve_alert_notifications.py XXX
 - examples/list_dashboards.py XXX
 - examples/list_hosts.py XXX
 - examples/list_metrics.py XXX

README.md

@@ -104,7 +104,7 @@ Deletes an alert.
 - **alert**: the alert object as returned by `get_alerts()`.
 
 **Success Return Value**  
-A string with the number of alerts deleted.  
+`None`.  
 **Example**  
 [examples/delete_alert.py](examples/delete_alert.py).  
 
@@ -119,11 +119,11 @@ Deletes a dashboard.
 **Example**  
 [examples/delete_dashboard.py](examples/delete_dashboard.py).  
 
-#### `delete_event(self, event_id)`  
+#### `delete_event(self, event)`  
 **Description**  
-Deletes an event given its id.  
+Deletes an event.  
 **Arguments**  
-- **event_id**: id of the event to be deleted.
+- **event**: the event object as returned by `get_events()`.
 
 **Success Return Value**  
 `None`.  

examples/delete_event.py

@@ -37,9 +37,9 @@
 # Delete the first event among the returned ones
 #
 for event in res[1]['events']:
-	print "Deleting event " + json.dumps(event)
+	print "Deleting event " + event['name']
 
-	res = sdclient.delete_event(event['id'])
+	res = sdclient.delete_event(event)
 
 	if not res[0]:
 		print res[1]

examples/list_metrics.py

@@ -37,4 +37,5 @@
     print res[1]
     sys.exit(1)
 
-print data
+for metric_id, metric in data.iteritems():
+	print "Metric name: " + metric_id + ", type: " + metric['type']

examples/resolve_alert_notifications.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+#
+# Resolve alert notifications from Sysdig Cloud
+#
+
+import os
+import sys
+import time
+sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])), '..'))
+from sdcclient import SdcClient
+
+#
+# Parse arguments
+#
+if len(sys.argv) != 2:
+    print 'usage: %s <sysdig-token>' % sys.argv[0]
+    print 'You can find your token at https://app.sysdigcloud.com/#/settings/user'
+    sys.exit(1)
+
+sdc_token = sys.argv[1]
+
+#
+# Instantiate the SDC client
+#
+sdclient = SdcClient(sdc_token)
+
+#
+# Get the unresolved notifications in the last day
+#
+res = sdclient.get_notifications(from_ts=int(time.time()-86400), to_ts=int(time.time()), resolved=False)
+
+if not res[0]:
+    print res[1]
+    sys.exit(1)
+
+#
+# Resolve them
+#
+notifications = res[1]['notifications']
+
+print "Resolving " + str(len(notifications)) + " notifications"
+for notification in notifications:
+    res = sdclient.update_notification_resolution(notification, True)
+    if not res[0]:
+        print res[1]
+        sys.exit(1)

sdcclient/_client.py

@@ -15,6 +15,7 @@ def __init__(self, token="", sdc_url='https://app.sysdigcloud.com'):
     def __checkResponse(self, r):
         if r.status_code >= 300:
             errorcode = r.status_code
+            self.lasterr = None
 
             try:
                 j = r.json()
@@ -25,8 +26,14 @@ def __checkResponse(self, r):
             if 'errors' in j:
                 if 'message' in j['errors'][0]:
                     self.lasterr = j['errors'][0]['message']
-                else:
-                    self.lasterr = j['errors'][0]['reason']
+
+                if 'reason' in j['errors'][0]:
+                    if self.lasterr is not None:
+                        self.lasterr += ' '
+                    else:
+                        self.lasrerr = ''
+
+                    self.lasterr += j['errors'][0]['reason']
             elif 'message' in j:
                 self.lasterr = j['message']
             else:
@@ -75,6 +82,18 @@ def get_notifications(self, from_ts, to_ts, state=None, resolved=None):
             return [False, self.lasterr]
         return [True, r.json()]
 
+    def update_notification_resolution(self, notification, resolved):
+        if 'id' not in notification:
+            return [False, "Invalid notification format"]
+
+        notification['resolved'] = resolved
+        data = {'notification': notification}
+
+        r = requests.put(self.url + '/api/notifications/' + str(notification['id']), headers=self.hdrs, data=json.dumps(data))
+        if not self.__checkResponse(r):
+            return [False, self.lasterr]
+        return [True, r.json()]
+
     def create_alert(self, name, description, severity, for_atleast_s, condition, segmentby=[],
                      segment_condition='ANY', filter='', notify='', enabled=True, annotations={}):
         #
@@ -530,8 +549,11 @@ def get_events(self, name=None, from_ts=None, to_ts=None, tags=None):
             return [False, self.lasterr]
         return [True, r.json()]
 
-    def delete_event(self, event_id):
-        r = requests.delete(self.url + '/api/events/' + str(event_id), headers=self.hdrs)
+    def delete_event(self, event):
+        if 'id' not in event:
+            return [False, "Invalid event format"]
+
+        r = requests.delete(self.url + '/api/events/' + str(event['id']), headers=self.hdrs)
         if not self.__checkResponse(r):
             return [False, self.lasterr]
         return [True, None]