change enum

problemset/models.py

@@ -6,15 +6,31 @@ from problem.models import Problem
 from utils.models import JSONField, RichTextField
 
 
+class ProblemSetStatus(models.TextChoices):
+    DRAFT = "draft", "Draft"
+    ACTIVE = "active", "Active"
+    ARCHIVED = "archived", "Archived"
+
+
+class ProblemSetDifficulty(models.TextChoices):
+    EASY = "Easy", "Easy"
+    MEDIUM = "Medium", "Medium"
+    HARD = "Hard", "Hard"
+
+
+class BadgeConditionType(models.TextChoices):
+    ALL_PROBLEMS = "all_problems", "All Problems"
+    PROBLEM_COUNT = "problem_count", "Problem Count"
+    SCORE = "score", "Score"
+
+
 class ProblemSet(models.Model):
     """题单模型"""
 
     title = models.TextField(verbose_name="题单标题")
     description = RichTextField(verbose_name="题单描述")
     # 创建者
-    created_by = models.ForeignKey(
-        User, on_delete=models.CASCADE, verbose_name="创建者"
-    )
+    created_by = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="创建者")
     # 创建时间
     create_time = models.DateTimeField(auto_now_add=True, verbose_name="创建时间")
     # 更新时间
@@ -22,11 +38,13 @@ class ProblemSet(models.Model):
     # 是否可见
     visible = models.BooleanField(default=True, verbose_name="是否可见")
     # 题单难度等级
-    difficulty = models.TextField(default="Easy", verbose_name="难度等级")
+    difficulty = models.TextField(
+        default=ProblemSetDifficulty.EASY,
+        choices=ProblemSetDifficulty.choices,
+        verbose_name="难度等级",
+    )
     # 题单状态
-    status = models.TextField(
-        default="draft", verbose_name="状态"
-    )  # active, archived, draft
+    status = models.TextField(default=ProblemSetStatus.DRAFT, choices=ProblemSetStatus.choices, verbose_name="状态")
     # 截止时间（到期后自动解除防作弊隐藏）
     end_time = models.DateTimeField(null=True, blank=True, verbose_name="截止时间")
 
@@ -43,9 +61,7 @@ class ProblemSet(models.Model):
 class ProblemSetProblem(models.Model):
     """题单题目关联模型"""
 
-    problemset = models.ForeignKey(
-        ProblemSet, on_delete=models.CASCADE, verbose_name="题单"
-    )
+    problemset = models.ForeignKey(ProblemSet, on_delete=models.CASCADE, verbose_name="题单")
     problem = models.ForeignKey(Problem, on_delete=models.CASCADE, verbose_name="题目")
     # 在题单中的顺序
     order = models.IntegerField(default=0, verbose_name="顺序")
@@ -58,7 +74,9 @@ class ProblemSetProblem(models.Model):
 
     class Meta:
         db_table = "problemset_problem"
-        unique_together = (("problemset", "problem"),)
+        constraints = [
+            models.UniqueConstraint(fields=["problemset", "problem"], name="unique_problemset_problem"),
+        ]
         ordering = ("order",)
         verbose_name = "题单题目"
         verbose_name_plural = "题单题目"
@@ -70,17 +88,13 @@ class ProblemSetProblem(models.Model):
 class ProblemSetBadge(models.Model):
     """题单奖章模型"""
 
-    problemset = models.ForeignKey(
-        ProblemSet, on_delete=models.CASCADE, verbose_name="题单"
-    )
+    problemset = models.ForeignKey(ProblemSet, on_delete=models.CASCADE, verbose_name="题单")
     name = models.TextField(verbose_name="奖章名称")
     description = models.TextField(verbose_name="奖章描述")
     # 奖章图标路径
     icon = models.TextField(verbose_name="奖章图标")
     # 获得条件：完成所有题目、完成指定数量题目、达到指定分数等
-    condition_type = models.TextField(
-        verbose_name="获得条件类型"
-    )  # all_problems, problem_count, score
+    condition_type = models.TextField(choices=BadgeConditionType.choices, verbose_name="获得条件类型")
     condition_value = models.IntegerField(default=0, verbose_name="条件值")
 
     class Meta:
@@ -90,17 +104,13 @@ class ProblemSetBadge(models.Model):
 
     def __str__(self):
         return f"{self.problemset.title} - {self.name}"
-    
+
     def recalculate_user_badges(self):
         """重新计算所有用户的徽章资格"""
         from django.db import transaction
 
         user_progresses = ProblemSetProgress.objects.filter(problemset=self.problemset)
-        new_badges = [
-            UserBadge(user=progress.user, badge=self)
-            for progress in user_progresses
-            if self._is_eligible(progress)
-        ]
+        new_badges = [UserBadge(user=progress.user, badge=self) for progress in user_progresses if self._is_eligible(progress)]
         with transaction.atomic():
             UserBadge.objects.filter(badge=self).delete()
             if new_badges:
@@ -118,9 +128,7 @@ class ProblemSetBadge(models.Model):
 
     def _check_user_badge_eligibility(self, progress):
         """检查并授予单个用户的徽章（供外部单次调用）"""
