update

flowchart/views/oj.py

@@ -1,3 +1,5 @@
+from django.utils import timezone
+
 from account.decorators import login_required
 from flowchart.models import FlowchartSubmission, FlowchartSubmissionStatus
 from flowchart.serializers import (
@@ -86,6 +88,12 @@ class FlowchartSubmissionListAPI(AsyncAPIView):
         elif request.user.is_regular_user():
             queryset = queryset.filter(user=request.user)
 
+        if request.GET.get("today") == "1":
+            now = timezone.now()
+            queryset = queryset.filter(
+                create_time__gte=now.replace(hour=0, minute=0, second=0, microsecond=0)
+            )
+
         data = await self.async_paginate_data(request, queryset)
         data["results"] = await self.async_serialize_data(
             FlowchartSubmissionListSerializer,

submission/views/oj.py

@@ -5,6 +5,7 @@ from django.utils import timezone
 
 from account.decorators import check_contest_permission, login_required
 from contest.models import ContestStatus
+from flowchart.models import FlowchartSubmission
 from judge.tasks import judge_task
 from options.options import SysOptions
 
@@ -266,7 +267,13 @@ class SubmissionExistsAPI(AsyncAPIView):
 class SubmissionsTodayCount(AsyncAPIView):
     async def get(self, request):
         now = timezone.now()
-        count = await Submission.objects.filter(
-            create_time__gte=now.replace(hour=0, minute=0, second=0, microsecond=0)
-        ).acount()
+        start = now.replace(hour=0, minute=0, second=0, microsecond=0)
+        if request.GET.get("language") == "Flowchart":
+            count = await FlowchartSubmission.objects.filter(
+                create_time__gte=start
+            ).acount()
+        else:
+            count = await Submission.objects.filter(
+                contest_id__isnull=True, create_time__gte=start
+            ).acount()
         return self.success(count)