Expose participation start and end times through API; fixes #1586

DMOJ · Feb 26, 2021 · 0c009fa · 0c009fa
1 parent 26cbe26
commit 0c009fa
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/judge/views/api/api_v2.py b/judge/views/api/api_v2.py
@@ -221,6 +221,10 @@ def get_object_data(self, contest):
             .order_by('-score', 'cumtime', 'tiebreaker')
         )
 
+        # Setting contest attribute to reduce db queries in .start and .end_time
+        for participation in participations:
+            participation.contest = contest
+
         return {
             'key': contest.key,
             'name': contest.name,
@@ -257,6 +261,8 @@ def get_object_data(self, contest):
             'rankings': [
                 {
                     'user': participation.username,
+                    'start_time': participation.start.isoformat(),
+                    'end_time': participation.end_time.isoformat(),
                     'score': participation.score,
                     'cumulative_time': participation.cumtime,
                     'tiebreaker': participation.tiebreaker,
@@ -303,6 +309,10 @@ def get_unfiltered_queryset(self):
             .only(
                 'user__user__username',
                 'contest__key',
+                'contest__start_time',
+                'contest__end_time',
+                'contest__time_limit',
+                'real_start',
                 'score',
                 'cumtime',
                 'tiebreaker',
@@ -315,6 +325,8 @@ def get_object_data(self, participation):
         return {
             'user': participation.user.username,
             'contest': participation.contest.key,
+            'start_time': participation.start.isoformat(),
+            'end_time': participation.end_time.isoformat(),
             'score': participation.score,
             'cumulative_time': participation.cumtime,
             'tiebreaker': participation.tiebreaker,