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

在前台添加今日提交数

Browse Source
This commit is contained in:
yuetsh
2025-04-14 12:32:49 +08:00
parent e5939d3592
commit 025276002d
2 changed files with 89 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
submission/urls/oj.py
View File

@@ -1,6 +1,11 @@
from django.urls import path
from ..views.oj import SubmissionAPI, SubmissionListAPI, ContestSubmissionListAPI, SubmissionExistsAPI
from ..views.oj import (
SubmissionAPI,
SubmissionListAPI,
ContestSubmissionListAPI,
SubmissionExistsAPI,
)
urlpatterns = [
path("submission", SubmissionAPI.as_view()),

100
submission/views/oj.py
View File

@@ -1,9 +1,11 @@
from datetime import datetime
import ipaddress
from account.decorators import login_required, check_contest_permission
from contest.models import ContestStatus, ContestRuleType
from judge.tasks import judge_task
from options.options import SysOptions
# from judge.dispatcher import JudgeDispatcher
from problem.models import Problem, ProblemRuleType
from utils.api import APIView, validate_serializer
@@ -11,8 +13,11 @@ from utils.cache import cache
from utils.captcha import Captcha
from utils.throttling import TokenBucket
from ..models import Submission
from ..serializers import (CreateSubmissionSerializer, SubmissionModelSerializer,
ShareSubmissionSerializer)
from ..serializers import (
CreateSubmissionSerializer,
SubmissionModelSerializer,
ShareSubmissionSerializer,
)
from ..serializers import SubmissionSafeModelSerializer, SubmissionListSerializer
@@ -22,8 +27,9 @@ class SubmissionAPI(APIView):
auth_method = getattr(request, "auth_method", "")
if auth_method == "api_key":
return
user_bucket = TokenBucket(key=str(request.user.id),
redis_conn=cache, **SysOptions.throttling["user"])
user_bucket = TokenBucket(
key=str(request.user.id), redis_conn=cache, **SysOptions.throttling["user"]
)
can_consume, wait = user_bucket.consume()
if not can_consume:
return "Please wait %d seconds" % (int(wait))
@@ -42,7 +48,10 @@ class SubmissionAPI(APIView):
if not request.user.is_contest_admin(contest):
user_ip = ipaddress.ip_address(request.session.get("ip"))
if contest.allowed_ip_ranges:
if not any(user_ip in ipaddress.ip_network(cidr, strict=False) for cidr in contest.allowed_ip_ranges):
if not any(
user_ip in ipaddress.ip_network(cidr, strict=False)
for cidr in contest.allowed_ip_ranges
):
return self.error("Your IP is not allowed in this contest")
@validate_serializer(CreateSubmissionSerializer)
@@ -66,18 +75,22 @@ class SubmissionAPI(APIView):
return self.error(error)
try:
problem = Problem.objects.get(id=data["problem_id"], contest_id=data.get("contest_id"), visible=True)
problem = Problem.objects.get(
id=data["problem_id"], contest_id=data.get("contest_id"), visible=True
)
except Problem.DoesNotExist:
return self.error("Problem not exist")
if data["language"] not in problem.languages:
return self.error(f"{data['language']} is not allowed in the problem")
submission = Submission.objects.create(user_id=request.user.id,
submission = Submission.objects.create(
user_id=request.user.id,
username=request.user.username,
language=data["language"],
code=data["code"],
problem_id=problem.id,
ip=request.session["ip"],
contest_id=data.get("contest_id"))
contest_id=data.get("contest_id"),
)
# use this for debug
# JudgeDispatcher(submission.id, problem.id).judge()
judge_task.send(submission.id, problem.id)
@@ -92,18 +105,25 @@ class SubmissionAPI(APIView):
if not submission_id:
return self.error("Parameter id doesn't exist")
try:
submission = Submission.objects.select_related("problem").get(id=submission_id)
submission = Submission.objects.select_related("problem").get(
id=submission_id
)
except Submission.DoesNotExist:
return self.error("Submission doesn't exist")
if not submission.check_user_permission(request.user):
return self.error("No permission for this submission")
if submission.problem.rule_type == ProblemRuleType.OI or request.user.is_admin_role():
if (
submission.problem.rule_type == ProblemRuleType.OI
or request.user.is_admin_role()
):
submission_data = SubmissionModelSerializer(submission).data
else:
submission_data = SubmissionSafeModelSerializer(submission).data
# 是否有权限取消共享
submission_data["can_unshare"] = submission.check_user_permission(request.user, check_share=False)
submission_data["can_unshare"] = submission.check_user_permission(
request.user, check_share=False
)
return self.success(submission_data)
@validate_serializer(ShareSubmissionSerializer)
@@ -113,12 +133,17 @@ class SubmissionAPI(APIView):
share submission
"""
try:
submission = Submission.objects.select_related("problem").get(id=request.data["id"])
submission = Submission.objects.select_related("problem").get(
id=request.data["id"]
)
except Submission.DoesNotExist:
return self.error("Submission doesn't exist")
if not submission.check_user_permission(request.user, check_share=False):
return self.error("No permission to share the submission")
if submission.contest and submission.contest.status == ContestStatus.CONTEST_UNDERWAY:
if (
submission.contest
and submission.contest.status == ContestStatus.CONTEST_UNDERWAY
):
return self.error("Can not share submission now")
submission.shared = request.data["shared"]
submission.save(update_fields=["shared"])
@@ -132,19 +157,26 @@ class SubmissionListAPI(APIView):
if request.GET.get("contest_id"):
return self.error("Parameter error")
submissions = Submission.objects.filter(contest_id__isnull=True).select_related("problem__created_by")
submissions = Submission.objects.filter(contest_id__isnull=True).select_related(
"problem__created_by"
)
problem_id = request.GET.get("problem_id")
myself = request.GET.get("myself")
result = request.GET.get("result")
username = request.GET.get("username")
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)
if not SysOptions.submission_list_show_all and not request.user.is_super_admin():
if (
not SysOptions.submission_list_show_all
and not request.user.is_super_admin()
):
return self.success({"results": [], "total": 0})
if myself and myself == "1":
@@ -155,7 +187,16 @@ class SubmissionListAPI(APIView):
submissions = submissions.filter(result=result)
data = self.paginate_data(request, submissions)
data["results"] = SubmissionListSerializer(data["results"], many=True, user=request.user).data
data["results"] = SubmissionListSerializer(
data["results"], many=True, user=request.user
).data
today = datetime.today()
today_submissions_count = Submission.objects.filter(
create_time__gte=datetime(today.year, today.month, today.day, 0, 0)
).count()
data["today_count"] = today_submissions_count
return self.success(data)
@@ -166,14 +207,18 @@ class ContestSubmissionListAPI(APIView):
return self.error("Limit is needed")
contest = self.contest
submissions = Submission.objects.filter(contest_id=contest.id).select_related("problem__created_by")
submissions = Submission.objects.filter(contest_id=contest.id).select_related(
"problem__created_by"
)
problem_id = request.GET.get("problem_id")
myself = request.GET.get("myself")
result = request.GET.get("result")
username = request.GET.get("username")
if problem_id:
try:
problem = Problem.objects.get(_id=problem_id, contest_id=contest.id, visible=True)
problem = Problem.objects.get(
_id=problem_id, contest_id=contest.id, visible=True
)
except Problem.DoesNotExist:
return self.error("Problem doesn't exist")
submissions = submissions.filter(problem=problem)
@@ -191,11 +236,15 @@ class ContestSubmissionListAPI(APIView):
# 封榜的时候只能看到自己的提交
if contest.rule_type == ContestRuleType.ACM:
if not contest.real_time_rank and not request.user.is_contest_admin(contest):
if not contest.real_time_rank and not request.user.is_contest_admin(
contest
):
submissions = submissions.filter(user_id=request.user.id)
data = self.paginate_data(request, submissions)
data["results"] = SubmissionListSerializer(data["results"], many=True, user=request.user).data
data["results"] = SubmissionListSerializer(
data["results"], many=True, user=request.user
).data
return self.success(data)
@@ -203,6 +252,9 @@ class SubmissionExistsAPI(APIView):
def get(self, request):
if not request.GET.get("problem_id"):
return self.error("Parameter error, problem_id is required")
return self.success(request.user.is_authenticated and
Submission.objects.filter(problem_id=request.GET["problem_id"],
user_id=request.user.id).exists())
return self.success(
request.user.is_authenticated
and Submission.objects.filter(
problem_id=request.GET["problem_id"], user_id=request.user.id
).exists()
)