chore(s3): create class and refactor

prowler/lib/outputs/compliance/compliance_output.py

@@ -1,5 +1,5 @@
 from csv import DictWriter
-from io import TextIOWrapper
+from pathlib import Path
 from typing import List
 
 from prowler.lib.check.compliance_models import ComplianceBaseModel
@@ -25,17 +25,21 @@
         create_file_descriptor: Method to create a file descriptor for writing data to a file.
     """
 
-    _data: list
-    _file_descriptor: TextIOWrapper
-
     def __init__(
         self,
         findings: List[Finding],
         compliance: ComplianceBaseModel,
         create_file_descriptor: bool = False,
         file_path: str = None,
+        file_extension: str = "",
     ) -> None:
         self._data = []
+
+        if not file_extension and file_path:
+            self._file_extension = "".join(Path(file_path).suffixes)
+        if file_extension:
+            self._file_extension = file_extension
+
         if findings:
             # Get the compliance name of the model
             compliance_name = (

prowler/lib/outputs/output.py

@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 from io import TextIOWrapper
-from os import path
+from pathlib import Path
 from typing import List
 
 from prowler.lib.logger import logger
@@ -40,13 +40,13 @@ def __init__(
         self._data = []
 
         if not file_extension and file_path:
-            _, self._file_extension = path.splitext(file_path)
+            self._file_extension = "".join(Path(file_path).suffixes)
         if file_extension:
             self._file_extension = file_extension
 
         if findings:
             self.transform(findings)
-            if create_file_descriptor:
+            if create_file_descriptor and file_path:
                 self.create_file_descriptor(file_path)
 
     @property
@@ -73,7 +73,7 @@ def transform(self, findings: List[Finding]):
     def batch_write_data_to_file(self) -> None:
         raise NotImplementedError
 
-    def create_file_descriptor(self, file_path) -> None:
+    def create_file_descriptor(self, file_path: str) -> None:
         """
         Creates a file descriptor for writing data to a file.