504 lines
17 KiB
Python
504 lines
17 KiB
Python
from django.conf.urls import url
|
|
from django.contrib import admin
|
|
from django.core.exceptions import PermissionDenied
|
|
from django.db import connection, transaction
|
|
from django.db.models import Q, TextField
|
|
from django.forms import ModelForm, ModelMultipleChoiceField, TextInput
|
|
from django.http import Http404, HttpResponseRedirect
|
|
from django.shortcuts import get_object_or_404
|
|
from django.urls import reverse, reverse_lazy
|
|
from django.utils import timezone
|
|
from django.utils.html import format_html
|
|
from django.utils.translation import gettext_lazy as _, ungettext
|
|
from reversion.admin import VersionAdmin
|
|
from reversion_compare.admin import CompareVersionAdmin
|
|
|
|
from django_ace import AceWidget
|
|
from judge.models import Contest, ContestProblem, ContestSubmission, Profile, Rating
|
|
from judge.ratings import rate_contest
|
|
from judge.widgets import (
|
|
AdminHeavySelect2MultipleWidget,
|
|
AdminHeavySelect2Widget,
|
|
AdminPagedownWidget,
|
|
AdminSelect2MultipleWidget,
|
|
AdminSelect2Widget,
|
|
HeavyPreviewAdminPageDownWidget,
|
|
)
|
|
|
|
|
|
class AdminHeavySelect2Widget(AdminHeavySelect2Widget):
|
|
@property
|
|
def is_hidden(self):
|
|
return False
|
|
|
|
|
|
class ContestTagForm(ModelForm):
|
|
contests = ModelMultipleChoiceField(
|
|
label=_("Included contests"),
|
|
queryset=Contest.objects.all(),
|
|
required=False,
|
|
widget=AdminHeavySelect2MultipleWidget(data_view="contest_select2"),
|
|
)
|
|
|
|
|
|
class ContestTagAdmin(admin.ModelAdmin):
|
|
fields = ("name", "color", "description", "contests")
|
|
list_display = ("name", "color")
|
|
actions_on_top = True
|
|
actions_on_bottom = True
|
|
form = ContestTagForm
|
|
|
|
if AdminPagedownWidget is not None:
|
|
formfield_overrides = {
|
|
TextField: {"widget": AdminPagedownWidget},
|
|
}
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
super(ContestTagAdmin, self).save_model(request, obj, form, change)
|
|
obj.contests.set(form.cleaned_data["contests"])
|
|
|
|
def get_form(self, request, obj=None, **kwargs):
|
|
form = super(ContestTagAdmin, self).get_form(request, obj, **kwargs)
|
|
if obj is not None:
|
|
form.base_fields["contests"].initial = obj.contests.all()
|
|
return form
|
|
|
|
|
|
class ContestProblemInlineForm(ModelForm):
|
|
class Meta:
|
|
widgets = {
|
|
"problem": AdminHeavySelect2Widget(data_view="problem_select2"),
|
|
"hidden_subtasks": TextInput(attrs={"size": "3"}),
|
|
"points": TextInput(attrs={"size": "1"}),
|
|
"order": TextInput(attrs={"size": "1"}),
|
|
"output_prefix_override": TextInput(attrs={"size": "1"}),
|
|
}
|
|
|
|
|
|
class ContestProblemInline(admin.TabularInline):
|
|
model = ContestProblem
|
|
verbose_name = _("Problem")
|
|
verbose_name_plural = "Problems"
|
|
fields = (
|
|
"problem",
|
|
"points",
|
|
"partial",
|
|
"is_pretested",
|
|
"max_submissions",
|
|
"hidden_subtasks",
|
|
"output_prefix_override",
|
|
"order",
|
|
"rejudge_column",
|
|
)
|
|
readonly_fields = ("rejudge_column",)
|
|
form = ContestProblemInlineForm
|
|
|
|
def rejudge_column(self, obj):
|
|
if obj.id is None:
|
|
return ""
|
|
return format_html(
|
|
'<a class="button rejudge-link" href="{}">Rejudge</a>',
|
|
reverse("admin:judge_contest_rejudge", args=(obj.contest.id, obj.id)),
|
|
)
|
|
|
|
rejudge_column.short_description = ""
|
|
|
|
|
|
class ContestForm(ModelForm):
|
|
def __init__(self, *args, **kwargs):
|
|
super(ContestForm, self).__init__(*args, **kwargs)
|
|
if "rate_exclude" in self.fields:
|
|
if self.instance and self.instance.id:
|
|
self.fields["rate_exclude"].queryset = Profile.objects.filter(
|
|
contest_history__contest=self.instance
|
|
).distinct()
|
|
else:
|
|
self.fields["rate_exclude"].queryset = Profile.objects.none()
|
|
self.fields["banned_users"].widget.can_add_related = False
|
|
self.fields["view_contest_scoreboard"].widget.can_add_related = False
|
|
|
|
def clean(self):
|
|
cleaned_data = super(ContestForm, self).clean()
|
|
cleaned_data["banned_users"].filter(
|
|
current_contest__contest=self.instance
|
|
).update(current_contest=None)
|
|
|
|
class Meta:
|
|
widgets = {
|
|
"authors": AdminHeavySelect2MultipleWidget(data_view="profile_select2"),
|
|
"curators": AdminHeavySelect2MultipleWidget(data_view="profile_select2"),
|
|
"testers": AdminHeavySelect2MultipleWidget(data_view="profile_select2"),
|
|
"private_contestants": AdminHeavySelect2MultipleWidget(
|
|
data_view="profile_select2", attrs={"style": "width: 100%"}
|
|
),
|
|
"organizations": AdminHeavySelect2MultipleWidget(
|
|
data_view="organization_select2"
|
|
),
|
|
"tags": AdminSelect2MultipleWidget,
|
|
"banned_users": AdminHeavySelect2MultipleWidget(
|
|
data_view="profile_select2", attrs={"style": "width: 100%"}
|
|
),
|
|
"view_contest_scoreboard": AdminHeavySelect2MultipleWidget(
|
|
data_view="profile_select2", attrs={"style": "width: 100%"}
|
|
),
|
|
}
|
|
|
|
if HeavyPreviewAdminPageDownWidget is not None:
|
|
widgets["description"] = HeavyPreviewAdminPageDownWidget(
|
|
preview=reverse_lazy("contest_preview")
|
|
)
|
|
|
|
|
|
class ContestAdmin(CompareVersionAdmin):
|
|
fieldsets = (
|
|
(None, {"fields": ("key", "name", "authors", "curators", "testers")}),
|
|
(
|
|
_("Settings"),
|
|
{
|
|
"fields": (
|
|
"is_visible",
|
|
"use_clarifications",
|
|
"hide_problem_tags",
|
|
"scoreboard_visibility",
|
|
"run_pretests_only",
|
|
"points_precision",
|
|
)
|
|
},
|
|
),
|
|
(
|
|
_("Scheduling"),
|
|
{"fields": ("start_time", "end_time", "time_limit", "freeze_after")},
|
|
),
|
|
(
|
|
_("Details"),
|
|
{
|
|
"fields": (
|
|
"description",
|
|
"og_image",
|
|
"logo_override_image",
|
|
"tags",
|
|
"summary",
|
|
)
|
|
},
|
|
),
|
|
(
|
|
_("Format"),
|
|
{"fields": ("format_name", "format_config", "problem_label_script")},
|
|
),
|
|
(
|
|
_("Rating"),
|
|
{
|
|
"fields": (
|
|
"is_rated",
|
|
"rate_all",
|
|
"rating_floor",
|
|
"rating_ceiling",
|
|
"rate_exclude",
|
|
)
|
|
},
|
|
),
|
|
(
|
|
_("Access"),
|
|
{
|
|
"fields": (
|
|
"access_code",
|
|
"private_contestants",
|
|
"organizations",
|
|
"view_contest_scoreboard",
|
|
)
|
|
},
|
|
),
|
|
(_("Justice"), {"fields": ("banned_users",)}),
|
|
)
|
|
list_display = (
|
|
"key",
|
|
"name",
|
|
"is_visible",
|
|
"is_rated",
|
|
"start_time",
|
|
"end_time",
|
|
"time_limit",
|
|
"user_count",
|
|
)
|
|
search_fields = ("key", "name")
|
|
inlines = [ContestProblemInline]
|
|
actions_on_top = True
|
|
actions_on_bottom = True
|
|
form = ContestForm
|
|
change_list_template = "admin/judge/contest/change_list.html"
|
|
filter_horizontal = ["rate_exclude"]
|
|
date_hierarchy = "start_time"
|
|
|
|
def get_actions(self, request):
|
|
actions = super(ContestAdmin, self).get_actions(request)
|
|
|
|
if request.user.has_perm(
|
|
"judge.change_contest_visibility"
|
|
) or request.user.has_perm("judge.create_private_contest"):
|
|
for action in ("make_visible", "make_hidden"):
|
|
actions[action] = self.get_action(action)
|
|
|
|
return actions
|
|
|
|
def get_queryset(self, request):
|
|
queryset = Contest.objects.all()
|
|
if request.user.has_perm("judge.edit_all_contest"):
|
|
return queryset
|
|
else:
|
|
return queryset.filter(
|
|
Q(authors=request.profile) | Q(curators=request.profile)
|
|
).distinct()
|
|
|
|
def get_readonly_fields(self, request, obj=None):
|
|
readonly = []
|
|
if not request.user.has_perm("judge.contest_rating"):
|
|
readonly += ["is_rated", "rate_all", "rate_exclude"]
|
|
if not request.user.has_perm("judge.contest_access_code"):
|
|
readonly += ["access_code"]
|
|
if not request.user.has_perm("judge.create_private_contest"):
|
|
readonly += [
|
|
"private_contestants",
|
|
"organizations",
|
|
]
|
|
if not request.user.has_perm("judge.change_contest_visibility"):
|
|
readonly += ["is_visible"]
|
|
if not request.user.has_perm("judge.contest_problem_label"):
|
|
readonly += ["problem_label_script"]
|
|
return readonly
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
# `is_visible` will not appear in `cleaned_data` if user cannot edit it
|
|
if form.cleaned_data.get("is_visible") and not request.user.has_perm(
|
|
"judge.change_contest_visibility"
|
|
):
|
|
if (
|
|
not len(form.cleaned_data["organizations"]) > 0
|
|
and not len(form.cleaned_data["private_contestants"]) > 0
|
|
):
|
|
raise PermissionDenied
|
|
if not request.user.has_perm("judge.create_private_contest"):
|
|
raise PermissionDenied
|
|
|
|
super().save_model(request, obj, form, change)
|
|
|
|
# We need this flag because `save_related` deals with the inlines, but does not know if we have already rescored
|
|
self._rescored = False
|
|
if form.changed_data and any(
|
|
f in form.changed_data
|
|
for f in (
|
|
"start_time",
|
|
"end_time",
|
|
"time_limit",
|
|
"format_config",
|
|
"format_name",
|
|
"freeze_after",
|
|
)
|
|
):
|
|
self._rescore(obj.key)
|
|
self._rescored = True
|
|
|
|
def save_related(self, request, form, formsets, change):
|
|
super().save_related(request, form, formsets, change)
|
|
# Only rescored if we did not already do so in `save_model`
|
|
if not self._rescored and any(formset.has_changed() for formset in formsets):
|
|
self._rescore(form.cleaned_data["key"])
|
|
obj = form.instance
|
|
obj.is_organization_private = obj.organizations.count() > 0
|
|
obj.is_private = obj.private_contestants.count() > 0
|
|
obj.save()
|
|
|
|
def has_change_permission(self, request, obj=None):
|
|
if not request.user.has_perm("judge.edit_own_contest"):
|
|
return False
|
|
if obj is None:
|
|
return True
|
|
return obj.is_editable_by(request.user)
|
|
|
|
def _rescore(self, contest_key):
|
|
from judge.tasks import rescore_contest
|
|
|
|
transaction.on_commit(rescore_contest.s(contest_key).delay)
|
|
|
|
def make_visible(self, request, queryset):
|
|
if not request.user.has_perm("judge.change_contest_visibility"):
|
|
queryset = queryset.filter(
|
|
Q(is_private=True) | Q(is_organization_private=True)
|
|
)
|
|
count = queryset.update(is_visible=True)
|
|
self.message_user(
|
|
request,
|
|
ungettext(
|
|
"%d contest successfully marked as visible.",
|
|
"%d contests successfully marked as visible.",
|
|
count,
|
|
)
|
|
% count,
|
|
)
|
|
|
|
make_visible.short_description = _("Mark contests as visible")
|
|
|
|
def make_hidden(self, request, queryset):
|
|
if not request.user.has_perm("judge.change_contest_visibility"):
|
|
queryset = queryset.filter(
|
|
Q(is_private=True) | Q(is_organization_private=True)
|
|
)
|
|
count = queryset.update(is_visible=True)
|
|
self.message_user(
|
|
request,
|
|
ungettext(
|
|
"%d contest successfully marked as hidden.",
|
|
"%d contests successfully marked as hidden.",
|
|
count,
|
|
)
|
|
% count,
|
|
)
|
|
|
|
make_hidden.short_description = _("Mark contests as hidden")
|
|
|
|
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"
|
|
),
|
|
] + super(ContestAdmin, self).get_urls()
|
|
|
|
def rejudge_view(self, request, contest_id, problem_id):
|
|
queryset = ContestSubmission.objects.filter(
|
|
problem_id=problem_id
|
|
).select_related("submission")
|
|
for model in queryset:
|
|
model.submission.judge(rejudge=True)
|
|
|
|
self.message_user(
|
|
request,
|
|
ungettext(
|
|
"%d submission was successfully scheduled for rejudging.",
|
|
"%d submissions were successfully scheduled for rejudging.",
|
|
len(queryset),
|
|
)
|
|
% len(queryset),
|
|
)
|
|
return HttpResponseRedirect(
|
|
reverse("admin:judge_contest_change", args=(contest_id,))
|
|
)
|
|
|
|
def rate_all_view(self, request):
|
|
if not request.user.has_perm("judge.contest_rating"):
|
|
raise PermissionDenied()
|
|
with transaction.atomic():
|
|
with connection.cursor() as cursor:
|
|
cursor.execute("TRUNCATE TABLE `%s`" % Rating._meta.db_table)
|
|
Profile.objects.update(rating=None)
|
|
for contest in Contest.objects.filter(
|
|
is_rated=True, end_time__lte=timezone.now()
|
|
).order_by("end_time"):
|
|
rate_contest(contest)
|
|
return HttpResponseRedirect(reverse("admin:judge_contest_changelist"))
|
|
|
|
def rate_view(self, request, id):
|
|
if not request.user.has_perm("judge.contest_rating"):
|
|
raise PermissionDenied()
|
|
contest = get_object_or_404(Contest, id=id)
|
|
if not contest.is_rated or not contest.ended:
|
|
raise Http404()
|
|
with transaction.atomic():
|
|
contest.rate()
|
|
return HttpResponseRedirect(
|
|
request.META.get("HTTP_REFERER", reverse("admin:judge_contest_changelist"))
|
|
)
|
|
|
|
def get_form(self, request, obj=None, **kwargs):
|
|
form = super(ContestAdmin, self).get_form(request, obj, **kwargs)
|
|
if "problem_label_script" in form.base_fields:
|
|
# form.base_fields['problem_label_script'] does not exist when the user has only view permission
|
|
# on the model.
|
|
form.base_fields["problem_label_script"].widget = AceWidget(
|
|
"lua", request.profile.ace_theme
|
|
)
|
|
|
|
perms = ("edit_own_contest", "edit_all_contest")
|
|
form.base_fields["curators"].queryset = Profile.objects.filter(
|
|
Q(user__is_superuser=True)
|
|
| Q(user__groups__permissions__codename__in=perms)
|
|
| Q(user__user_permissions__codename__in=perms),
|
|
).distinct()
|
|
return form
|
|
|
|
|
|
class ContestParticipationForm(ModelForm):
|
|
class Meta:
|
|
widgets = {
|
|
"contest": AdminSelect2Widget(),
|
|
"user": AdminHeavySelect2Widget(data_view="profile_select2"),
|
|
}
|
|
|
|
|
|
class ContestParticipationAdmin(admin.ModelAdmin):
|
|
fields = ("contest", "user", "real_start", "virtual", "is_disqualified")
|
|
list_display = (
|
|
"contest",
|
|
"username",
|
|
"show_virtual",
|
|
"real_start",
|
|
"score",
|
|
"cumtime",
|
|
"tiebreaker",
|
|
)
|
|
actions = ["recalculate_results"]
|
|
actions_on_bottom = actions_on_top = True
|
|
search_fields = ("contest__key", "contest__name", "user__user__username")
|
|
form = ContestParticipationForm
|
|
date_hierarchy = "real_start"
|
|
|
|
def get_queryset(self, request):
|
|
return (
|
|
super(ContestParticipationAdmin, self)
|
|
.get_queryset(request)
|
|
.only(
|
|
"contest__name",
|
|
"contest__format_name",
|
|
"contest__format_config",
|
|
"user__user__username",
|
|
"real_start",
|
|
"score",
|
|
"cumtime",
|
|
"tiebreaker",
|
|
"virtual",
|
|
)
|
|
)
|
|
|
|
def save_model(self, request, obj, form, change):
|
|
super().save_model(request, obj, form, change)
|
|
if form.changed_data and "is_disqualified" in form.changed_data:
|
|
obj.set_disqualified(obj.is_disqualified)
|
|
|
|
def recalculate_results(self, request, queryset):
|
|
count = 0
|
|
for participation in queryset:
|
|
participation.recompute_results()
|
|
count += 1
|
|
self.message_user(
|
|
request,
|
|
ungettext(
|
|
"%d participation recalculated.",
|
|
"%d participations recalculated.",
|
|
count,
|
|
)
|
|
% count,
|
|
)
|
|
|
|
recalculate_results.short_description = _("Recalculate results")
|
|
|
|
def username(self, obj):
|
|
return obj.user.username
|
|
|
|
username.short_description = _("username")
|
|
username.admin_order_field = "user__user__username"
|
|
|
|
def show_virtual(self, obj):
|
|
return obj.virtual or "-"
|
|
|
|
show_virtual.short_description = _("virtual")
|
|
show_virtual.admin_order_field = "virtual"
|