[import-document] added error log to track bug

OpenCTI-Platform · Jun 6, 2024 · e415eff · e415eff
1 parent 14216f4
commit e415eff
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/internal-import-file/import-document/src/reportimporter/core.py b/internal-import-file/import-document/src/reportimporter/core.py
@@ -387,12 +387,20 @@ def _process_parsing_results(
                 )
                 if len(entity_stix_bundle["objects"]) == 0:
                     raise ValueError("Entity cannot be found or exported")
+
                 entity_stix = [
                     object
                     for object in entity_stix_bundle["objects"]
                     if object["id"] == match[RESULT_FORMAT_MATCH]
-                ][0]
-                entities.append(entity_stix)
+                ]
+                if len(entity_stix) == 0:
+                    log_message = "Couldn't find entity {} bundle {}".format(
+                        match, entity_stix_bundle
+                    )
+                    self.helper.log_error(log_message)
+                    raise ValueError("Entity cannot be found in bundle")
+
+                entities.append(entity_stix[0])
             else:
                 self.helper.log_info("Odd data received: {}".format(match))