Adds New Delete File Endpoint and Refactors Endpoints, Adds VirusChecker

storage_service/depends/depend_s3_service.py

@@ -1,6 +1,6 @@
 from storage_service.config.config_s3 import get_config_s3
-from storage_service.service.amazon_s3_service import AmazonS3Service
-from storage_service.service.storage_service import StorageService
+from storage_service.service.storage.amazon_s3_service import AmazonS3Service
+from storage_service.service.storage.storage_service import StorageService
 from storage_service.utils.enums.storage_type import StorageType
 
 from dotenv import load_dotenv

storage_service/depends/depend_virus_checker_service.py

@@ -0,0 +1,26 @@
+import os
+from functools import cache
+
+from storage_service.config.config_virus_checker import get_virus_checker_api_key
+from storage_service.service.virus_checker.virus_total_service import VirusTotalService
+from storage_service.service.virus_checker.virus_checker_service import VirusCheckerService
+
+from dotenv import load_dotenv
+
+from storage_service.utils.enums.virus_checker_type import VirusCheckerType
+
+
+@cache
+def dependency_virus_checker_service() -> VirusCheckerService:
+    load_dotenv()
+
+    virus_checker_config = get_virus_checker_api_key()
+
+    if not virus_checker_config["api_key"]:
+        raise RuntimeError("Virus Checker API Key not found")
+
+    virus_checker_type_var = os.environ.get("VIRUS_CHECKER_TYPE")
+    if VirusCheckerType(virus_checker_type_var) == VirusCheckerType.TOTAL_VIRUS:
+        return VirusTotalService(**get_virus_checker_api_key())
+
+    raise RuntimeError("Invalid Virus Checker Type")