NDOJ/judge/admin/submission.py

from functools import partial
from operator import itemgetter

from django.conf import settings
from django.conf.urls import url
from django.contrib import admin, messages
from django.core.cache import cache
from django.core.exceptions import PermissionDenied
from django.db.models import Q
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404
from django.utils.html import format_html
from django.utils.translation import gettext, gettext_lazy as _, pgettext, ungettext

from django_ace import AceWidget
from judge.models import (
    ContestParticipation,
    ContestProblem,
    ContestSubmission,
    Profile,
    Submission,
    SubmissionSource,
    SubmissionTestCase,
)
from judge.utils.raw_sql import use_straight_join


class SubmissionStatusFilter(admin.SimpleListFilter):
    parameter_name = title = "status"
    __lookups = (
        ("None", _("None")),
        ("NotDone", _("Not done")),
        ("EX", _("Exceptional")),
    ) + Submission.STATUS
    __handles = set(map(itemgetter(0), Submission.STATUS))

    def lookups(self, request, model_admin):
        return self.__lookups

    def queryset(self, request, queryset):
        if self.value() == "None":
            return queryset.filter(status=None)
        elif self.value() == "NotDone":
            return queryset.exclude(status__in=["D", "IE", "CE", "AB"])
        elif self.value() == "EX":
            return queryset.exclude(status__in=["D", "CE", "G", "AB"])
        elif self.value() in self.__handles:
            return queryset.filter(status=self.value())


class SubmissionResultFilter(admin.SimpleListFilter):
    parameter_name = title = "result"
    __lookups = (("None", _("None")), ("BAD", _("Unaccepted"))) + Submission.RESULT
    __handles = set(map(itemgetter(0), Submission.RESULT))

    def lookups(self, request, model_admin):
        return self.__lookups

    def queryset(self, request, queryset):
        if self.value() == "None":
            return queryset.filter(result=None)
        elif self.value() == "BAD":
            return queryset.exclude(result="AC")
        elif self.value() in self.__handles:
            return queryset.filter(result=self.value())


class SubmissionTestCaseInline(admin.TabularInline):
    fields = ("case", "batch", "status", "time", "memory", "points", "total")
    readonly_fields = ("case", "batch", "total")
    model = SubmissionTestCase
    can_delete = False
    max_num = 0


class ContestSubmissionInline(admin.StackedInline):
    fields = ("problem", "participation", "points")
    model = ContestSubmission

    def get_formset(self, request, obj=None, **kwargs):
        kwargs["formfield_callback"] = partial(
            self.formfield_for_dbfield, request=request, obj=obj
        )
        return super(ContestSubmissionInline, self).get_formset(request, obj, **kwargs)

    def formfield_for_dbfield(self, db_field, **kwargs):
        submission = kwargs.pop("obj", None)
        label = None
        if submission:
            if db_field.name == "participation":
                kwargs["queryset"] = ContestParticipation.objects.filter(
                    user=submission.user, contest__problems=submission.problem
                ).only("id", "contest__name")

                def label(obj):
                    return obj.contest.name

            elif db_field.name == "problem":
                kwargs["queryset"] = ContestProblem.objects.filter(
                    problem=submission.problem
                ).only("id", "problem__name", "contest__name")

                def label(obj):
                    return pgettext("contest problem", "%(problem)s in %(contest)s") % {
                        "problem": obj.problem.name,
                        "contest": obj.contest.name,
                    }

        field = super(ContestSubmissionInline, self).formfield_for_dbfield(
            db_field, **kwargs
        )
        if label is not None:
            field.label_from_instance = label
        return field


class SubmissionSourceInline(admin.StackedInline):
    fields = ("source",)
    model = SubmissionSource
    can_delete = False
    extra = 0

    def get_formset(self, request, obj=None, **kwargs):
        kwargs.setdefault("widgets", {})["source"] = AceWidget(
            mode=obj and obj.language.ace, theme=request.profile.ace_theme
        )
        return super().get_formset(request, obj, **kwargs)


