Add contest freeze

judge/contest_format/atcoder.py

@@ -10,7 +10,7 @@ from django.utils.translation import gettext_lazy
 
 from judge.contest_format.default import DefaultContestFormat
 from judge.contest_format.registry import register_contest_format
-from judge.timezone import from_database_time
+from judge.timezone import from_database_time, to_database_time
 from judge.utils.timedelta import nice_repr
 
 
@@ -54,6 +54,10 @@ class AtCoderContestFormat(DefaultContestFormat):
         points = 0
         format_data = {}
 
+        frozen_time = self.contest.end_time
+        if self.contest.freeze_after:
+            frozen_time = participation.start + self.contest.freeze_after
+
         with connection.cursor() as cursor:
             cursor.execute(
                 """
@@ -66,9 +70,10 @@ class AtCoderContestFormat(DefaultContestFormat):
                 FROM judge_contestproblem cp INNER JOIN
                      judge_contestsubmission cs ON (cs.problem_id = cp.id AND cs.participation_id = %s) LEFT OUTER JOIN
                      judge_submission sub ON (sub.id = cs.submission_id)
+                WHERE sub.date < %s
                 GROUP BY cp.id
             """,
-                (participation.id, participation.id),
+                (participation.id, participation.id, to_database_time(frozen_time)),
             )
 
             for score, time, prob in cursor.fetchall():
@@ -100,6 +105,7 @@ class AtCoderContestFormat(DefaultContestFormat):
                 format_data[str(prob)] = {"time": dt, "points": score, "penalty": prev}
                 points += score
 
+        self.handle_frozen_state(participation, format_data)
         participation.cumtime = cumtime + penalty
         participation.score = points
         participation.tiebreaker = 0
@@ -114,7 +120,7 @@ class AtCoderContestFormat(DefaultContestFormat):
                     '<small style="color:red"> ({penalty})</small>',
                     penalty=floatformat(format_data["penalty"]),
                 )
-                if format_data["penalty"]
+                if format_data.get("penalty")
                 else ""
             )
             return format_html(
@@ -129,6 +135,7 @@ class AtCoderContestFormat(DefaultContestFormat):
                     + self.best_solution_state(
                         format_data["points"], contest_problem.points
                     )
+                    + (" frozen" if format_data.get("frozen") else "")
                 ),
                 url=reverse(
                     "contest_user_submissions_ajax",