104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
from account.decorators import super_admin_required
|
|
from judge.tasks import judge_task
|
|
# from judge.dispatcher import JudgeDispatcher
|
|
from utils.api import APIView
|
|
from ..models import Submission, JudgeStatus
|
|
from account.models import User, AdminType
|
|
from problem.models import Problem
|
|
from django.db.models import Count, Q
|
|
|
|
|
|
class SubmissionRejudgeAPI(APIView):
|
|
@super_admin_required
|
|
def get(self, request):
|
|
id = request.GET.get("id")
|
|
if not id:
|
|
return self.error("Parameter error, id is required")
|
|
try:
|
|
submission = Submission.objects.select_related("problem").get(id=id, contest_id__isnull=True)
|
|
except Submission.DoesNotExist:
|
|
return self.error("Submission does not exists")
|
|
submission.statistic_info = {}
|
|
submission.save()
|
|
|
|
judge_task.send(submission.id, submission.problem.id)
|
|
return self.success()
|
|
|
|
class SubmissionStatisticsAPI(APIView):
|
|
@super_admin_required
|
|
def get(self, request):
|
|
start = request.GET.get("start")
|
|
end = request.GET.get("end")
|
|
|
|
if not start or not end:
|
|
return self.error("start and end is required")
|
|
|
|
submissions = Submission.objects.filter(contest_id__isnull=True,
|
|
create_time__gte=start,
|
|
create_time__lte=end).select_related("problem__created_by")
|
|
|
|
problem_id = request.GET.get("problem_id")
|
|
|
|
if problem_id:
|
|
try:
|
|
problem = Problem.objects.get(_id=problem_id, contest_id__isnull=True, visible=True)
|
|
except Problem.DoesNotExist:
|
|
return self.error("Problem doesn't exist")
|
|
submissions = submissions.filter(problem=problem)
|
|
|
|
username = request.GET.get("username")
|
|
|
|
# 统计人数
|
|
person_count = 0
|
|
all_persions = []
|
|
if username:
|
|
submissions = submissions.filter(username__icontains=username)
|
|
all_persions = User.objects.filter(username__icontains=username,
|
|
is_disabled=False,
|
|
admin_type=AdminType.REGULAR_USER).values_list("username", flat=True)
|
|
person_count = all_persions.count()
|
|
|
|
submission_count = submissions.count()
|
|
accepted_count = submissions.filter(result=JudgeStatus.ACCEPTED).count()
|
|
|
|
try:
|
|
correct_rate = round(accepted_count/submission_count*100, 2)
|
|
except ZeroDivisionError:
|
|
correct_rate = 0
|
|
|
|
counts = submissions.values("username").annotate(submission_count=Count("id", distinct=True),
|
|
accepted_count=Count("id", filter=Q(result=JudgeStatus.ACCEPTED), distinct=True),
|
|
).order_by("-submission_count")
|
|
|
|
accepted = []
|
|
for item in counts:
|
|
if item["accepted_count"] > 0:
|
|
rate = round(item["accepted_count"]/item["submission_count"]*100, 2)
|
|
item["correct_rate"] = f"{rate}%"
|
|
accepted.append(item)
|
|
|
|
unaccepted = []
|
|
if person_count:
|
|
unaccepted = list(set(all_persions) - set([item['username'] for item in accepted]))
|
|
|
|
# 统计人数完成率
|
|
person_rate = 0
|
|
if person_count:
|
|
person_rate = round(len(accepted)/person_count*100, 2)
|
|
# 下面是做一些超出 100% 的操作，比如有人已经删号了，但是提交记录还在
|
|
if person_rate >= 100:
|
|
person_rate = 100
|
|
# 搜出来的人数比提交人数还多的情况
|
|
if person_count < len(accepted):
|
|
person_count = len(accepted)
|
|
|
|
return self.success({
|
|
"submission_count": submission_count,
|
|
"accepted_count": accepted_count,
|
|
"correct_rate": f"{correct_rate}%",
|
|
"person_count": person_count,
|
|
"person_rate": f"{person_rate}%",
|
|
"data": accepted,
|
|
"data_unaccepted": unaccepted
|
|
})
|
|
|