[client-python] fixes

Author: Archidoit

pycti/api/opencti_api_internal_file.py

@@ -5,7 +5,8 @@ def __init__(self, api):
         self.api = api
 
     def delete(self, **kwargs):
-        fileName = kwargs.get("fileName", None)
+        item = kwargs.get("item", None)
+        fileName = self.api.get_attribute_in_extension("id", item)
         if fileName is not None:
             query = """
                 mutation InternalFileDelete($fileName: String) {

pycti/utils/opencti_stix2.py

@@ -2599,23 +2599,8 @@ def element_operation_delete(self, item, operation):
         else:
             # Element is not knowledge we need to use the right api
             stix_helper = self.get_internal_helper().get(item["type"])
-
-            self.opencti.app_logger.info(stix_helper)
-            self.opencti.app_logger.info(
-                "-------- TYPE ----------",
-                {"type": item["type"]},
-            )
-            if item["type"] == "internalfile":
-                self.opencti.app_logger.info(
-                    "-------- internal file ----------",
-                    {"type": item["type"]},
-                )
-                self.opencti.app_logger.info(item)
-                fileName = self.opencti.get_attribute_in_extension("id", item)
-                self.opencti.app_logger.info(fileName)
-                stix_helper.delete(fileName=fileName)
-            elif stix_helper and hasattr(stix_helper, "delete"):
-                stix_helper.delete(id=item["id"])
+            if stix_helper and hasattr(stix_helper, "delete"):
+                stix_helper.delete(id=item["id"], item=item)
             else:
                 raise ValueError(
                     "Delete operation or not found stix helper", {"type": item["type"]}