feature: user avatar upload route added & file upload task added

Author: DataRohit

docker-compose.yml

@@ -94,6 +94,7 @@ services:
     container_name: minio
     ports:
       - "9090:9090"
+      - "9000:9000"
     volumes:
       - minio_data:/data
     env_file:

pkg/storage.py

@@ -0,0 +1,14 @@
+from minio import Minio
+
+from .config import Config
+
+minio_client = Minio(
+    endpoint=Config.MINIO_STORAGE_ENDPOINT,
+    access_key=Config.MINIO_ACCESS_KEY,
+    secret_key=Config.MINIO_SECRET_KEY,
+    secure=False,
+)
+
+if not minio_client.bucket_exists(Config.MINIO_STORAGE_BUCKET):
+    minio_client.make_bucket(Config.MINIO_STORAGE_BUCKET)
+    print(f"Bucket {Config.MINIO_STORAGE_BUCKET} created")

requirements.txt

@@ -3,12 +3,15 @@ alembic==1.13.3
 amqp==5.2.0
 annotated-types==0.7.0
 anyio==4.6.0
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
 asgiref==3.8.1
 asyncpg==0.29.0
 bcrypt==4.2.0
 billiard==4.2.1
 celery==5.4.0
 certifi==2024.8.30
+cffi==1.17.1
 click==8.1.7
 click-didyoumean==0.3.1
 click-plugins==1.1.1
@@ -32,9 +35,12 @@ Mako==1.3.5
 markdown-it-py==3.0.0
 MarkupSafe==2.1.5
 mdurl==0.1.2
+minio==7.2.9
 prometheus_client==0.21.0
 prompt_toolkit==3.0.48
 psycopg2-binary==2.9.9
+pycparser==2.22
+pycryptodome==3.21.0
 pydantic==2.9.2
 pydantic-settings==2.5.2
 pydantic_core==2.23.4
@@ -57,6 +63,7 @@ tornado==6.4.1
 typer==0.12.5
 typing_extensions==4.12.2
 tzdata==2024.2
+urllib3==2.2.3
 uvicorn==0.31.0
 uvloop==0.20.0
 vine==5.1.0

src/profile/routes.py

@@ -1,23 +1,26 @@
 import json
+import os
 
-from fastapi import APIRouter, Depends, status
+from fastapi import APIRouter, Depends, Request, status
 from fastapi.responses import JSONResponse
 from sqlmodel.ext.asyncio.session import AsyncSession
 
+from pkg.config import Config
 from pkg.db import get_session
 from pkg.utils import get_current_user_uid
 from src.auth.service import UserService
 
 from .schemas import UserProfileResponseSchema, UserProfileUpdateSchema
 from .service import UserProfileService
+from .tasks import upload_user_avatar_image_task
 
 profile_router = APIRouter()
 
 user_service = UserService()
 user_profile_service = UserProfileService()
 
 
-@profile_router.put("/", status_code=status.HTTP_200_OK)
+@profile_router.patch("/update-profile", status_code=status.HTTP_200_OK)
 async def update_user_profile(
     profile_data: UserProfileUpdateSchema,
     session: AsyncSession = Depends(get_session),
@@ -39,3 +42,48 @@ async def update_user_profile(
             ).model_dump(),
         },
     )
+
+
+@profile_router.patch("/update-avatar", status_code=status.HTTP_200_OK)
+async def update_user_avatar(
+    request: Request,
+    session: AsyncSession = Depends(get_session),
+    user_uid: str = Depends(get_current_user_uid),
+):
+    user_profile = await user_profile_service.get_user_profile_by_user_uid(
+        user_uid, session
+    )
+
+    form = await request.form()
+    avatar_image = form.get("avatar_image")
+
+    if not avatar_image:
+        return JSONResponse(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            content={"message": "Avatar image is required"},
+        )
+
+    avatar_image_content = await avatar_image.read()
+    avatar_image_file_extension = os.path.splitext(avatar_image.filename)[1]
+    avatar_image_file_name = f"avatars/{user_uid}{avatar_image_file_extension}"
+
+    upload_user_avatar_image_task.delay(
+        avatar_image_content,
+        avatar_image_file_name,
+        avatar_image.content_type,
+    )
+
+    file_url = f"http://{Config.MINIO_STORAGE_ENDPOINT}/{Config.MINIO_STORAGE_BUCKET}/{avatar_image_file_name}"
+    await user_profile_service.update_user_profile_avatar(
+        user_profile, file_url, session
+    )
+
+    return JSONResponse(
+        status_code=status.HTTP_200_OK,
+        content={
+            "message": "User avatar updated successfully",
+            "user_profile": UserProfileResponseSchema(
+                **json.loads(user_profile.model_dump_json())
+            ).model_dump(),
+        },
+    )

src/profile/service.py

@@ -24,3 +24,11 @@ async def get_user_profile_by_user_uid(
         )
         user_profile = result.scalars().first()
         return user_profile
+
+    async def update_user_profile_avatar(
+        self, user_profile: UserProfile, user_avatar_url: str, session: AsyncSession
+    ):
+        user_profile.avatar = user_avatar_url
+        await session.commit()
+        await session.refresh(user_profile)
+        return user_profile

src/profile/tasks.py

@@ -0,0 +1,20 @@
+from io import BytesIO
+
+from pkg.celery_app import celery_app
+from pkg.config import Config
+from pkg.storage import minio_client
+
+
+@celery_app.task
+def upload_user_avatar_image_task(
+    image_content: bytes, filename: str, content_type: str
+):
+    file_data = BytesIO(image_content)
+
+    minio_client.put_object(
+        bucket_name=Config.MINIO_STORAGE_BUCKET,
+        object_name=filename,
+        data=file_data,
+        length=len(image_content),
+        content_type=content_type,
+    )