[import-document] Fix self file is not defined

OpenCTI-Platform · May 26, 2022 · c7f36d9 · c7f36d9
1 parent 849a0a8
commit c7f36d9
Showing 1 changed file with 9 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
@@ -68,6 +68,8 @@ def __init__(self) -> None:
         else:
             raise FileNotFoundError(f"{entity_config_file} was not found")
 
+        self.file = None
+
     def _process_message(self, data: Dict) -> str:
         self.helper.log_info("Processing new message")
         file_name = self._download_import_file(data)
@@ -86,6 +88,7 @@ def _process_message(self, data: Dict) -> str:
 
         # Parse report
         parser = ReportParser(self.helper, entity_indicators, self.observable_config)
+
         if data["file_id"].startswith("import/global"):
             file_data = open(file_name, "rb").read()
             file_data_encoded = base64.b64encode(file_data)
@@ -355,7 +358,9 @@ def _process_parsed_objects(
                 report_types=entity["report_types"],
                 object_refs=observables + entities,
                 allow_custom=True,
-                custom_properties={"x_opencti_files": [self.file]},
+                custom_properties={
+                    "x_opencti_files": [self.file] if self.file is not None else []
+                },
             )
             observables.append(report)
         elif entity is not None:
@@ -375,7 +380,9 @@ def _process_parsed_objects(
                 report_types=["threat-report"],
                 object_refs=observables + entities,
                 allow_custom=True,
-                custom_properties={"x_opencti_files": [self.file]},
+                custom_properties={
+                    "x_opencti_files": [self.file] if self.file is not None else []
+                },
             )
             observables.append(report)
         bundles_sent = []