testing file read vs pil read and convert

Author: Eldies

app/storage_manager.py

@@ -28,6 +28,12 @@ def get_image(self, uuid: list[str]) -> Image.Image:
             raise StorageManagerException("Not Found")
         return Image.open(os.path.join(self.path_for_uuid(uuid), "file"))
 
+    def read_file(self, uuid: list[str]) -> bytes:
+        if not self.uuid_exists(uuid):
+            raise StorageManagerException("Not Found")
+        with open(os.path.join(self.path_for_uuid(uuid), "file"), "rb") as f:
+            return f.read()
+
 
 def get_storage_manager() -> StorageManager:
     return StorageManager()

app/views.py

@@ -62,11 +62,23 @@ async def post_image(
 @router_api.get("/image/{client_id}/{uuid}")
 async def get_image(client_id: str, uuid: str) -> Response:
     try:
+        logger.debug("start file")
+        get_storage_manager().read_file([client_id, uuid])
+        logger.debug("finished file")
+        logger.debug("start image")
         image = get_storage_manager().get_image([client_id, uuid])
+        logger.debug("image 0")
+        form = image.format
+        assert form is not None
+        logger.debug("image 1")
         img_stream = io.BytesIO()
-        assert image.format is not None
-        image.save(img_stream, image.format, quality="keep")
-        return Response(content=img_stream.getvalue(), media_type=Image.MIME[image.format])
+        logger.debug("image 2")
+        image.save(img_stream, form, quality="keep")
+        logger.debug("image 3")
+        image.tobytes()
+        image_bytes = img_stream.getvalue()
+        logger.debug("finished image")
+        return Response(content=image_bytes, media_type=Image.MIME[form])
     except StorageManagerException as e:
         raise HTTPException(
             status_code=status.HTTP_404_NOT_FOUND,