[client] Adapt client python to limit monitoring impact (#944)

Author: richard-julien

pycti/api/opencti_api_client.py

@@ -145,13 +145,11 @@ def __init__(self, url, token, log_level="info", ssl_verify=False, proxies={}):
             )
 
     def set_applicant_id_header(self, applicant_id):
-        """set the request header with the specified token
-
-        :param applicant_id: OpenCTI caller id
-        :type applicant_id: str
-        """
         self.request_headers["opencti-applicant-id"] = applicant_id
 
+    def set_work_id_header(self, work_id):
+        self.request_headers["opencti-work-id"] = work_id
+
     def set_retry_number(self, retry_number):
         self.request_headers["opencti-retry-number"] = (
             "" if retry_number is None else str(retry_number)

pycti/api/opencti_api_work.py

@@ -12,9 +12,7 @@ def to_received(self, work_id: str, message: str):
         query = """
             mutation workToReceived($id: ID!, $message: String) {
                 workEdit(id: $id) {
-                    toReceived (message: $message) {
-                      id
-                    }
+                    toReceived (message: $message)
                 }
             }
            """
@@ -25,9 +23,7 @@ def to_processed(self, work_id: str, message: str, in_error: bool = False):
         query = """
             mutation workToProcessed($id: ID!, $message: String, $inError: Boolean) {
                 workEdit(id: $id) {
-                    toProcessed (message: $message, inError: $inError) {
-                      id
-                    }
+                    toProcessed (message: $message, inError: $inError)
                 }
             }
            """
@@ -38,9 +34,7 @@ def report_expectation(self, work_id: str, error):
         query = """
             mutation reportExpectation($id: ID!, $error: WorkErrorInput) {
                 workEdit(id: $id) {
-                    reportExpectation(error: $error) {
-                      id
-                    }
+                    reportExpectation(error: $error)
                 }
             }
            """
@@ -56,9 +50,7 @@ def add_expectations(self, work_id: str, expectations: int):
         query = """
             mutation addExpectations($id: ID!, $expectations: Int) {
                 workEdit(id: $id) {
-                    addExpectations(expectations: $expectations) {
-                      id
-                    }
+                    addExpectations(expectations: $expectations)
                 }
             }
            """