class SubmissionAdmin(admin.ModelAdmin):
    readonly_fields = ("user", "problem", "date", "judged_date")
    fields = (
        "user",
        "problem",
        "date",
        "judged_date",
        "time",
        "memory",
        "points",
        "language",
        "status",
        "result",
        "case_points",
        "case_total",
        "judged_on",
        "error",
    )
    actions = ("judge", "recalculate_score")
    list_display = (
        "id",
        "problem_code",
        "problem_name",
        "user_column",
        "execution_time",
        "pretty_memory",
        "points",
        "language_column",
        "status",
        "result",
        "judge_column",
    )
    list_filter = ("language", SubmissionStatusFilter, SubmissionResultFilter)
    search_fields = ("problem__code", "problem__name", "user__user__username")
    actions_on_top = True
    actions_on_bottom = True
    inlines = [
        SubmissionSourceInline,
        SubmissionTestCaseInline,
        ContestSubmissionInline,
    ]

    def get_queryset(self, request):
        queryset = Submission.objects.select_related(
            "problem", "user__user", "language"
        ).only(
            "problem__code",
            "problem__name",
            "user__user__username",
            "language__name",
            "time",
            "memory",
            "points",
            "status",
            "result",
        )
        use_straight_join(queryset)
        if not request.user.has_perm("judge.edit_all_problem"):
            id = request.profile.id
            queryset = queryset.filter(
                Q(problem__authors__id=id) | Q(problem__curators__id=id)
            ).distinct()
        return queryset

    def has_add_permission(self, request):
        return False

    def lookup_allowed(self, key, value):
        return super(SubmissionAdmin, self).lookup_allowed(key, value) or key in (
            "problem__code",
        )

    def save_model(self, request, obj, form, change):
        super().save_model(request, obj, form, change)
        if "case_points" in form.changed_data or "case_total" in form.changed_data:
            obj.update_contest()

    def judge(self, request, queryset):
        if not request.user.has_perm(
            "judge.rejudge_submission"
        ) or not request.user.has_perm("judge.edit_own_problem"):
            self.message_user(
                request,
                gettext("You do not have the permission to rejudge submissions."),
                level=messages.ERROR,
            )
            return
        queryset = queryset.order_by("id")
        if (
            not request.user.has_perm("judge.rejudge_submission_lot")
            and queryset.count() > settings.DMOJ_SUBMISSIONS_REJUDGE_LIMIT
        ):
            self.message_user(
                request,
                gettext(
                    "You do not have the permission to rejudge THAT many submissions."
                ),
                level=messages.ERROR,
            )
            return
        if not request.user.has_perm("judge.edit_all_problem"):
            id = request.profile.id
            queryset = queryset.filter(
                Q(problem__authors__id=id) | Q(problem__curators__id=id)
            )
        judged = len(queryset)
        for model in queryset:
            model.judge(rejudge=True, batch_rejudge=True)
        self.message_user(
            request,
            ungettext(
                "%d submission was successfully scheduled for rejudging.",
                "%d submissions were successfully scheduled for rejudging.",
                judged,
            )
            % judged,
        )

    judge.short_description = _("Rejudge the selected submissions")

    def recalculate_score(self, request, queryset):
        if not request.user.has_perm("judge.rejudge_submission"):
            self.message_user(
                request,
                gettext("You do not have the permission to rejudge submissions."),
                level=messages.ERROR,
            )
            return
        submissions = list(
            queryset.defer(None)
            .select_related(None)
            .select_related("problem")
            .only(
                "points",
                "case_points",
                "case_total",
                "problem__partial",
                "problem__points",
            )
        )
        for submission in submissions:
            submission.points = round(
                submission.case_points
                / submission.case_total
                * submission.problem.points
                if submission.case_total
                else 0,
                1,
            )
            if (
                not submission.problem.partial
                and submission.points < submission.problem.points
            ):
                submission.points = 0
            submission.save()
            submission.update_contest()

        for profile in Profile.objects.filter(
            id__in=queryset.values_list("user_id", flat=True).distinct()
        ):
            profile.calculate_points()
            cache.delete("user_complete:%d" % profile.id)
            cache.delete("user_attempted:%d" % profile.id)

        for participation in ContestParticipation.objects.filter(
            id__in=queryset.values_list("contest__participation_id")
        ).prefetch_related("contest"):
            participation.recompute_results()

        self.message_user(
            request,
            ungettext(
                "%d submission were successfully rescored.",
                "%d submissions were successfully rescored.",
                len(submissions),
            )
            % len(submissions),
        )

    recalculate_score.short_description = _("Rescore the selected submissions")

    def problem_code(self, obj):
        return obj.problem.code

    problem_code.short_description = _("Problem code")
    problem_code.admin_order_field = "problem__code"

    def problem_name(self, obj):
        return obj.problem.name

    problem_name.short_description = _("Problem name")
    problem_name.admin_order_field = "problem__name"

    def user_column(self, obj):
        return obj.user.user.username

    user_column.admin_order_field = "user__user__username"
    user_column.short_description = _("User")

    def execution_time(self, obj):
        return round(obj.time, 2) if obj.time is not None else "None"

    execution_time.short_description = _("Time")
    execution_time.admin_order_field = "time"

    def pretty_memory(self, obj):
        memory = obj.memory
        if memory is None:
            return gettext("None")
        if memory < 1000:
            return gettext("%d KB") % memory
        else:
            return gettext("%.2f MB") % (memory / 1024)

    pretty_memory.admin_order_field = "memory"
    pretty_memory.short_description = _("Memory")

    def language_column(self, obj):
        return obj.language.name

    language_column.admin_order_field = "language__name"
    language_column.short_description = _("Language")

    def judge_column(self, obj):
        return format_html(
            '<input type="button" value="Rejudge" onclick="location.href=\'{}/judge/\'" />',
            obj.id,
        )

    judge_column.short_description = ""

    def get_urls(self):
        return [
            url(r"^(\d+)/judge/$", self.judge_view, name="judge_submission_rejudge"),
        ] + super(SubmissionAdmin, self).get_urls()

    def judge_view(self, request, id):
        if not request.user.has_perm(
            "judge.rejudge_submission"
        ) or not request.user.has_perm("judge.edit_own_problem"):
            raise PermissionDenied()
        submission = get_object_or_404(Submission, id=id)
        if not request.user.has_perm(
            "judge.edit_all_problem"
        ) and not submission.problem.is_editor(request.profile):
            raise PermissionDenied()
        submission.judge(rejudge=True)
        return HttpResponseRedirect(request.META.get("HTTP_REFERER", "/"))