feat: favorites

backend/src/modules/users/repository.py

@@ -4,7 +4,7 @@
 
 from src.modules.users.schemas import CreateUser
 from src.storages.mongo.users import User
-
+from fastapi import HTTPException
 
 # noinspection PyMethodMayBeStatic
 class UserRepository:
@@ -34,4 +34,8 @@ async def is_banned(self, user_id: str | PydanticObjectId) -> bool:
         return False
 
 
+    async def upsert_favorites(self, user_id: PydanticObjectId, favorite_items: list[PydanticObjectId]) -> list[PydanticObjectId]: 
+        user = await User.find_one(User.id == user_id).update({"$set": {"favorites": list(set(favorite_items))}})
+        return favorite_items
+
 user_repository: UserRepository = UserRepository()

backend/src/modules/users/routes.py

@@ -3,7 +3,8 @@
 from src.api.dependencies import USER_AUTH
 from src.api.exceptions import IncorrectCredentialsException
 from src.modules.users.repository import user_repository
-from src.modules.users.schemas import ViewUser
+from src.modules.users.schemas import ViewUser, UpdateFavoriteReq
+from beanie import PydanticObjectId
 
 router = APIRouter(
     prefix="/users",
@@ -67,3 +68,7 @@ async def logout(request: Request) -> None:
     """
     request.session.clear()
     return None
+
+@router.put("/favorites")
+async def update_favorites(user: USER_AUTH, req: UpdateFavoriteReq) -> list[PydanticObjectId]:
+    return await user_repository.upsert_favorites(user_id=user.user_id, favorite_items=req.favorite_ids)

backend/src/modules/users/schemas.py

@@ -9,6 +9,8 @@ class CreateUser(BaseSchema):
     login: str
     password: str
 
+class UpdateFavoriteReq(BaseSchema):
+    favorite_ids: list[PydanticObjectId]
 
 class ViewUser(BaseSchema):
     id: PydanticObjectId

backend/src/storages/mongo/users.py

@@ -4,6 +4,7 @@
 
 from pymongo import IndexModel
 
+from beanie import PydanticObjectId
 from src.pydantic_base import BaseSchema
 from src.storages.mongo.__base__ import CustomDocument
 
@@ -16,6 +17,7 @@ class UserRole(StrEnum):
 class UserSchema(BaseSchema):
     login: str
     password_hash: str
+    favorites: list[PydanticObjectId]
     role: UserRole = UserRole.DEFAULT