add set headers functions to webhooks property, change errors in ai detection client to custom copyleaks errors

Author: Elazarcl

copyleaks/clients/ai_detection_client.py

@@ -23,9 +23,10 @@
  SOFTWARE.
 '''
 
-from aifc import Error
 import requests
 from copyleaks.consts import Consts
+from copyleaks.exceptions.command_error import CommandError
+from copyleaks.exceptions.under_maintenance_error import UnderMaintenanceError
 from copyleaks.helpers.copyleaks_client_helper import CopyleaksClientHelper
 
 class _AIDetectionClient:
@@ -45,9 +46,9 @@ def __submit(url, auth_token, scan_id, submission):
         if response.ok:
             return response.json()
         elif response.status_code == 503:
-            raise Error()
+            raise UnderMaintenanceError()
         else:
-            raise Error(response)
+            raise CommandError(response)
 
     @staticmethod
     def submit_natural_language(auth_token, scan_id, submission):

copyleaks/models/submit/properties/submit_webhooks.py

@@ -58,3 +58,39 @@ def set_status(self, value):
         assert value
 
         self.status = value
+
+    def get_status_headers(self):
+        '''
+            Get the headers for the status webhook.
+        '''
+        return self.statusHeaders
+
+    def set_status_headers(self, value):
+        '''
+            Parameters: 
+                value: List of list of strings.
+
+            Example:
+                [["header-key1", "header-value1"], ["header-key2", "header-value2"]]
+        '''
+        assert isinstance(value, list)
+        self.statusHeaders = value
+
+    def get_new_result_headers(self):
+        '''
+            Get the headers for the new result webhook.
+        '''
+        return self.newResultHeaders
+
+    def set_new_result_headers(self, value):
+        '''
+            Set the headers for the new result webhook.
+
+            Parameters: 
+                value: List of list of strings.
+
+            Example:
+                [["header-key1", "header-value1"], ["header-key2", "header-value2"]]
+        '''
+        assert isinstance(value, list)
+        self.newResultHeaders = value