From 5ab70499d5f6ef0f1e8531fc5a4b0cfa77dd893a Mon Sep 17 00:00:00 2001 From: Vitor Hideyoshi Date: Thu, 24 Aug 2023 01:35:05 -0300 Subject: [PATCH] Fixes File Not Found Empty - Bucket Case --- storage_service/service/amazon_s3_service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/storage_service/service/amazon_s3_service.py b/storage_service/service/amazon_s3_service.py index 9a4f8e3..c0d2e75 100644 --- a/storage_service/service/amazon_s3_service.py +++ b/storage_service/service/amazon_s3_service.py @@ -29,7 +29,7 @@ class AmazonS3Service(StorageService): ) def get_temp_upload_link( - self, file_name, file_type: FileType + self, file_name, file_type: FileType ) -> dict[str, str | Any]: return { "presigned_url": self._get_presigned_write_url(file_name, file_type), @@ -62,7 +62,7 @@ class AmazonS3Service(StorageService): def _get_presigned_read_url(self, file_name) -> str | None: result = self.s3.list_objects(Bucket=self.bucket_name, Prefix=file_name) - if file_name in map(lambda x: x["Key"], result["Contents"]): + if "Contents" in result and file_name in map(lambda x: x["Key"], result["Contents"]): return self.s3.generate_presigned_url( "get_object", Params={"Bucket": self.bucket_name, "Key": file_name},