feat: 增加查询记录带上是否点赞的功能

service/record_service.py

@@ -1,29 +1,18 @@
-from sqlalchemy import select, desc, func
+from sqlalchemy import select, func
 from sqlalchemy.orm import Session, selectinload
 
-from models.family import Records, Comments
+from models.family import Records, Comments, Likes
 from schemas.record import RecordCreate, RecordUpdate, RecordResponse, RecordPageResponse
 
 
-def get_all_records(db: Session, skip: int = 0, limit: int = 20) -> list[RecordResponse]:
-    result = db.execute(
+def get_all_records_by_page(db: Session, user_id: int, page: int = 1, size: int = 20) -> RecordPageResponse:
+    base_query = (
         select(Records)
         .options(
             selectinload(Records.user),
             selectinload(Records.comments).selectinload(Comments.user),
         )
-        .order_by(desc(Records.create_time))
-        .offset(skip)
-        .limit(limit)
-    )
-    records = result.scalars().all()
-    return [RecordResponse.model_validate(r) for r in records]
-
-
-def get_all_records_by_page(db: Session, page: int = 1, size: int = 20) -> RecordPageResponse:
-    base_query = select(Records).options(
-        selectinload(Records.user),
-        selectinload(Records.comments).selectinload(Comments.user),
+        .order_by(Records.create_time.desc())
     )
 
     # 总数
@@ -32,15 +21,34 @@ def get_all_records_by_page(db: Session, page: int = 1, size: int = 20) -> Recor
     )
 
     # 分页数据
-    records = db.execute(
-        base_query
-        .order_by(Records.create_time.desc())
-        .offset((page - 1) * size)
-        .limit(size)
-    ).scalars().all()
+    records = (
+        db.execute(base_query.offset((page - 1) * size).limit(size))
+        .scalars()
+        .all()
+    )
+
+    # 批量判断当前用户点赞状态
+    liked_record_ids: set[int] = set()
+
+    if user_id and records:
+        liked_record_ids = set(
+            db.scalars(
+                select(Likes.record_id).where(
+                    Likes.record_id.in_([r.id for r in records]),
+                    Likes.user_id == user_id,
+                )
+            ).all()
+        )
+
+    # 组装响应
+    response_records = []
+    for r in records:
+        resp = RecordResponse.model_validate(r, from_attributes=True)
+        resp.liked = r.id in liked_record_ids
+        response_records.append(resp)
 
     return RecordPageResponse(
-        records=[RecordResponse.model_validate(r) for r in records],
+        records=response_records,
         total=total,
         page=page,
         size=size,