from django.db import models from django.utils.timezone import now from account.models import User from problem.models import Problem from utils.models import RichTextField, JSONField class ProblemSet(models.Model): """题单模型""" title = models.TextField(verbose_name="题单标题") description = RichTextField(verbose_name="题单描述") # 创建者 created_by = models.ForeignKey( User, on_delete=models.CASCADE, verbose_name="创建者" ) # 创建时间 create_time = models.DateTimeField(auto_now_add=True, verbose_name="创建时间") # 更新时间 last_update_time = models.DateTimeField(auto_now=True, verbose_name="更新时间") # 是否可见 visible = models.BooleanField(default=True, verbose_name="是否可见") # 题单难度等级 difficulty = models.TextField(default="Easy", verbose_name="难度等级") # 题单状态 status = models.TextField( default="draft", verbose_name="状态" ) # active, archived, draft class Meta: db_table = "problemset" ordering = ("-create_time",) verbose_name = "题单" verbose_name_plural = "题单" def __str__(self): return self.title class ProblemSetProblem(models.Model): """题单题目关联模型""" 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="顺序") # 是否为必做题 is_required = models.BooleanField(default=True, verbose_name="是否必做") # 题目在题单中的分值 score = models.IntegerField(default=0, verbose_name="分值") # 题目提示信息 hint = models.TextField(null=True, blank=True, verbose_name="提示") class Meta: db_table = "problemset_problem" unique_together = (("problemset", "problem"),) ordering = ("order",) verbose_name = "题单题目" verbose_name_plural = "题单题目" def __str__(self): return f"{self.problemset.title} - {self.problem.title}" class ProblemSetBadge(models.Model): """题单奖章模型""" 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_value = models.IntegerField(default=0, verbose_name="条件值") class Meta: db_table = "problemset_badge" verbose_name = "题单奖章" verbose_name_plural = "题单奖章" def __str__(self): return f"{self.problemset.title} - {self.name}" def recalculate_user_badges(self): """重新计算所有用户的徽章资格""" # 获取所有已加入该题单的用户进度 user_progresses = ProblemSetProgress.objects.filter(problemset=self.problemset) # 删除该徽章的所有现有用户徽章记录 UserBadge.objects.filter(badge=self).delete() # 重新评估每个用户的徽章资格 for progress in user_progresses: self._check_user_badge_eligibility(progress) def _check_user_badge_eligibility(self, progress): """检查用户是否符合该徽章的条件""" # 检查是否已经拥有该徽章 if UserBadge.objects.filter(user=progress.user, badge=self).exists(): return False # 根据条件类型检查用户是否符合条件 if self.condition_type == "all_problems": if progress.completed_problems_count == progress.total_problems_count: UserBadge.objects.create(user=progress.user, badge=self) return True elif self.condition_type == "problem_count": if progress.completed_problems_count >= self.condition_value: UserBadge.objects.create(user=progress.user, badge=self) return True elif self.condition_type == "score": if progress.total_score >= self.condition_value: UserBadge.objects.create(user=progress.user, badge=self) return True return False class ProblemSetProgress(models.Model): """题单进度模型""" 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="加入时间") # 完成时间 complete_time = models.DateTimeField(null=True, blank=True, verbose_name="完成时间") # 是否完成 is_completed = models.BooleanField(default=False, verbose_name="是否完成") # 完成进度百分比 progress_percentage = models.FloatField(default=0.0, verbose_name="完成进度") # 已完成的题目数量 completed_problems_count = models.IntegerField( default=0, verbose_name="已完成题目数" ) # 总题目数量 total_problems_count = models.IntegerField(default=0, verbose_name="总题目数") # 获得的总分 total_score = models.IntegerField(default=0, verbose_name="总分") # 用户在该题单中的详细进度信息 # {"problem_id": {"score": 20, "submit_time": "2024-01-01T00:00:00Z"}} progress_detail = JSONField(default=dict, verbose_name="详细进度") class Meta: db_table = "problemset_progress" unique_together = (("problemset", "user"),) verbose_name = "题单进度" verbose_name_plural = "题单进度" def __str__(self): return f"{self.user.username} - {self.problemset.title}" def update_progress(self): """更新进度信息""" # 获取题单中的所有题目 problemset_problems = ProblemSetProblem.objects.filter( problemset=self.problemset ) self.total_problems_count = problemset_problems.count() # 计算已完成题目数 completed_count = 0 total_score = 0 for psp in problemset_problems: problem_id = str(psp.problem.id) if problem_id in self.progress_detail: problem_progress = self.progress_detail[problem_id] completed_count += 1 total_score += problem_progress.get("score", 0) self.completed_problems_count = completed_count self.total_score = total_score # 计算完成百分比 if self.total_problems_count > 0: self.progress_percentage = ( completed_count / self.total_problems_count ) * 100 else: self.progress_percentage = 0 # 检查是否完成 self.is_completed = completed_count == self.total_problems_count if self.is_completed and not self.complete_time: self.complete_time = now() self.save() @classmethod def sync_all_progress_for_problemset(cls, problemset): """同步指定题单的所有用户进度""" progresses = cls.objects.filter(problemset=problemset) for progress in progresses: progress.update_progress() return progresses.count() class ProblemSetSubmission(models.Model): """题单提交记录模型""" 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="题目" ) class Meta: db_table = "problemset_submission" ordering = ("-submission__create_time",) verbose_name = "题单提交记录" verbose_name_plural = "题单提交记录" indexes = [ models.Index(fields=["problemset", "user"]), models.Index(fields=["problemset", "problem"]), models.Index(fields=["user"]), ] def __str__(self): return f"{self.user.username} - {self.problemset.title} - {self.problem.title}" @property 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="奖章" ) # 获得时间 earned_time = models.DateTimeField(auto_now_add=True, verbose_name="获得时间") class Meta: db_table = "user_badge" unique_together = (("user", "badge"),) verbose_name = "用户奖章" verbose_name_plural = "用户奖章" def __str__(self): return f"{self.user.username} - {self.badge.name}"