add image deletion api (#12)

* add image deletion api

* psycho

---------

Co-authored-by: Josh Smith <[email protected]>

app/adapters/s3.py

@@ -13,13 +13,13 @@
 async def upload(
     body: bytes,
     file_name: str,
-    folder: str,
+    directory: str,
     content_type: str | None = None,
     acl: str | None = None,
 ) -> None:
     params: dict[str, Any] = {
         "Bucket": settings.AWS_S3_BUCKET_NAME,
-        "Key": f"{folder}/{file_name}",
+        "Key": f"{directory}/{file_name}",
         "Body": body,
     }
     if content_type is not None:
@@ -60,3 +60,16 @@ async def download(file_name: str, directory: str) -> DownloadResponse | None:
         "body": body,
         "content_type": content_type,
     }
+
+
+async def delete(file_name: str, directory: str) -> None:
+    try:
+        await app.clients.s3_client.delete_object(
+            Bucket=settings.AWS_S3_BUCKET_NAME,
+            Key=f"{directory}/{file_name}",
+        )
+    except app.clients.s3_client.exceptions.NoSuchKey:
+        return None
+    except (botocore.exceptions.BotoCoreError, botocore.exceptions.ClientError) as exc:
+        logging.error("Failed to download file from S3", exc_info=exc)
+        return None

app/api/v1/avatars.py

@@ -84,3 +84,21 @@ async def get_avatar(file_path: str):
         content=download_response["body"],
         media_type=download_response["content_type"],
     )
+
+
+@router.delete("/api/v1/users/{user_id}/avatar")
+async def delete_avatar(
+    user_id: int,
+    authorization: AdminAuthorization = Depends(authorize_admin),
+):
+    data = await app.usecases.images.delete_image(
+        image_type=ImageType.USER_AVATAR,
+        no_ext_file_name=f"{user_id}",
+        authorization=authorization,
+    )
+    if isinstance(data, Error):
+        return Response(
+            status_code=_get_status_code_for_error(data.code),
+            content=data.user_feedback,
+        )
+    return Response(status_code=201)

app/usecases/images.py

@@ -143,7 +143,7 @@ async def upload_image(
     await s3.upload(
         body=image_content,
         file_name=f"{no_ext_file_name}.{image_format.lower()}",
-        folder=image_type.get_s3_folder(),
+        directory=image_type.get_s3_folder(),
         content_type=image_mime_type,
     )
     logging.info(
@@ -157,3 +157,24 @@ async def upload_image(
     )
 
     return None
+
+
+async def delete_image(
+    image_type: ImageType,
+    no_ext_file_name: str,
+    authorization: AbstractAuthorization,
+) -> None | Error:
+    for ext in ["png", "jpg", "jpeg", "gif", "webp"]:
+        await s3.delete(
+            file_name=f"{no_ext_file_name}.{ext}",
+            directory=image_type.get_s3_folder(),
+        )
+
+    logging.info(
+        "Deleted image",
+        extra={
+            "file_name": no_ext_file_name,
+            "image_type": image_type,
+            "authorization": authorization.format_for_logs(),
+        },
+    )