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},