[client] Try/expect expectations

Samuel Hassine · Samuel Hassine · commit eaa3f8a9c356 · 2020-12-15T00:09:59.000+01:00
diff --git a/pycti/api/opencti_api_work.py b/pycti/api/opencti_api_work.py
@@ -44,7 +44,10 @@ def report_expectation(self, work_id: str, error):
                 }
             }
            """
-        self.api.query(query, {"id": work_id, "error": error})
+        try:
+            self.api.query(query, {"id": work_id, "error": error})
+        except:
+            self.api.log("error", "Cannot report expectation")
 
     def add_expectations(self, work_id: str, expectations: int):
         logging.info(
@@ -59,7 +62,10 @@ def add_expectations(self, work_id: str, expectations: int):
                 }
             }
            """
-        self.api.query(query, {"id": work_id, "expectations": expectations})
+        try:
+            self.api.query(query, {"id": work_id, "expectations": expectations})
+        except:
+            self.api.log("error", "Cannot report expectation")
 
     def initiate_work(self, connector_id: str, friendly_name: str) -> str:
         logging.info("Initiate work for " + connector_id)

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,10 @@ def report_expectation(self, work_id: str, error):`
`44`	`44`	`}`
`45`	`45`	`}`
`46`	`46`	`"""`
`47`		`- self.api.query(query, {"id": work_id, "error": error})`
	`47`	`+ try:`
	`48`	`+ self.api.query(query, {"id": work_id, "error": error})`
	`49`	`+ except:`
	`50`	`+ self.api.log("error", "Cannot report expectation")`
`48`	`51`
`49`	`52`	`def add_expectations(self, work_id: str, expectations: int):`
`50`	`53`	`logging.info(`
`@@ -59,7 +62,10 @@ def add_expectations(self, work_id: str, expectations: int):`
`59`	`62`	`}`
`60`	`63`	`}`
`61`	`64`	`"""`
`62`		`- self.api.query(query, {"id": work_id, "expectations": expectations})`
	`65`	`+ try:`
	`66`	`+ self.api.query(query, {"id": work_id, "expectations": expectations})`
	`67`	`+ except:`
	`68`	`+ self.api.log("error", "Cannot report expectation")`
`63`	`69`
`64`	`70`	`def initiate_work(self, connector_id: str, friendly_name: str) -> str:`
`65`	`71`	`logging.info("Initiate work for " + connector_id)`