更新 ContestRank的生成方法

contest/models.py

@@ -5,7 +5,8 @@ from django.utils.timezone import now
 from account.models import User
 from problem.models import AbstractProblem
 from group.models import Group
-from utils.models import RichTextField
+from utils.models import RichTextField, JsonField
+from judge.judger.result import result
 
 
 GROUP_CONTEST = 0
@@ -104,3 +105,66 @@ class ContestSubmission(models.Model):
 
     class Meta:
         db_table = "contest_submission"
+
+
+class ContestRank(models.Model):
+    user = models.ForeignKey(User)
+    contest = models.ForeignKey(Contest)
+    total_submission_number = models.IntegerField(default=0)
+    total_ac_number = models.IntegerField(default=0)
+    # ac 的题目才要加到这个字段里面 = ac 时间 + 错误次数 * 20 * 60
+    # 没有 ac 的题目不计算罚时 单位是秒
+    total_time = models.IntegerField(default=0)
+    # 数据结构{23: {"is_ac": True, "ac_time": 8999, "error_number": 2, "is_first_ac": True}}
+    # key 是比赛题目的id
+    submission_info = JsonField(default={})
+
+    def update_rank(self, submission):
+        if not submission.contest_id or submission.contest_id != self.contest_id:
+            raise ValueError("Error submission type")
+
+        # 这道题以前提交过
+        if submission.problem_id in self.problem_info:
+            info = self.submission_info[submission.problem_id]
+            # 如果这道题目已经 ac 了就跳过
+            if info["is_ac"]:
+                return
+
+            self.total_submission_number += 1
+
+            if submission.result == result["accepted"]:
+
+                self.total_ac_number += 1
+
+                info["is_ac"] = True
+                info["ac_time"] = (submission.create_time - self.contest.start_time).total_seconds()
+
+                # 之前已经提交过，但是是错误的，这次提交是正确的。错误的题目不计入罚时
+                self.total_time += (info["ac_time"] + info["error_time"] * 20 * 60)
+                problem = ContestProblem.objects.get(id=submission.problem_id)
+                if problem.total_accepted_number == 0:
+                    info["is_first_ac"] = True
+
+            else:
+                info["error_number"] += 1
+                info["is_ac"] = False
+
+        else:
+            # 第一次提交这道题目
+            self.total_submission_number += 1
+            info = {"is_ac": False, "ac_time": 0, "error_number": 0, "is_first_ac": False}
+            if submission.result == result["accepted"]:
+                self.total_ac_number += 1
+                info["is_ac"] = True
+                info["ac_time"] = (submission.create_time - self.contest.start_time).total_seconds()
+                self.total_time += info["ac_time"]
+                problem = ContestProblem.objects.get(id=submission.problem_id)
+
+                if problem.total_accepted_number == 0:
+                    info["is_first_ac"] = True
+
+            else:
+                info["is_ac"] = False
+                info["error_number"] = 1
+        self.submission_info[submission.problem_id] = info
+        self.save()