feat(access_key): Implement teamId in create_access_key and implement update_access_key (#251)

Signed-off-by: Daniele De Lorenzi <[email protected]>

Author: dark-vex

examples/create_access_keys.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python
+#
+# List all the access keys in a Sysdig Monitor environment. The token you provide must
+# have Admin rights.
+#
+
+import sys
+
+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')
+    print('For this script to work, the user for the token must have Admin rights')
+    sys.exit(1)
+
+sdc_token = sys.argv[1]
+
+# Maximum number of agents allowed to connect for this access key. Set to '' if not required
+agent_limit = ''
+# Number of agent licenses that are ALWAYS available to this access key. This directly counts against the maximum number of available licenses. Set to '' if not required.
+agent_reserved = ''
+# Team ID to which to assign the access key. Team ID must be valid. Set to '' if not required.
+team_id = ''
+
+
+#
+# Instantiate the SDC client
+#
+sdclient = SdcClient(sdc_token, 'https://app.sysdigcloud.com')
+
+#
+# Get the configuration
+#
+ok, res = sdclient.create_access_key(
+    agent_limit,
+    agent_reserved,
+    team_id)
+
+if ok:
+    print('Access Key: {}\nTeam ID: {}\nAgent Limit: {}\nAgent Reserved: {}\n==========='.format(res['customerAccessKey']['accessKey'], res['customerAccessKey']['teamId'], res['customerAccessKey']['limit'], res['customerAccessKey']['reservation']))
+else:
+    print(res)
+    sys.exit(1)

examples/update_access_keys.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python
+#
+# List all the access keys in a Sysdig Monitor environment. The token you provide must
+# have Admin rights.
+#
+
+import sys
+
+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')
+    print('For this script to work, the user for the token must have Admin rights')
+    sys.exit(1)
+
+sdc_token = sys.argv[1]
+
+# Access Key that needs to be updated
+accessKey = ''
+# Maximum number of agents allowed to connect for this access key. Set to '' if not required
+agent_limit = ''
+# Number of agent licenses that are ALWAYS available to this access key. This directly counts against the maximum number of available licenses. Set to '' if not required.
+agent_reserved = ''
+# Team ID to which to assign the access key. Team ID must be valid. Set to '' if not required.
+team_id = ''
+
+
+#
+# Instantiate the SDC client
+#
+sdclient = SdcClient(sdc_token, 'https://app.sysdigcloud.com')
+
+#
+# Get the configuration
+#
+if accessKey:
+    ok, res = sdclient.update_access_key(
+        accessKey,
+        agent_limit,
+        agent_reserved,
+        team_id)
+else:
+    print('Please specify the Access Key that you would like to be updated')
+    sys.exit(1)
+
+if ok:
+    print('Access Key: {}\nTeam ID: {}\nAgent Limit: {}\nAgent Reserved: {}\n==========='.format(res['customerAccessKey']['accessKey'], res['customerAccessKey']['teamId'], res['customerAccessKey']['limit'], res['customerAccessKey']['reservation']))
+else:
+    print(res)
+    sys.exit(1)

sdcclient/_common.py

@@ -1044,15 +1044,42 @@ def list_access_keys(self):
         res = self.http.get(self.url + '/api/customer/accessKeys', headers=self.hdrs, verify=self.ssl_verify)
         return self._request_result(res)
 
-    def create_access_key(self):
+    def create_access_key(self, agent_limit=None, agent_reserved=None, team_id=None):
         '''
         **Description**
             Create a new access key for Sysdig Monitor/Secure
 
         **Reslut**
             The access keys object
         '''
-        res = self.http.post(self.url + '/api/customer/accessKeys', headers=self.hdrs, verify=self.ssl_verify)
+        access_key_payload = {
+            "customerAccessKey": {
+                "limit": agent_limit,
+                "reservation": agent_reserved,
+                "teamId": team_id
+            }
+        }
+
+        res = self.http.post(self.url + '/api/customer/accessKeys', headers=self.hdrs, verify=self.ssl_verify, data=json.dumps(access_key_payload))
+        return self._request_result(res)
+
+    def update_access_key(self, access_key, agent_limit=None, agent_reserved=None, team_id=None):
+        '''
+        **Description**
+            Create a new access key for Sysdig Monitor/Secure
+
+        **Reslut**
+            The access keys object
+        '''
+        access_key_payload = {
+            "customerAccessKey": {
+                "limit": agent_limit,
+                "reservation": agent_reserved,
+                "teamId": team_id
+            }
+        }
+
+        res = self.http.put(self.url + '/api/customer/accessKeys/' + access_key, headers=self.hdrs, verify=self.ssl_verify, data=json.dumps(access_key_payload))
         return self._request_result(res)
 
     def disable_access_key(self, access_key):