[client] black

Author: marieflorescontact

examples/get_all_TA_using_pagination.py

@@ -0,0 +1,18 @@
+# coding: utf-8
+from more_itertools.more import first
+
+from pycti import OpenCTIApiClient
+
+# Variables
+api_url = "http://localhost:4000"
+api_token = "d434ce02-e58e-4cac-8b4c-42bf16748e84"
+
+# OpenCTI initialization
+opencti_api_client = OpenCTIApiClient(api_url, api_token)
+
+# List all TA
+actors = opencti_api_client.threat_actor.list()
+
+# Print
+
+print(actors)

pycti/entities/opencti_threat_actor_group.py

@@ -216,7 +216,9 @@ def list(self, **kwargs) -> dict:
             final_data = final_data + data
             while result["data"]["threatActorsGroup"]["pageInfo"]["hasNextPage"]:
                 after = result["data"]["threatActorsGroup"]["pageInfo"]["endCursor"]
-                self.opencti.app_logger.info("Listing threatActorsGroup", {"after": after})
+                self.opencti.app_logger.info(
+                    "Listing threatActorsGroup", {"after": after}
+                )
                 result = self.opencti.query(
                     query,
                     {
@@ -228,7 +230,9 @@ def list(self, **kwargs) -> dict:
                         "orderMode": order_mode,
                     },
                 )
-                data = self.opencti.process_multiple(result["data"]["threatActorsGroup"])
+                data = self.opencti.process_multiple(
+                    result["data"]["threatActorsGroup"]
+                )
                 final_data = final_data + data
             return final_data
         else:

pycti/entities/opencti_threat_actor_individual.py

@@ -213,11 +213,17 @@ def list(self, **kwargs) -> dict:
         )
         if get_all:
             final_data = []
-            data = self.opencti.process_multiple(result["data"]["threatActorsIndividuals"])
+            data = self.opencti.process_multiple(
+                result["data"]["threatActorsIndividuals"]
+            )
             final_data = final_data + data
             while result["data"]["threatActorsIndividuals"]["pageInfo"]["hasNextPage"]:
-                after = result["data"]["threatActorsIndividuals"]["pageInfo"]["endCursor"]
-                self.opencti.app_logger.info("Listing threatActorsIndividuals", {"after": after})
+                after = result["data"]["threatActorsIndividuals"]["pageInfo"][
+                    "endCursor"
+                ]
+                self.opencti.app_logger.info(
+                    "Listing threatActorsIndividuals", {"after": after}
+                )
                 result = self.opencti.query(
                     query,
                     {
@@ -229,7 +235,9 @@ def list(self, **kwargs) -> dict:
                         "orderMode": order_mode,
                     },
                 )
-                data = self.opencti.process_multiple(result["data"]["threatActorsIndividuals"])
+                data = self.opencti.process_multiple(
+                    result["data"]["threatActorsIndividuals"]
+                )
                 final_data = final_data + data
             return final_data
         else: