xuyue/OnlineJudge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

新增AI分析

Browse Source
This commit is contained in:
yuetsh
2025-09-24 01:02:29 +08:00
parent aba908c0aa
commit 6e1e3ef0c6
16 changed files with 790 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

159
submission/views/admin.py
View File

@@ -1,6 +1,6 @@
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
@@ -8,6 +8,12 @@ from problem.models import Problem
from django.db.models import Count, Q
def get_real_name(username, class_name):
if class_name and username.startswith("ks"):
return username[len(f"ks{class_name}"):]
return username
class SubmissionRejudgeAPI(APIView):
@super_admin_required
def get(self, request):
@@ -15,7 +21,9 @@ class SubmissionRejudgeAPI(APIView):
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)
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 = {}
@@ -23,82 +31,123 @@ class SubmissionRejudgeAPI(APIView):
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")
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)
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_persons = []
all_users_dict = {}
if username:
submissions = submissions.filter(username__icontains=username)
all_persons = User.objects.filter(username__icontains=username,
is_disabled=False,
admin_type=AdminType.REGULAR_USER).values_list("username", flat=True)
person_count = all_persons.count()
all_users_dict = {
user["username"]: user["class_name"]
for user in User.objects.filter(
username__icontains=username,
is_disabled=False,
admin_type=AdminType.REGULAR_USER,
).values("username", "class_name")
}
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
# 优化：一次性获取所有统计数据
submission_stats = submissions.aggregate(
total_count=Count("id"),
accepted_count=Count("id", filter=Q(result=JudgeStatus.ACCEPTED)),
)
submission_count = submission_stats["total_count"]
accepted_count = submission_stats["accepted_count"]
correct_rate = (
round(accepted_count / submission_count * 100, 2) if submission_count else 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")
# 优化：获取用户提交统计
user_submissions = (
submissions.values("username")
.annotate(
submission_count=Count("id"),
accepted_count=Count("id", filter=Q(result=JudgeStatus.ACCEPTED)),
)
.order_by("-submission_count")
)
# 获取所有有提交记录的用户的class_name信息
submitted_usernames = {item["username"] for item in user_submissions}
if submitted_usernames:
submitted_users_dict = {
user["username"]: user["class_name"]
for user in User.objects.filter(
username__in=submitted_usernames
).values("username", "class_name")
}
else:
submitted_users_dict = {}
# 处理有提交记录的用户
accepted = []
for item in counts:
for item in user_submissions:
username_key = item["username"]
if item["accepted_count"] > 0:
rate = round(item["accepted_count"]/item["submission_count"]*100, 2)
item["correct_rate"] = f"{rate}%"
accepted.append(item)
rate = round(item["accepted_count"] / item["submission_count"] * 100, 2)
accepted.append(
{
"username": username_key,
"class_name": submitted_users_dict.get(username_key),
"submission_count": item["submission_count"],
"accepted_count": item["accepted_count"],
"correct_rate": f"{rate}%",
}
)
# 处理无提交记录的用户，只返回姓名列表
unaccepted = []
if len(accepted) > 0:
unaccepted = list(set(all_persons) - set([item['username'] for item in accepted]))
# 统计人数完成率
if all_users_dict:
unaccepted_usernames = set(all_users_dict.keys()) - submitted_usernames
for username in unaccepted_usernames:
class_name = all_users_dict[username]
real_name = get_real_name(username, class_name)
unaccepted.append(real_name)
# 计算人数完成率
person_count = len(all_users_dict) if all_users_dict else 0
person_rate = 0
if person_count:
person_rate = round(len(accepted)/person_count*100, 2)
# 下面是做一些超出 100% 的操作，比如有人已经删号了，但是提交记录还在
if person_rate >= 100:
person_rate = 100
# 搜出来的人数比提交人数还多的情况
person_rate = min(100, round(len(accepted) / person_count * 100, 2))
# 处理已删除用户但提交记录仍存在的情况
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
})
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,
}
)