-        if self._is_eligible(progress) and not UserBadge.objects.filter(
-            user=progress.user, badge=self
-        ).exists():
+        if self._is_eligible(progress) and not UserBadge.objects.filter(user=progress.user, badge=self).exists():
             UserBadge.objects.create(user=progress.user, badge=self)
             return True
         return False
@@ -129,9 +137,7 @@ class ProblemSetBadge(models.Model):
 class ProblemSetProgress(models.Model):
     """题单进度模型"""
 
-    problemset = models.ForeignKey(
-        ProblemSet, on_delete=models.CASCADE, verbose_name="题单"
-    )
+    problemset = models.ForeignKey(ProblemSet, on_delete=models.CASCADE, verbose_name="题单")
     user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="用户")
     # 加入时间
     join_time = models.DateTimeField(auto_now_add=True, verbose_name="加入时间")
@@ -142,9 +148,7 @@ class ProblemSetProgress(models.Model):
     # 完成进度百分比
     progress_percentage = models.FloatField(default=0.0, verbose_name="完成进度")
     # 已完成的题目数量
-    completed_problems_count = models.IntegerField(
-        default=0, verbose_name="已完成题目数"
-    )
+    completed_problems_count = models.IntegerField(default=0, verbose_name="已完成题目数")
     # 总题目数量
     total_problems_count = models.IntegerField(default=0, verbose_name="总题目数")
     # 获得的总分
@@ -155,7 +159,9 @@ class ProblemSetProgress(models.Model):
 
     class Meta:
         db_table = "problemset_progress"
-        unique_together = (("problemset", "user"),)
+        constraints = [
+            models.UniqueConstraint(fields=["problemset", "user"], name="unique_problemset_progress_user"),
+        ]
         verbose_name = "题单进度"
         verbose_name_plural = "题单进度"
 
@@ -165,9 +171,7 @@ class ProblemSetProgress(models.Model):
     def update_progress(self):
         """更新进度信息"""
         # 获取题单中的所有题目
-        problemset_problems = ProblemSetProblem.objects.filter(
-            problemset=self.problemset
-        )
+        problemset_problems = ProblemSetProblem.objects.filter(problemset=self.problemset)
         self.total_problems_count = problemset_problems.count()
 
         # 获取当前题单中所有题目的ID集合（直接用 problem_id FK 字段，无需额外查询）
@@ -199,9 +203,7 @@ class ProblemSetProgress(models.Model):
 
         # 计算完成百分比
         if self.total_problems_count > 0:
-            self.progress_percentage = (
-                completed_count / self.total_problems_count
-            ) * 100
+            self.progress_percentage = (completed_count / self.total_problems_count) * 100
         else:
             self.progress_percentage = 0
 
@@ -223,17 +225,11 @@ class ProblemSetProgress(models.Model):
 
 class ProblemSetSubmission(models.Model):
     """题单提交记录模型"""
-    
-    problemset = models.ForeignKey(
-        ProblemSet, on_delete=models.CASCADE, verbose_name="题单"
-    )
+
+    problemset = models.ForeignKey(ProblemSet, on_delete=models.CASCADE, verbose_name="题单")
     user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="用户")
-    submission = models.ForeignKey(
-        "submission.Submission", on_delete=models.CASCADE, verbose_name="提交记录"
-    )
-    problem = models.ForeignKey(
-        "problem.Problem", on_delete=models.CASCADE, verbose_name="题目"
-    )
+    submission = models.ForeignKey("submission.Submission", on_delete=models.CASCADE, verbose_name="提交记录")
+    problem = models.ForeignKey("problem.Problem", on_delete=models.CASCADE, verbose_name="题目")
 
     class Meta:
         db_table = "problemset_submission"
@@ -253,34 +249,33 @@ class ProblemSetSubmission(models.Model):
     def submit_time(self):
         """提交时间"""
         return self.submission.create_time
-    
+
     @property
     def result(self):
         """提交结果"""
         return self.submission.result
-    
+
     @property
     def language(self):
         """编程语言"""
         return self.submission.language
 
+
 class UserBadge(models.Model):
     """用户奖章模型"""
 
     user = models.ForeignKey(User, on_delete=models.CASCADE, verbose_name="用户")
-    badge = models.ForeignKey(
-        ProblemSetBadge, on_delete=models.CASCADE, verbose_name="奖章"
-    )
+    badge = models.ForeignKey(ProblemSetBadge, on_delete=models.CASCADE, verbose_name="奖章")
     # 获得时间
     earned_time = models.DateTimeField(auto_now_add=True, verbose_name="获得时间")
 
     class Meta:
         db_table = "user_badge"
-        unique_together = (("user", "badge"),)
+        constraints = [
+            models.UniqueConstraint(fields=["user", "badge"], name="unique_user_badge"),
+        ]
         verbose_name = "用户奖章"
         verbose_name_plural = "用户奖章"
 
     def __str__(self):
         return f"{self.user.username} - {self.badge.name}"
-
-