Implements Better Abstraction on Top of StorageService and File Handlers

Adds Expiration Time to Config

Reformats Project

resize_image_service/utils/enums/file_type.py

@@ -4,6 +4,3 @@ from enum import Enum
 class FileType(Enum):
     PNG = "png"
     JPEG = "jpeg"
-
-
-CONTENT_TYPE = {FileType.PNG: "image/png", FileType.JPEG: "image/jpeg"}

resize_image_service/utils/enums/storage_type.py

@@ -0,0 +1,5 @@
+from enum import Enum
+
+
+class StorageType(Enum):
+    S3_STORAGE = "s3"

resize_image_service/utils/file_handler/__init__.py

@@ -0,0 +1,9 @@
+from resize_image_service.utils.enums.file_type import FileType
+from resize_image_service.utils.file_handler.handlers.image_handler import (
+    image_handler,
+)
+
+FILE_HANDLER = {
+    FileType.PNG: {"content_type": "image/png", "handler": image_handler},
+    FileType.JPEG: {"content_type": "image/jpeg", "handler": image_handler},
+}

resize_image_service/utils/file_handler/handlers/image_handler.py

@@ -0,0 +1,20 @@
+from PIL import Image
+
+import io
+
+
+def image_handler(file_bytes: io.BytesIO) -> io.BytesIO:
+    img = Image.open(file_bytes)
+
+    img.thumbnail((320, 320))
+
+    data = list(img.getdata())
+    image_without_exif = Image.new(img.mode, img.size)
+    image_without_exif.putdata(data)
+
+    new_byte_img = io.BytesIO()
+    img.save(new_byte_img, format="PNG")
+
+    new_byte_img.seek(0)
+
+    return new_byte_img