xuyue/OnlineJudge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

API 调整

Browse Source
This commit is contained in:
yuetsh
2024-07-01 01:23:32 +08:00
committed by yuetsh
parent 5bcccf12e4
commit 145797076d
2 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comment/urls/oj.py
View File

@@ -1,8 +1,9 @@
from django.urls import re_path as url from django.urls import re_path as url
from ..views.oj import CommentAPI from ..views.oj import CommentAPI, CommentStatisticsAPI
urlpatterns = [ urlpatterns = [
url(r"^comment/?$", CommentAPI.as_view(), name="comment_api"), url(r"^comment/?$", CommentAPI.as_view(), name="comment_api"),
url(r"^comment/statistics?$", CommentStatisticsAPI.as_view(), name="comment_statistics_api"),
] ]

21
comment/views/oj.py
View File

@@ -48,18 +48,23 @@ class CommentAPI(APIView):
) )
return self.success() return self.success()
@login_required
def get(self, request): def get(self, request):
problem_id = request.GET.get("problem_id") problem_id = request.GET.get("problem_id")
my_comment = request.GET.get("my_comment") comment = (
Comment.objects.select_related("problem")
if my_comment: .filter(user=request.user, problem_id=problem_id, visible=True)
comment = ( .first()
Comment.objects.select_related("problem") )
.filter(user=request.user, problem_id=problem_id, visible=True) if comment:
.first()
)
return self.success(CommentSerializer(comment).data) return self.success(CommentSerializer(comment).data)
else:
return self.success()
class CommentStatisticsAPI(APIView):
def get(self, request):
problem_id = request.GET.get("problem_id")
comments = Comment.objects.select_related("problem").filter( comments = Comment.objects.select_related("problem").filter(
problem_id=problem_id, visible=True problem_id=problem_id, visible=True
) )