change enum

problem/migrations/0008_alter_problem_unique_together_and_more.py

@@ -0,0 +1,33 @@
+# Generated by Django 6.0.4 on 2026-05-09 08:18
+
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("contest", "0004_alter_acmcontestrank_unique_together_and_more"),
+        ("problem", "0007_problem_problem_visible_idx"),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.AlterUniqueTogether(
+            name="problem",
+            unique_together=set(),
+        ),
+        migrations.AlterField(
+            model_name="problem",
+            name="difficulty",
+            field=models.TextField(choices=[("Low", "Low"), ("Mid", "Mid"), ("High", "High")]),
+        ),
+        migrations.AlterField(
+            model_name="problem",
+            name="rule_type",
+            field=models.TextField(choices=[("ACM", "ACM"), ("OI", "OI")]),
+        ),
+        migrations.AddConstraint(
+            model_name="problem",
+            constraint=models.UniqueConstraint(fields=("_id", "contest"), name="unique_problem_id_contest"),
+        ),
+    ]

problem/models.py

@@ -2,7 +2,7 @@ from django.db import models
 
 from account.models import User
 from contest.models import Contest
-from utils.constants import Choices
+from utils.constants import Difficulty
 from utils.models import RichTextField
 
 
@@ -13,25 +13,19 @@ class ProblemTag(models.Model):
         db_table = "problem_tag"
 
 
-class ProblemRuleType(Choices):
-    ACM = "ACM"
-    OI = "OI"
+class ProblemRuleType(models.TextChoices):
+    ACM = "ACM", "ACM"
+    OI = "OI", "OI"
 
 
-class ProblemDifficulty(object):
-    High = "High"
-    Mid = "Mid"
-    Low = "Low"
-
-
-class ProblemIOMode(Choices):
-    standard = "Standard IO"
-    file = "File IO"
+class ProblemIOMode(models.TextChoices):
+    STANDARD = "Standard IO", "Standard IO"
+    FILE = "File IO", "File IO"
 
 
 def _default_io_mode():
     return {
-        "io_mode": ProblemIOMode.standard,
+        "io_mode": ProblemIOMode.STANDARD,
         "input": "input.txt",
         "output": "output.txt",
     }
@@ -66,9 +60,9 @@ class Problem(models.Model):
     memory_limit = models.IntegerField()
     # io mode
     io_mode = models.JSONField(default=_default_io_mode)
-    rule_type = models.TextField()
+    rule_type = models.TextField(choices=ProblemRuleType.choices)
     visible = models.BooleanField(default=True)
-    difficulty = models.TextField()
+    difficulty = models.TextField(choices=Difficulty.choices)
     tags = models.ManyToManyField(ProblemTag)
     source = models.TextField(null=True)
     prompt = models.TextField(null=True)
@@ -81,7 +75,7 @@ class Problem(models.Model):
     # {JudgeStatus.ACCEPTED: 3, JudgeStatus.WRONG_ANSWER: 11}, the number means count
     statistic_info = models.JSONField(default=dict)
     share_submission = models.BooleanField(default=False)
-    
+
     # 流程图相关字段
     allow_flowchart = models.BooleanField(default=False)  # 是否允许/需要提交流程图
     mermaid_code = models.TextField(null=True, blank=True)  # 流程图答案(Mermaid代码)
@@ -91,7 +85,9 @@ class Problem(models.Model):
 
     class Meta:
         db_table = "problem"
-        unique_together = (("_id", "contest"),)
+        constraints = [
+            models.UniqueConstraint(fields=["_id", "contest"], name="unique_problem_id_contest"),
+        ]
         ordering = ("create_time",)
         indexes = [
             models.Index(fields=["contest", "visible"], name="problem_contest_visible_idx"),

problem/serializers.py

@@ -38,7 +38,7 @@ class CreateProblemCodeTemplateSerializer(serializers.Serializer):
 
 
 class ProblemIOModeSerializer(serializers.Serializer):
-    io_mode = serializers.ChoiceField(choices=ProblemIOMode.choices())
+    io_mode = serializers.ChoiceField(choices=ProblemIOMode.choices)
     input = serializers.CharField()
     output = serializers.CharField()
 
@@ -59,22 +59,16 @@ class CreateOrEditProblemSerializer(serializers.Serializer):
     output_description = serializers.CharField()
     samples = serializers.ListField(child=CreateSampleSerializer(), allow_empty=False)
     test_case_id = serializers.CharField(max_length=32)
-    test_case_score = serializers.ListField(
-        child=CreateTestCaseScoreSerializer(), allow_empty=True
-    )
+    test_case_score = serializers.ListField(child=CreateTestCaseScoreSerializer(), allow_empty=True)
     time_limit = serializers.IntegerField(min_value=1, max_value=1000 * 60)
     memory_limit = serializers.IntegerField(min_value=1, max_value=1024)
     languages = LanguageNameMultiChoiceField()
     template = serializers.DictField(child=serializers.CharField(min_length=1))
-    rule_type = serializers.ChoiceField(
-        choices=[ProblemRuleType.ACM, ProblemRuleType.OI]
-    )
+    rule_type = serializers.ChoiceField(choices=ProblemRuleType.choices)
     io_mode = ProblemIOModeSerializer()
     visible = serializers.BooleanField()
-    difficulty = serializers.ChoiceField(choices=Difficulty.choices())
-    tags = serializers.ListField(
-        child=serializers.CharField(max_length=32), allow_empty=False
-    )
+    difficulty = serializers.ChoiceField(choices=Difficulty.choices)
+    tags = serializers.ListField(child=serializers.CharField(max_length=32), allow_empty=False)
     hint = serializers.CharField(allow_blank=True, allow_null=True)
     source = serializers.CharField(max_length=256, allow_blank=True, allow_null=True)
     prompt = serializers.CharField(allow_blank=True, allow_null=True)
@@ -88,13 +82,9 @@ class CreateOrEditProblemSerializer(serializers.Serializer):
     # 流程图相关字段
     allow_flowchart = serializers.BooleanField(required=False, default=False)
     show_flowchart = serializers.BooleanField(required=False, default=False)
-    mermaid_code = serializers.CharField(
-        allow_blank=True, allow_null=True, required=False
-    )
+    mermaid_code = serializers.CharField(allow_blank=True, allow_null=True, required=False)
 
-    flowchart_hint = serializers.CharField(
-        allow_blank=True, allow_null=True, required=False
-    )
+    flowchart_hint = serializers.CharField(allow_blank=True, allow_null=True, required=False)
 
 
 class CreateProblemSerializer(CreateOrEditProblemSerializer):
@@ -220,6 +210,7 @@ class ProblemSafeSerializer(BaseProblemSerializer):
             return None
         return obj.flowchart_data
 
+
 class ContestProblemMakePublicSerializer(serializers.Serializer):
     id = serializers.IntegerField()
     display_id = serializers.CharField(max_length=32)