From 2b65acf91abc71065b56f5c56bd320031ec01c5e Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Fri, 5 Jun 2026 08:51:57 -0600
Subject: [PATCH] update

---
 flowchart/views/oj.py  |  8 ++++++++
 submission/views/oj.py | 13 ++++++++++---
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/flowchart/views/oj.py b/flowchart/views/oj.py
index 60b0e63..13bca32 100644
--- a/flowchart/views/oj.py
+++ b/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,
diff --git a/submission/views/oj.py b/submission/views/oj.py
index 3b4810e..59dea1b 100644
--- a/submission/views/oj.py
+++ b/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)