[FIX] 토론 좋아요 수 반영

Author: SeoYeomm

crud/discussion.py

@@ -1,5 +1,5 @@
 from sqlalchemy.orm import Session
-from models.discussion import Discussion
+from models.discussion import Discussion, DiscussionLike
 from schemas.discussion import DiscussionCreate, DiscussionLikeResponse
 from typing import List
 
@@ -19,20 +19,32 @@ def get_discussions_by_question(db: Session, question_id: int):
 
 # 토론 좋아요 추가
 def add_like_to_discussion(db: Session, discussion_id: int, user_id: int):
-    discussion = discussion = db.query(Discussion).filter(Discussion.discussion_id == discussion_id).first()
-    # 좋아요 토글: 좋아요가 1 이상이면 취소, 0이면 추가
-    if discussion.like > 0:
-        discussion.like -= 1  # 좋아요 취소
+    discussion = db.query(Discussion).filter(Discussion.discussion_id == discussion_id).first()
+    
+    # 사용자가 이미 좋아요를 눌렀는지 확인
+    existing_like = db.query(DiscussionLike).filter(
+        DiscussionLike.discussion_id == discussion_id,
+        DiscussionLike.user_id == user_id
+    ).first()
+
+    if existing_like:
+        # 이미 눌렀다면 삭제 (좋아요 취소)
+        db.delete(existing_like)
+        discussion.like -= 1
         liked = False
     else:
-        discussion.like += 1  # 좋아요 추가
+        # 좋아요 추가
+        new_like = DiscussionLike(discussion_id=discussion_id, user_id=user_id)
+        db.add(new_like)
+        discussion.like += 1
         liked = True
 
     db.commit()
     db.refresh(discussion)
 
-    return {"discussion_id": discussion_id, "like": discussion.like, "liked": liked}    
+    return {"discussion_id": discussion_id, "like": discussion.like, "liked": liked}
 
 # 좋아요 누른 토론 조회
 def get_liked_discussions_by_user(db: Session, user_id: int) -> List[Discussion]:
-    return db.query(Discussion).filter(Discussion.user_id == user_id, Discussion.like > 0).all()
+    return db.query(Discussion).join(DiscussionLike, Discussion.discussion_id == DiscussionLike.discussion_id)\
+        .filter(DiscussionLike.user_id == user_id).all()

models/discussion.py

@@ -13,3 +13,10 @@ class Discussion(Base):
     comment_exist = Column(Boolean, default=False, nullable=False)
     created_date = Column(DateTime, server_default=func.now(), nullable=False)
     updated_date = Column(DateTime, onupdate=func.now(), nullable=True)
+
+class DiscussionLike(Base):
+    __tablename__ = "discussion_likes"
+
+    discussion_likes_id = Column(BigInteger, primary_key=True, autoincrement=True)
+    discussion_id = Column(BigInteger, ForeignKey("discussions.discussion_id"), nullable=False)
+    user_id = Column(BigInteger, ForeignKey("users.user_id"), nullable=False)