92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
"""
|
|
评论 CRUD 操作
|
|
"""
|
|
from typing import Optional, List, Tuple
|
|
from app.models.comment import Comment
|
|
from app.models.post import Post
|
|
|
|
|
|
class CommentCRUD:
|
|
"""评论 CRUD 操作类"""
|
|
|
|
@staticmethod
|
|
async def get_by_id(comment_id: str) -> Optional[Comment]:
|
|
"""根据 ID 获取评论"""
|
|
return await Comment.filter(id=comment_id).first()
|
|
|
|
@staticmethod
|
|
async def get_by_post(
|
|
post_id: str,
|
|
page: int = 1,
|
|
page_size: int = 20,
|
|
approved_only: bool = True
|
|
) -> Tuple[List[Comment], int]:
|
|
"""获取文章的所有评论（树形结构）"""
|
|
query = Comment.filter(post_id=post_id)
|
|
|
|
if approved_only:
|
|
query = query.filter(is_approved=True)
|
|
|
|
total = await query.count()
|
|
comments = await query \
|
|
.prefetch_related("author", "replies__author") \
|
|
.filter(parent_id=None) \
|
|
.offset((page - 1) * page_size) \
|
|
.limit(page_size) \
|
|
.order_by("created_at")
|
|
|
|
return comments, total
|
|
|
|
@staticmethod
|
|
async def create(
|
|
content: str,
|
|
author_id: str,
|
|
post_id: str,
|
|
parent_id: Optional[str] = None,
|
|
is_approved: bool = True
|
|
) -> Comment:
|
|
"""创建评论"""
|
|
comment = await Comment.create(
|
|
content=content,
|
|
author_id=author_id,
|
|
post_id=post_id,
|
|
parent_id=parent_id,
|
|
is_approved=is_approved
|
|
)
|
|
|
|
# 更新文章的评论数
|
|
await Post.filter(id=post_id).update(comment_count=Post.comment_count + 1)
|
|
|
|
return comment
|
|
|
|
@staticmethod
|
|
async def update(comment_id: str, **kwargs) -> Optional[Comment]:
|
|
"""更新评论"""
|
|
comment = await Comment.filter(id=comment_id).first()
|
|
if comment:
|
|
for key, value in kwargs.items():
|
|
if value is not None and hasattr(comment, key):
|
|
setattr(comment, key, value)
|
|
await comment.save()
|
|
return comment
|
|
|
|
@staticmethod
|
|
async def delete(comment_id: str) -> bool:
|
|
"""删除评论"""
|
|
comment = await Comment.filter(id=comment_id).first()
|
|
if comment:
|
|
post_id = comment.post_id
|
|
await comment.delete()
|
|
# 更新文章的评论数
|
|
await Post.filter(id=post_id).update(comment_count=Post.comment_count - 1)
|
|
return True
|
|
return False
|
|
|
|
@staticmethod
|
|
async def approve(comment_id: str) -> Optional[Comment]:
|
|
"""审核通过评论"""
|
|
return await CommentCRUD.update(comment_id, is_approved=True)
|
|
|
|
|
|
comment_crud = CommentCRUD()
|