Merge pull request #45 from 9git9git/BE-기능-수정

✨ 캐릭터 도감의 isCollected를 구분하기 위한 기능추가

Author: GideokKim

app/api/v1/endpoints/character.py

@@ -8,6 +8,7 @@
     delete_character_by_id,
     update_character_by_id,
     get_character_collection,
+    register_user_character,
 )
 from sqlalchemy.ext.asyncio import AsyncSession
 from app.db.session import get_db
@@ -33,7 +34,7 @@ async def post_character(
         )
 
 
-@router.get("/", response_model=ResponseBase[List[CharacterResponse]])
+@router.get("/all", response_model=ResponseBase[List[CharacterResponse]])
 async def get_characters(
     db: AsyncSession = Depends(get_db),
 ) -> ResponseBase[List[CharacterResponse]]:
@@ -98,6 +99,14 @@ async def delete_character(
 
 
 # 서비스 로직
+@router.post("/collect")
+async def collect_character_for_user(
+    user_id: UUID,
+    character_id: UUID,
+    db: AsyncSession = Depends(get_db),
+):
+    await register_user_character(db, user_id, character_id)
+    return {"message": "수집 완료!"}
 
 
 @router.get("/", response_model=ResponseBase[List[CharacterResponse]])

app/services/character.py

@@ -15,6 +15,8 @@
 from sqlalchemy.orm import selectinload
 from app.models.user import UserCharacter
 from app.models.user import Character as CharacterModel
+from uuid import uuid4
+from datetime import datetime
 
 
 async def add_character(
@@ -81,6 +83,19 @@ async def delete_character_by_id(db: AsyncSession, character_id: UUID) -> bool:
 # 서비스 로직
 
 
+async def register_user_character(
+    db: AsyncSession, user_id: UUID, character_id: UUID
+) -> None:
+    user_character = UserCharacter(
+        id=uuid4(),
+        user_id=user_id,
+        character_id=character_id,
+        collected_at=datetime.utcnow(),
+    )
+    db.add(user_character)
+    await db.commit()
+
+
 async def get_character_collection(
     db: AsyncSession, user_id: UUID
 ) -> List[CharacterResponse]: