Reformat using black
This commit is contained in:
cuom1999
parent
efee4ad081
commit
a87fb49918
221 changed files with 19127 additions and 7310 deletions
|
|
@ -13,239 +13,367 @@ 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.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
|
||||
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':
|
||||
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() == "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
|
||||
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':
|
||||
if self.value() == "None":
|
||||
return queryset.filter(result=None)
|
||||
elif self.value() == 'BAD':
|
||||
return queryset.exclude(result='AC')
|
||||
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')
|
||||
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')
|
||||
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)
|
||||
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)
|
||||
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')
|
||||
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')
|
||||
|
||||
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,
|
||||
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)
|
||||
|
||||
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',)
|
||||
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)
|
||||
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')
|
||||
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]
|
||||
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',
|
||||
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'):
|
||||
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()
|
||||
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 has_change_permission(self, request, obj=None):
|
||||
if not request.user.has_perm('judge.edit_own_problem'):
|
||||
if not request.user.has_perm("judge.edit_own_problem"):
|
||||
return False
|
||||
if request.user.has_perm('judge.edit_all_problem') or obj is None:
|
||||
if request.user.has_perm("judge.edit_all_problem") or obj is None:
|
||||
return True
|
||||
return obj.problem.is_editor(request.profile)
|
||||
|
||||
def lookup_allowed(self, key, value):
|
||||
return super(SubmissionAdmin, self).lookup_allowed(key, value) or key in ('problem__code',)
|
||||
return super(SubmissionAdmin, self).lookup_allowed(key, value) or key in (
|
||||
"problem__code",
|
||||
)
|
||||
|
||||
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)
|
||||
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)
|
||||
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'):
|
||||
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))
|
||||
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')
|
||||
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)
|
||||
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'))
|
||||
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 = 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()):
|
||||
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)
|
||||
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'):
|
||||
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')
|
||||
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'
|
||||
|
||||
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'
|
||||
|
||||
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')
|
||||
|
||||
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'
|
||||
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')
|
||||
return gettext("None")
|
||||
if memory < 1000:
|
||||
return gettext('%d KB') % memory
|
||||
return gettext("%d KB") % memory
|
||||
else:
|
||||
return gettext('%.2f MB') % (memory / 1024)
|
||||
pretty_memory.admin_order_field = 'memory'
|
||||
pretty_memory.short_description = _('Memory')
|
||||
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')
|
||||
|
||||
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 = ''
|
||||
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'),
|
||||
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'):
|
||||
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):
|
||||
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', '/'))
|
||||
return HttpResponseRedirect(request.META.get("HTTP_REFERER", "/"))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue