fix: 로또 번호 추천 버그 수정 및 지난회차 1등 당첨금 필드 추가 (#15)

Co-authored-by: Gukhee Jo <[email protected]>

Author: 9keyyyy

src/lotto/entities/schemas.py

@@ -69,6 +69,7 @@ class LottoRecommendationContent(BaseModel):
     )
     strong_element: FiveElements = Field(description="강한 기운", examples=["화(火)"])
     weak_element: FiveElements = Field(description="상충되는 기운", examples=["목(木)"])
+    last_prize_amount: int = Field(description="지난 회차 1등 당첨금")
 
 
 class LottoRecommendation(CommonBase):

src/lotto/repository.py

@@ -108,3 +108,13 @@ async def get_excluded_numbers(
         )
         result = await self.session.execute(query)
         return [row[0] for row in result.fetchall()]
+
+    async def get_latest_prize_amount(self) -> Optional[int]:
+        """가장 최신 회차의 1등 당첨금을 조회합니다."""
+        query = (
+            select(LottoDraws.first_prize_amount)
+            .order_by(desc(LottoDraws.round))
+            .limit(1)
+        )
+        result = await self.session.execute(query)
+        return result.scalar_one_or_none()

src/lotto/service.py

@@ -70,8 +70,11 @@ async def create_lotto_recommendation(self, user_id: str) -> LottoRecommendation
         # 3. 통계 데이터 조회
         frequent_nums = await self.lotto_repository.get_frequent_numbers(limit=10)
         infrequent_nums = await self.lotto_repository.get_excluded_numbers(limit=2)
+        
+        # 4. 최신 회차 당첨금 조회
+        latest_prize_amount = await self.lotto_repository.get_latest_prize_amount()
 
-        # 4. HCX API 호출하여 로또 추천 생성
+        # 5. HCX API 호출하여 로또 추천 생성
         hcx_client = HCXClient()
 
         # 사용자 사주 정보 사용
@@ -81,16 +84,8 @@ async def create_lotto_recommendation(self, user_id: str) -> LottoRecommendation
             "month_pillar": four_pillar.get("month_pillar"),
             "day_pillar": four_pillar.get("day_pillar"),
             "time_pillar": four_pillar.get("time_pillar"),
-            "strong_element": (
-                four_pillar.get("strong_elements")[0]
-                if four_pillar.get("strong_elements")
-                else None
-            ),
-            "weak_element": (
-                four_pillar.get("weak_elements")[0]
-                if four_pillar.get("weak_elements")
-                else None
-            ),
+            "strong_element": four_pillar.get("strong_element"),
+            "weak_element": four_pillar.get("weak_element"),
         }
 
         # HCXUtils를 사용하여 프롬프트 가져오기
@@ -125,6 +120,7 @@ async def create_lotto_recommendation(self, user_id: str) -> LottoRecommendation
                 infrequent_nums=infrequent_nums,
                 strong_element=lotto_prompt_data["strong_element"],
                 weak_element=lotto_prompt_data["weak_element"],
+                last_prize_amount=latest_prize_amount or 0,
             )
 
         except Exception as e:
@@ -133,7 +129,7 @@ async def create_lotto_recommendation(self, user_id: str) -> LottoRecommendation
                 status_code=400, detail=f"로또 추천 생성 실패: {str(e)}"
             )
 
-        # 5. 데이터베이스에 저장
+        # 6. 데이터베이스에 저장
         recommendation = await self.lotto_repository.create_lotto_recommendation(
             user_id=user_id, round=latest_round + 1, content=content.model_dump()
         )

src/users/router.py

@@ -38,7 +38,7 @@ async def create_user(user_create: UserCreate, user_service: UserService = Depen
     - name: 사용자 이름
     - birth_date: 사용자 생년월일시 (YYYY-MM-DD HH:MM:SS 형식)
       - 생시를 모르는 경우 입력하지 않음 (YYYY-MM-DD 형식)
-      - 선택한 시간 범위에서 가장 빠른 시간으로 입력 (예, 23:00:00 ~ 00:59:59의 경우 23:00:00로 입
+      - 선택한 시간 범위에서 가장 빠른 시간으로 입력 (ex. 23:00:00 ~ 00:59:59의 경우 23:00:00로 입력)
     - gender: 사용자 성별 (M/F)
     """
     return await user_service.create_user(user_create)
@@ -66,15 +66,15 @@ async def get_user_four_pillar(user_id: str, user_service: UserService = Depends
     """
     return await user_service.get_user_four_pillar(user_id)
 
+
 @user_router.put("/{user_id}", response_model=UserDetail)
 async def update_user(
-    user_id: str,
-    user_update: UserUpdate,
-    user_service: UserService = Depends()
+    user_id: str, user_update: UserUpdate, user_service: UserService = Depends()
 ):
     """사용자 정보를 수정합니다."""
     return await user_service.update_user(user_id, user_update)
 
+
 @user_router.post("/{user_id}/lotto-recommendation", response_model=LottoRecommendation)
 async def create_lotto_recommendation(
     user_id: str,
@@ -90,6 +90,7 @@ async def create_lotto_recommendation(
             status_code=500, detail=f"로또 추천 생성 중 오류가 발생했습니다: {str(e)}"
         )
 
+
 @user_router.get(
     "/{user_id}/lotto-recommendation", response_model=Optional[LottoRecommendation]
 )
@@ -99,20 +100,24 @@ async def get_lotto_recommendation(
 ):
     """
     사용자의 최신 로또 추천을 조회합니다.
-    
+
     content:
     - reason: 사주 기반 로또 추천 이유
     - num1, num2: 기운과 잘 맞는 숫자 (1-45)
     - num3, num4: 재물운 좋을 때 나오는 숫자 (1-45)
     - num5, num6: 최근 자주 나온 번호 (1-45)
     - cold_nums: 기운과 상충하는 숫자 (1-3개)
     - infrequent_nums: 등장 빈도가 낮은 숫자 (1-3개)
+    - last_prize_amount: 지난 회차 1등 당첨금 (1인당 수령액)
     - strong_element: 강한 기운
     - weak_element: 상충되는 기운
     """
     return await lotto_service.get_lotto_recommendation(user_id=user_id)
 
-@user_router.get("/{user_id}/daily-fortunes", response_model=List[UserDailyFortuneSummary])
+
+@user_router.get(
+    "/{user_id}/daily-fortunes", response_model=List[UserDailyFortuneSummary]
+)
 async def get_user_daily_fortunes(
     user_id: str,
     fortune_date: date = Query(default=date.today()),