87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
from celery import shared_task
|
|
from django.core.cache import cache
|
|
from django.utils.translation import gettext as _
|
|
|
|
from judge.models import Problem, Profile, Submission, SubmissionTestCase
|
|
from judge.utils.celery import Progress
|
|
|
|
__all__ = ("apply_submission_filter", "rejudge_problem_filter", "rescore_problem")
|
|
|
|
|
|
def apply_submission_filter(queryset, id_range, languages, results, contest):
|
|
if id_range:
|
|
start, end = id_range
|
|
queryset = queryset.filter(id__gte=start, id__lte=end)
|
|
if languages:
|
|
queryset = queryset.filter(language_id__in=languages)
|
|
if results:
|
|
queryset = queryset.filter(result__in=results)
|
|
if contest:
|
|
queryset = queryset.filter(contest_object=contest)
|
|
queryset = queryset.exclude(status__in=Submission.IN_PROGRESS_GRADING_STATUS)
|
|
return queryset
|
|
|
|
|
|
@shared_task(bind=True)
|
|
def rejudge_problem_filter(
|
|
self, problem_id, id_range=None, languages=None, results=None, contest=None
|
|
):
|
|
queryset = Submission.objects.filter(problem_id=problem_id)
|
|
queryset = apply_submission_filter(queryset, id_range, languages, results, contest)
|
|
|
|
rejudged = 0
|
|
with Progress(self, queryset.count() * 2) as p:
|
|
for submission in queryset:
|
|
SubmissionTestCase.objects.filter(submission_id=submission.id).delete()
|
|
rejudged += 1
|
|
if rejudged % 10 == 0:
|
|
p.done = rejudged
|
|
|
|
for submission in queryset.iterator():
|
|
submission.judge(rejudge=True, batch_rejudge=True, delete_testcases=False)
|
|
rejudged += 1
|
|
if rejudged % 10 == 0:
|
|
p.done = rejudged
|
|
return rejudged
|
|
|
|
|
|
@shared_task(bind=True)
|
|
def rescore_problem(self, problem_id):
|
|
problem = Problem.objects.get(id=problem_id)
|
|
submissions = Submission.objects.filter(problem_id=problem_id)
|
|
|
|
with Progress(self, submissions.count(), stage=_("Modifying submissions")) as p:
|
|
rescored = 0
|
|
for submission in submissions.iterator():
|
|
submission.points = round(
|
|
submission.case_points / submission.case_total * problem.points
|
|
if submission.case_total
|
|
else 0,
|
|
1,
|
|
)
|
|
if not problem.partial and submission.points < problem.points:
|
|
submission.points = 0
|
|
submission.save(update_fields=["points"])
|
|
submission.update_contest()
|
|
rescored += 1
|
|
if rescored % 10 == 0:
|
|
p.done = rescored
|
|
|
|
with Progress(
|
|
self,
|
|
submissions.values("user_id").distinct().count(),
|
|
stage=_("Recalculating user points"),
|
|
) as p:
|
|
users = 0
|
|
profiles = Profile.objects.filter(
|
|
id__in=submissions.values_list("user_id", flat=True).distinct()
|
|
)
|
|
for profile in profiles.iterator():
|
|
profile._updating_stats_only = True
|
|
profile.calculate_points()
|
|
cache.delete("user_complete:%d" % profile.id)
|
|
cache.delete("user_attempted:%d" % profile.id)
|
|
users += 1
|
|
if users % 10 == 0:
|
|
p.done = users
|
|
return rescored
|