增加比赛的判题

submission/views.py

@@ -1,21 +1,22 @@
 # coding=utf-8
 import json
-import redis
 
+import redis
 from django.shortcuts import render
+from django.core.paginator import Paginator
 
 from rest_framework.views import APIView
 
-from judge.judger.result import result
+from problem.models import Problem
 from judge.judger_controller.tasks import judge
 from judge.judger_controller.settings import redis_config
 from account.decorators import login_required
 from account.models import SUPER_ADMIN
-from problem.models import Problem
+from contest.models import Contest, ContestProblem
+from contest.decorators import check_user_contest_permission
 from utils.shortcuts import serializer_invalid_response, error_response, success_response, error_page, paginate
 from .models import Submission
-from .serializers import CreateSubmissionSerializer, SubmissionSerializer
-from django.core.paginator import Paginator
+from .serializers import CreateSubmissionSerializer, SubmissionSerializer, CreateContestSubmissionSerializer
 
 
 class SubmissionAPIView(APIView):
@@ -136,3 +137,44 @@ def my_submission_list_page(request, page=1):
     return render(request, "oj/submission/my_submissions_list.html",
                   {"submissions": current_page, "page": int(page),
                    "previous_page": previous_page, "next_page": next_page, "start_id": int(page) * 20 - 20})
+
+
+
+class ContestSubmissionAPIView(APIView):
+    @check_user_contest_permission
+    def post(self, request):
+        """
+        创建比赛的提交
+        ---
+        request_serializer: ConestSubmissionSerializer
+        """
+        serializer = CreateContestSubmissionSerializer(data=request.data)
+        if serializer.is_valid():
+            data = serializer.data
+            try:
+                contest = Contest.objects.get(id=data["contest_id"])
+            except Contest.DoesNotExist:
+                return error_response(u"比赛不存在")
+            try:
+                problem = ContestProblem.objects.get(contest=contest, id=data["problem_id"])
+                # 更新题目提交计数器
+                problem.total_submit_number += 1
+                problem.save()
+            except Problem.DoesNotExist:
+                return error_response(u"题目不存在")
+
+            submission = Submission.objects.create(user_id=request.user.id, language=int(data["language"]),
+                                                   contest_id=contest.id, code=data["code"], problem_id=problem.id)
+            try:
+                judge.delay(submission.id, problem.time_limit, problem.memory_limit, problem.test_case_id)
+            except Exception:
+                return error_response(u"提交判题任务失败")
+
+            # 增加redis 中判题队列长度的计数器
+            r = redis.Redis(host=redis_config["host"], port=redis_config["port"], db=redis_config["db"])
+            r.incr("judge_queue_length")
+
+            return success_response({"submission_id": submission.id})
+
+        else:
+            return serializer_invalid_response(serializer)