Add create and list notification channels (#83)

Author: omerazr

examples/list_notification_channels.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python
+#
+# Post a user event to Sysdig Cloud
+#
+
+import os
+import sys
+import json
+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)
+
+#
+# Post the event
+#
+ok, res = sdclient.list_notification_channels()
+
+#
+# Return the result
+#
+if ok:
+    print(json.dumps(res['notificationChannels'], indent=4))
+else:
+    print(res)
+    sys.exit(1)

sdcclient/_common.py

@@ -108,6 +108,19 @@ def get_n_connected_agents(self):
         data = res.json()
         return [True, data['total']]
 
+    def list_notification_channels(self):
+        '''**Description**
+            List all configured Notification Channels
+
+        **Arguments**
+            none
+
+        **Success Return Value**
+            A JSON representation of all the notification channels
+        '''
+        res = requests.get(self.url + '/api/notificationChannels', headers=self.hdrs, verify=self.ssl_verify)
+        return self._request_result(res)
+
     def get_notification_ids(self, channels=None):
         '''**Description**
             Get an array of all configured Notification Channel IDs, or a filtered subset of them.
@@ -126,7 +139,7 @@ def get_notification_ids(self, channels=None):
         res = requests.get(self.url + '/api/notificationChannels', headers=self.hdrs, verify=self.ssl_verify)
 
         if not self._checkResponse(res):
-            return [False, self.lasterr]
+            return False, self.lasterr
 
         ids = []
 
@@ -184,9 +197,9 @@ def get_notification_ids(self, channels=None):
                                 found = True
                                 ids.append(ch['id'])
             if not found:
-                return [False, "Channel not found: " + str(c)]
+                return False, "Channel not found: " + str(c)
 
-        return [True, ids]
+        return True, ids
 
     def create_email_notification_channel(self, channel_name, email_recipients):
         channel_json = {
@@ -203,13 +216,26 @@ def create_email_notification_channel(self, channel_name, email_recipients):
         res = requests.post(self.url + '/api/notificationChannels', headers=self.hdrs, data=json.dumps(channel_json), verify=self.ssl_verify)
         return self._request_result(res)
 
+    def create_notification_channel(self, channel):
+        channel["id"] = None
+        channel["version"] = None
+        channel["createdOn"] = None
+        channel["modifiedOn"] = None
+        channel_json = {
+            'notificationChannel': channel
+        }
+
+        res = requests.post(self.url + '/api/notificationChannels', headers=self.hdrs, data=json.dumps(channel_json),
+                            verify=self.ssl_verify)
+        return self._request_result(res)
+
     def get_notification_channel(self, id):
 
         res = requests.get(self.url + '/api/notificationChannels/' + str(id), headers=self.hdrs, verify=self.ssl_verify)
         if not self._checkResponse(res):
-            return [False, self.lasterr]
+            return False, self.lasterr
 
-        return [True, res.json()['notificationChannel']]
+        return True, res.json()['notificationChannel']
 
     def update_notification_channel(self, channel):
         if 'id' not in channel:
@@ -224,8 +250,8 @@ def delete_notification_channel(self, channel):
 
         res = requests.delete(self.url + '/api/notificationChannels/' + str(channel['id']), headers=self.hdrs, verify=self.ssl_verify)
         if not self._checkResponse(res):
-            return [False, self.lasterr]
-        return [True, None]
+            return False, self.lasterr
+        return True, None
 
     def get_data_retention_info(self):
         '''**Description**

test/test_monitor_apis.sh

@@ -43,5 +43,6 @@ date; $SCRIPTDIR/../examples/list_users.py $PYTHON_SDC_TEST_MONITOR_API_TOKEN
 date; $SCRIPTDIR/../examples/list_sysdig_captures.py $PYTHON_SDC_TEST_MONITOR_API_TOKEN
 date; $SCRIPTDIR/../examples/create_sysdig_capture.py $PYTHON_SDC_TEST_MONITOR_API_TOKEN $AGENT_HOSTNAME $CAPTURE_NAME 10
 date; $SCRIPTDIR/../examples/notification_channels.py -c $CHANNEL_NAME $PYTHON_SDC_TEST_MONITOR_API_TOKEN
+date; $SCRIPTDIR/../examples/list_notification_channels.py $PYTHON_SDC_TEST_MONITOR_API_TOKEN
 date; $SCRIPTDIR/../examples/user_team_mgmt.py $PYTHON_SDC_TEST_MONITOR_API_TOKEN $TEAM_NAME [email protected]
 date; $SCRIPTDIR/../examples/user_team_mgmt_extended.py $PYTHON_SDC_TEST_MONITOR_API_TOKEN $TEAM_NAME [email protected]