diff --git a/submission/urls/oj.py b/submission/urls/oj.py index cca081c..cafc4d0 100644 --- a/submission/urls/oj.py +++ b/submission/urls/oj.py @@ -5,11 +5,13 @@ from ..views.oj import ( SubmissionListAPI, ContestSubmissionListAPI, SubmissionExistsAPI, + SubmissionsTodayCount, ) urlpatterns = [ path("submission", SubmissionAPI.as_view()), path("submissions", SubmissionListAPI.as_view()), + path("submissions/today_count", SubmissionsTodayCount.as_view()), path("submission_exists", SubmissionExistsAPI.as_view()), path("contest_submissions", ContestSubmissionListAPI.as_view()), ] diff --git a/submission/views/oj.py b/submission/views/oj.py index c914b26..325b309 100644 --- a/submission/views/oj.py +++ b/submission/views/oj.py @@ -190,13 +190,6 @@ class SubmissionListAPI(APIView): data["results"] = SubmissionListSerializer( data["results"], many=True, user=request.user ).data - - today = datetime.today() - today_submissions_count = Submission.objects.filter( - create_time__gte=datetime(today.year, today.month, today.day, 0, 0) - ).count() - data["today_count"] = today_submissions_count - return self.success(data) @@ -258,3 +251,12 @@ class SubmissionExistsAPI(APIView): problem_id=request.GET["problem_id"], user_id=request.user.id ).exists() ) + + +class SubmissionsTodayCount(APIView): + def get(self, request): + today = datetime.today() + count = Submission.objects.filter( + create_time__gte=datetime(today.year, today.month, today.day, 0, 0) + ).count() + return self.success(count)