From c4ddfa6841f97ed6e7063555955ad13eba1c0f29 Mon Sep 17 00:00:00 2001
From: yuetsh <517252939@qq.com>
Date: Sun, 12 Apr 2026 06:00:39 -0600
Subject: [PATCH] add flowchart

---
 problem/serializers.py | 1 +
 problem/views/oj.py    | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/problem/serializers.py b/problem/serializers.py
index 342ef67..3a70c00 100644
--- a/problem/serializers.py
+++ b/problem/serializers.py
@@ -185,6 +185,7 @@ class ProblemListSerializer(BaseProblemSerializer):
             "tags",
             "contest",
             "allow_flowchart",
+            "show_flowchart",
         ]
 
 
diff --git a/problem/views/oj.py b/problem/views/oj.py
index 14c9133..2da9452 100644
--- a/problem/views/oj.py
+++ b/problem/views/oj.py
@@ -7,7 +7,7 @@ from submission.models import Submission, JudgeStatus
 from utils.api import APIView
 from account.decorators import check_contest_permission
 from utils.constants import CacheKey
-from ..models import ProblemTag, Problem, ProblemRuleType
+from ..models import ProblemTag, Problem
 from ..serializers import (
     ProblemSerializer,
     TagSerializer,
@@ -124,8 +124,10 @@ class ProblemAPI(APIView):
 
         # 排序
         sort = request.GET.get("sort")
-        if sort:
+        if sort and sort != "flowchart":
             problems = problems.order_by(sort)
+        if sort and sort == "flowchart":
+            problems = problems.order_by("-allow_flowchart", "-show_flowchart", "-create_time")
 
         # 根据profile 为做过的题目添加标记
         data = self.paginate_data(request, problems, ProblemListSerializer)