Add functionality to login with api-key

Author: chkp-orso

cpapi/mgmt_api.py

@@ -139,8 +139,15 @@ def save_debug_data(self):
             out_file = open(self.debug_file, 'w+')
             out_file.write(json.dumps(self.api_calls, indent=4, sort_keys=True))
 
-    def login(self, username, password, continue_last_session=False, domain=None, read_only=False,
-              payload=None):
+    def _build_payload(self, username=None, password=None, api_key=None):
+        payload = {}
+        if username: payload['user'] = username
+        if password: payload['password'] = password
+        if api_key: payload['api-key'] = api_key
+        return payload
+
+    def login(self, username=None, password=None, continue_last_session=False, domain=None, read_only=False,
+              payload=None, api_key=None):
         """
         performs a 'login' API call to the management server
 
@@ -155,7 +162,7 @@ def login(self, username, password, continue_last_session=False, domain=None, re
         :returns: APIResponse object
         :side-effects: updates the class's uid and server variables
         """
-        credentials = {"user": username, "password": password}
+        credentials = self._build_payload(username, password, api_key)
 
         if self.context == "web_api":
             credentials.update({"continue-last-session": continue_last_session,
@@ -310,7 +317,10 @@ def api_call(self, command, payload=None, sid=None, wait_for_task=True):
         # would not appear as plaintext in the debug file.
         if command == "login":
             json_data = compatible_loads(_data)
-            json_data["password"] = "****"
+            if "password" in json_data:
+                json_data["password"] = "****"
+            if "api-key" in json_data:
+                json_data["api-key"] = "****"
             _data = json.dumps(json_data)
 
         # Store the request and the reply (for debug purpose).