From 159b2b4cc0bd98f59fd823c2c395774d17c42e23 Mon Sep 17 00:00:00 2001
From: cuom1999 <lephuocdinh99@gmail.com>
Date: Thu, 23 Nov 2023 23:16:01 -0600
Subject: [PATCH] Paginate contest summary

---
 dmoj/urls.py                            |   5 +-
 judge/views/contests.py                 | 102 ++++++++++++++----------
 templates/contest/contests_summary.html |   7 +-
 3 files changed, 68 insertions(+), 46 deletions(-)

diff --git a/dmoj/urls.py b/dmoj/urls.py
index 8ac0a29..adcb18d 100644
--- a/dmoj/urls.py
+++ b/dmoj/urls.py
@@ -518,9 +518,8 @@ urlpatterns = [
     ),
     url(r"^contests/", paged_list_view(contests.ContestList, "contest_list")),
     url(
-        r"^contests/summary/(?P<key>\w+)$",
-        contests.contests_summary_view,
-        name="contests_summary",
+        r"^contests/summary/(?P<key>\w+)/",
+        paged_list_view(contests.ContestsSummaryView, "contests_summary"),
     ),
     url(r"^course/", paged_list_view(course.CourseList, "course_list")),
     url(
diff --git a/judge/views/contests.py b/judge/views/contests.py
index 80e5fb5..1ae746b 100644
--- a/judge/views/contests.py
+++ b/judge/views/contests.py
@@ -1426,55 +1426,73 @@ ContestsSummaryData = namedtuple(
 )
 
 
-def contests_summary_view(request, key):
-    try:
-        contests_summary = ContestsSummary.objects.get(key=key)
-    except:
-        raise Http404()
+class ContestsSummaryView(DiggPaginatorMixin, ListView):
+    paginate_by = 50
+    template_name = "contest/contests_summary.html"
 
-    cache_key = "csv:" + key
-    context = cache.get(cache_key)
-    if context:
-        return render(request, "contest/contests_summary.html", context)
+    def get(self, *args, **kwargs):
+        try:
+            self.contests_summary = ContestsSummary.objects.get(key=kwargs["key"])
+        except:
+            raise Http404()
+        return super().get(*args, **kwargs)
 
-    scores_system = contests_summary.scores
-    contests = contests_summary.contests.all()
-    total_points = defaultdict(int)
-    result_per_contest = defaultdict(lambda: [(0, 0)] * len(contests))
-    user_css_class = {}
+    def _get_contests_and_ranking(self):
+        contests_summary = self.contests_summary
+        cache_key = "csv:" + contests_summary.key
+        result = cache.get(cache_key)
+        if result:
+            return result
 
-    for i in range(len(contests)):
-        contest = contests[i]
-        users, problems = get_contest_ranking_list(request, contest)
-        for rank, user in users:
-            curr_score = 0
-            if rank - 1 < len(scores_system):
-                curr_score = scores_system[rank - 1]
-            total_points[user.user] += curr_score
-            result_per_contest[user.user][i] = (curr_score, rank)
-            user_css_class[user.user] = user.css_class
+        scores_system = contests_summary.scores
+        contests = contests_summary.contests.all()
+        total_points = defaultdict(int)
+        result_per_contest = defaultdict(lambda: [(0, 0)] * len(contests))
+        user_css_class = {}
 
-    sorted_total_points = [
-        ContestsSummaryData(
-            user=user,
-            points=total_points[user],
-            point_contests=result_per_contest[user],
-            css_class=user_css_class[user],
-        )
-        for user in total_points
-    ]
+        for i in range(len(contests)):
+            contest = contests[i]
+            users, problems = get_contest_ranking_list(self.request, contest)
+            for rank, user in users:
+                curr_score = 0
+                if rank - 1 < len(scores_system):
+                    curr_score = scores_system[rank - 1]
+                total_points[user.user] += curr_score
+                result_per_contest[user.user][i] = (curr_score, rank)
+                user_css_class[user.user] = user.css_class
 
-    sorted_total_points.sort(key=lambda x: x.points, reverse=True)
-    total_rank = ranker(sorted_total_points)
+        sorted_total_points = [
+            ContestsSummaryData(
+                user=user,
+                points=total_points[user],
+                point_contests=result_per_contest[user],
+                css_class=user_css_class[user],
+            )
+            for user in total_points
+        ]
 
-    context = {
-        "total_rank": list(total_rank),
-        "title": _("Contests Summary"),
-        "contests": contests,
-    }
-    cache.set(cache_key, context)
+        sorted_total_points.sort(key=lambda x: x.points, reverse=True)
+        total_rank = ranker(sorted_total_points)
 
-    return render(request, "contest/contests_summary.html", context)
+        result = {
+            "total_rank": list(total_rank),
+            "contests": contests,
+        }
+        cache.set(cache_key, result)
+        return result
+
+    def get_queryset(self):
+        rank_and_contests = self._get_contests_and_ranking()
+        total_rank = rank_and_contests["total_rank"]
+        self.contests = rank_and_contests["contests"]
+        return total_rank
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context["contests"] = self.contests
+        context["title"] = _("Contests")
+        context["first_page_href"] = "."
+        return context
 
 
 @receiver([post_save, post_delete], sender=ContestsSummary)
diff --git a/templates/contest/contests_summary.html b/templates/contest/contests_summary.html
index a7fbe12..3d5a999 100644
--- a/templates/contest/contests_summary.html
+++ b/templates/contest/contests_summary.html
@@ -42,7 +42,7 @@
       </tr>
     </thead>
     <tbody>
-      {% for rank, item in total_rank %}
+      {% for rank, item in page_obj %}
         <tr>
           <td>
             {{ rank }}
@@ -68,4 +68,9 @@
       {% endfor %}
     </tbody>
   </table>
+  {% if page_obj and page_obj.num_pages > 1 %}
+    <div style="margin-top: 10px;">
+      {% include "list-pages.html" %}
+    </div>
+  {% endif %}
 {% endblock %}