[WIP] UserChallengeStatus 오류 수정

getcloser/backend/app/models/challenges.py

@@ -6,8 +6,9 @@
 class UserChallengeStatus(Base):
     __tablename__ = "user_challenge_status"
 
-    user_id = Column(Integer, primary_key=True, index=True)
-    challenge_id = Column(Integer, ForeignKey("challenge_questions.id"), index=True)
+    id = Column(Integer, primary_key=True, index=True)
+    user_id = Column(Integer, index=True)
+    challenge_id = Column(Integer, ForeignKey("challenge_questions.id"), nullable=True)
 
     is_correct = Column(Boolean, default=False)
     submitted_at = Column(DateTime(timezone=True))

getcloser/backend/app/services/challenge_service.py

@@ -36,16 +36,34 @@ def assign_challenges_logic(my_id: str, members: list, db: Session) -> list:
     available_ids = members.copy()
     random.shuffle(available_ids)
 
+    members = list(set(members))
     for user_id in members:
         possible_ids = [uid for uid in available_ids if uid != user_id]
         if not possible_ids:
             raise ValueError(f"{user_id}에게 할당할 문제 부족")
 
         assigned_user_id = random.choice(possible_ids)
         assigned_question = random.choice([q for q in team_questions if q.user_id == assigned_user_id])
-
+        
         available_ids.remove(assigned_user_id)
 
+        # ✅ UserChallengeStatus 업데이트
+        user_status = (
+            db.query(UserChallengeStatus)
+            .filter(UserChallengeStatus.user_id == user_id)
+            .first()
+        )
+
+        if not user_status:
+            raise HTTPException(status_code=404, detail="UserChallengeStatus 없음")
+
+        user_status.challenge_id = assigned_question.id
+        user_status.submitted_at = None
+        user_status.is_correct = False
+        user_status.is_redeemed = False
+
+        db.add(user_status)
+
         assigned_list.append(AssignedChallenge(
             user_id=user_id,
             assigned_challenge_id=assigned_question.id,
@@ -54,6 +72,8 @@ def assign_challenges_logic(my_id: str, members: list, db: Session) -> list:
             answer=assigned_question.answer
         ))
 
+        db.commit()
+
     return assigned_list[0]