initial commit

core/utils/storage.py

@@ -0,0 +1,33 @@
+from typing import Optional, Union
+
+from config.env import env
+
+
+class Storage:
+
+    storages = ["AWS", "MINIO", "FILE", "STATIC"]
+
+    def __init__(self, storage: Union[str], storage_type: Union[str] = "default") -> None:
+        self.storage = storage
+        self.sorage_type = storage_type
+        if storage not in self.storages:
+            raise ValueError(f"Invalid storage type: {storage}")
+
+    def get_backend(self) -> Optional[str]:
+        match self.storage:  
+            case "AWS" | "MINIO": 
+                return "storages.backends.s3boto3.S3Boto3Storage"
+            case "FILE":
+                return "django.core.files.storage.FileSystemStorage"
+            case "STATIC":
+                return "django.contrib.staticfiles.storage.StaticFilesStorage"
+
+    def get_options(self) -> Optional[dict[str, object]]:
+        match self.storage:
+            case "AWS" | "MINIO":
+                if self.sorage_type == "default":
+                    return {"bucket_name": env.str("STORAGE_BUCKET_MEDIA")}
+                elif self.sorage_type == "static":
+                    return {"bucket_name": env.str("STORAGE_BUCKET_STATIC")}
+            case _:
+                return {}