2017-05-09 06:41:26 +00:00
|
|
|
from django_redis import get_redis_connection
|
2017-05-08 09:29:01 +00:00
|
|
|
|
|
|
|
from account.decorators import login_required
|
|
|
|
from account.models import AdminType, User
|
2017-07-05 13:09:14 +00:00
|
|
|
from problem.models import Problem, ProblemRuleType
|
2017-05-10 09:20:52 +00:00
|
|
|
from submission.tasks import judge_task
|
2017-05-15 08:42:15 +00:00
|
|
|
# from judge.dispatcher import JudgeDispatcher
|
2017-05-08 09:29:01 +00:00
|
|
|
from utils.api import APIView, validate_serializer
|
|
|
|
from utils.throttling import TokenBucket, BucketController
|
2017-05-15 05:09:54 +00:00
|
|
|
from ..models import Submission, JudgeStatus
|
2017-05-15 08:42:15 +00:00
|
|
|
from ..serializers import CreateSubmissionSerializer, SubmissionModelSerializer
|
2017-07-05 13:09:14 +00:00
|
|
|
from ..serializers import SubmissionSafeSerializer, SubmissionListSerializer
|
|
|
|
|
2017-05-08 09:29:01 +00:00
|
|
|
|
2017-05-15 08:42:15 +00:00
|
|
|
def _submit(response, user, problem_id, language, code, contest_id=None):
|
2017-05-15 05:09:54 +00:00
|
|
|
# TODO: 预设默认值,需修改
|
|
|
|
controller = BucketController(user_id=user.id,
|
|
|
|
redis_conn=get_redis_connection("Throttling"),
|
|
|
|
default_capacity=30)
|
|
|
|
bucket = TokenBucket(fill_rate=10, capacity=20,
|
|
|
|
last_capacity=controller.last_capacity,
|
|
|
|
last_timestamp=controller.last_timestamp)
|
|
|
|
|
|
|
|
if bucket.consume():
|
|
|
|
controller.last_capacity -= 1
|
|
|
|
else:
|
|
|
|
return response.error("Please wait %d seconds" % int(bucket.expected_time() + 1))
|
|
|
|
|
|
|
|
try:
|
2017-07-04 12:59:25 +00:00
|
|
|
problem = Problem.objects.get(_id=problem_id)
|
2017-05-15 05:09:54 +00:00
|
|
|
except Problem.DoesNotExist:
|
|
|
|
return response.error("Problem not exist")
|
|
|
|
|
|
|
|
submission = Submission.objects.create(user_id=user.id,
|
|
|
|
language=language,
|
|
|
|
code=code,
|
2017-07-04 12:59:25 +00:00
|
|
|
problem_id=problem._id,
|
2017-05-15 08:42:15 +00:00
|
|
|
contest_id=contest_id)
|
2017-07-04 09:32:50 +00:00
|
|
|
# todo 暂时保留 方便排错
|
2017-05-15 08:42:15 +00:00
|
|
|
# JudgeDispatcher(submission.id, problem.id).judge()
|
2017-05-15 05:09:54 +00:00
|
|
|
judge_task.delay(submission.id, problem.id)
|
|
|
|
return response.success({"submission_id": submission.id})
|
|
|
|
|
|
|
|
|
2017-05-08 09:29:01 +00:00
|
|
|
class SubmissionAPI(APIView):
|
|
|
|
@validate_serializer(CreateSubmissionSerializer)
|
2017-05-15 08:42:15 +00:00
|
|
|
@login_required
|
2017-05-08 09:29:01 +00:00
|
|
|
def post(self, request):
|
|
|
|
data = request.data
|
2017-05-15 05:09:54 +00:00
|
|
|
return _submit(self, request.user, data["problem_id"], data["language"], data["code"])
|
2017-05-08 09:29:01 +00:00
|
|
|
|
|
|
|
@login_required
|
|
|
|
def get(self, request):
|
2017-05-15 08:42:15 +00:00
|
|
|
submission_id = request.GET.get("id")
|
2017-07-05 13:09:14 +00:00
|
|
|
if not submission_id:
|
|
|
|
return self.error("Parameter id doesn't exist.")
|
|
|
|
try:
|
|
|
|
submission = Submission.objects.get(id=submission_id, user_id=request.user.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.")
|
2017-05-08 09:29:01 +00:00
|
|
|
|
2017-07-05 13:09:14 +00:00
|
|
|
# check problem'rule is ACM or IO.
|
|
|
|
if Problem.objects.filter(_id=submission.problem_id,
|
|
|
|
visible=True,
|
|
|
|
rule_type=ProblemRuleType.ACM
|
|
|
|
).exists():
|
|
|
|
return self.success(SubmissionSafeSerializer(submission).data)
|
|
|
|
return self.success(SubmissionModelSerializer(submission).data)
|
|
|
|
|
|
|
|
|
|
|
|
class SubmissionListAPI(APIView):
|
|
|
|
def get(self, request):
|
2017-07-04 12:59:25 +00:00
|
|
|
contest_id = request.GET.get("contest_id")
|
|
|
|
if contest_id:
|
2017-07-05 13:09:14 +00:00
|
|
|
subs = Submission.objects.filter(contest_id=contest_id)
|
2017-07-04 12:59:25 +00:00
|
|
|
else:
|
|
|
|
subs = Submission.objects.filter(contest_id__isnull=True)
|
|
|
|
|
2017-07-04 10:03:45 +00:00
|
|
|
problem_id = request.GET.get("problem_id")
|
2017-07-04 09:32:50 +00:00
|
|
|
if problem_id:
|
|
|
|
subs = subs.filter(problem_id=problem_id)
|
2017-05-08 09:29:01 +00:00
|
|
|
|
2017-07-06 08:09:38 +00:00
|
|
|
if request.GET.get("myself") and request.GET["myself"] == "1":
|
2017-07-04 09:32:50 +00:00
|
|
|
subs = subs.filter(user_id=request.user.id)
|
2017-07-06 13:08:34 +00:00
|
|
|
data = self.paginate_data(request, subs)
|
|
|
|
data["results"] = SubmissionListSerializer(data["results"], many=True, user=request.user).data
|
|
|
|
return self.success(data)
|
2017-05-15 05:09:54 +00:00
|
|
|
|
|
|
|
|