30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
from account.decorators import super_admin_required
|
|
from comment.serializers import CommentListSerializer
|
|
from problem.models import Problem
|
|
from utils.api import APIView
|
|
from comment.models import Comment
|
|
|
|
|
|
class CommentAPI(APIView):
|
|
@super_admin_required
|
|
def get(self, request):
|
|
comments = Comment.objects.select_related("problem").exclude(content="")
|
|
problem_id = request.GET.get("problem")
|
|
if problem_id:
|
|
try:
|
|
# 这里如果题目不可见，也需要显示该题目的评论
|
|
problem = Problem.objects.get(_id=problem_id, contest_id__isnull=True)
|
|
except Problem.DoesNotExist:
|
|
return self.error("Problem doesn't exist")
|
|
comments = comments.filter(problem=problem)
|
|
return self.success(
|
|
self.paginate_data(request, comments, CommentListSerializer)
|
|
)
|
|
|
|
@super_admin_required
|
|
def delete(self, request):
|
|
id = request.GET.get("id")
|
|
if id:
|
|
Comment.objects.filter(id=id).delete()
|
|
return self.success()
|