Add actions for contest locking; #595

DMOJ · May 13, 2020 · 551fedc · 551fedc
1 parent c5bd903
commit 551fedc
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/judge/admin/contest.py b/judge/admin/contest.py
@@ -139,6 +139,10 @@ def get_actions(self, request):
             for action in ('make_visible', 'make_hidden'):
                 actions[action] = self.get_action(action)
 
+        if request.user.has_perm('judge.contest_lock'):
+            for action in ('set_locked', 'set_unlocked'):
+                actions[action] = self.get_action(action)
+
         return actions
 
     def get_queryset(self, request):
@@ -212,13 +216,45 @@ def make_hidden(self, request, queryset):
                                              count) % count)
     make_hidden.short_description = _('Mark contests as hidden')
 
+    def set_locked(self, request, queryset):
+        for row in queryset:
+            self.set_is_locked(row, True)
+        count = queryset.count()
+        self.message_user(request, ungettext('%d contest successfully locked.',
+                                             '%d contests successfully locked.',
+                                             count) % count)
+    set_locked.short_description = _('Lock contest submissions')
+
+    def set_unlocked(self, request, queryset):
+        for row in queryset:
+            self.set_is_locked(row, False)
+        count = queryset.count()
+        self.message_user(request, ungettext('%d contest successfully unlocked.',
+                                             '%d contests successfully unlocked.',
+                                             count) % count)
+    set_unlocked.short_description = _('Unlock contest submissions')
+
+    def set_is_locked(self, contest, is_locked):
+        with transaction.atomic():
+            contest.is_locked = is_locked
+            contest.save()
+            ContestSubmission.objects.filter(participation__contest=contest).update(submission__is_locked=is_locked)
+
     def get_urls(self):
         return [
             url(r'^rate/all/$', self.rate_all_view, name='judge_contest_rate_all'),
             url(r'^(\d+)/rate/$', self.rate_view, name='judge_contest_rate'),
             url(r'^(\d+)/judge/(\d+)/$', self.rejudge_view, name='judge_contest_rejudge'),
+            url(r'^(\d+)/lock/$', self.lock_view, name='judge_contest_lock'),
         ] + super(ContestAdmin, self).get_urls()
 
+    def lock_view(self, request, id):
+        if not request.user.has_perm('judge.contest_lock'):
+            raise PermissionDenied()
+        contest = get_object_or_404(Contest, id=id)
+        self.set_is_locked(contest, not contest.is_locked)
+        return HttpResponseRedirect(reverse('admin:judge_contest_change', args=(id,)))
+
     def rejudge_view(self, request, contest_id, problem_id):
         queryset = ContestSubmission.objects.filter(problem_id=problem_id).select_related('submission')
         for model in queryset:

diff --git a/templates/admin/judge/contest/change_form.html b/templates/admin/judge/contest/change_form.html
@@ -20,4 +20,11 @@
             <span class="text">{% trans "Rate" %}</span>
         </a>
     {% endif %}
+    {% if original and perms.judge.contest_lock %}
+        <a style="display: none" title="{% trans "Toggle Lock" %}" href="{% url 'admin:judge_contest_lock' original.pk %}"
+           class="button rerate-link">
+            <i class="fa fa-lg fa-lock"></i>
+            <span class="text">{% trans "Toggle Lock" %}</span>
+        </a>
+    {% endif %}
 {% endblock %}