后端比赛和比赛题目的部分，还有要增加和修改的地方

contest/serializers.py

@@ -0,0 +1,95 @@
+# coding=utf-8
+import json
+from rest_framework import serializers
+
+from account.models import User
+from .models import Contest, ContestProblem
+
+
+class CreateContestSerializer(serializers.Serializer):
+    title = serializers.CharField(max_length=40)
+    description = serializers.CharField(max_length=5000)
+    mode = serializers.IntegerField()
+    show_rank = serializers.BooleanField()
+    password = serializers.CharField(max_length=30, required=False)
+    start_time = serializers.DateTimeField()
+    end_time = serializers.DateTimeField()
+
+
+class ContestSerializer(serializers.ModelSerializer):
+    class UserSerializer(serializers.ModelSerializer):
+        class Meta:
+            model = User
+            fields = ["username"]
+
+    created_by = UserSerializer()
+
+    class Meta:
+        model = Contest
+
+
+class EditContestSerializer(serializers.Serializer):
+    title = serializers.CharField(max_length=40)
+    description = serializers.CharField(max_length=10000)
+    mode = serializers.IntegerField()
+    show_rank = serializers.BooleanField()
+    password = serializers.CharField(max_length=30, required=False)
+    start_time = serializers.DateTimeField()
+    end_time = serializers.DateTimeField()
+
+
+class ContestProblemSampleSerializer(serializers.ListField):
+    input = serializers.CharField(max_length=3000)
+    output = serializers.CharField(max_length=3000)
+
+
+class JSONField(serializers.Field):
+    def to_representation(self, value):
+        return json.loads(value)
+
+
+class CreateContestProblemSerializer(serializers.Serializer):
+    title = serializers.CharField(max_length=50)
+    description = serializers.CharField(max_length=10000)
+    input_description = serializers.CharField(max_length=10000)
+    output_description = serializers.CharField(max_length=10000)
+    # [{"input": "1 1", "output": "2"}]
+    samples = ContestProblemSampleSerializer()
+    test_case_id = serializers.CharField(max_length=40)
+    time_limit = serializers.IntegerField()
+    memory_limit = serializers.IntegerField()
+    difficulty = serializers.IntegerField()
+    hint = serializers.CharField(max_length=3000, allow_blank=True)
+    sort_index = serializers.CharField(max_length=30)
+
+
+class ContestProblemSerializer(serializers.ModelSerializer):
+    samples = JSONField()
+
+    class ContestSerializer(serializers.ModelSerializer):
+        class Meta:
+            model = Contest
+            fields = ["title"]
+
+    created_by = ContestSerializer()
+
+    class Meta:
+        model = ContestProblem
+
+
+class EditContestProblemSerializer(serializers.Serializer):
+    id = serializers.IntegerField()
+    title = serializers.CharField(max_length=50)
+    description = serializers.CharField(max_length=10000)
+    input_description = serializers.CharField(max_length=10000)
+    output_description = serializers.CharField(max_length=10000)
+    test_case_id = serializers.CharField(max_length=40)
+    time_limit = serializers.IntegerField()
+    memory_limit = serializers.IntegerField()
+    difficulty = serializers.IntegerField()
+    samples = ContestProblemSampleSerializer()
+    hint = serializers.CharField(max_length=3000, allow_blank=True)
+    visible = serializers.BooleanField()
+    sort_index = serializers.CharField(max_length=30)
+
+

contest/views.py

@@ -1,3 +1,170 @@
+# coding=utf-8
+import json
 from django.shortcuts import render
+from django.db import IntegrityError
+from django.db.models import Q
+from rest_framework.views import APIView
 
-# Create your views here.
+from .models import Contest, ContestProblem
+from .serializers import (CreateContestSerializer, ContestSerializer, EditContestSerializer,
+                          CreateContestProblemSerializer, ContestProblemSerializer, EditContestProblemSerializer)
+from utils.shortcuts import (serializer_invalid_response, error_response,
+                             success_response, paginate, rand_str, error_page)
+
+
+def contest_page(request, contest_id):
+    pass
+
+
+class ContestAdminAPIView(APIView):
+    def post(self, request):
+        """
+        比赛发布json api接口
+        ---
+        request_serializer: CreateContestSerializer
+        response_serializer: ContestSerializer
+        """
+        serializer = CreateContestSerializer(data=request.data)
+        if serializer.is_valid():
+            data = serializer.data
+            try:
+                contest = Contest.objects.create(title=data["title"], description=data["description"],
+                                                 mode=data["mode"], show_rank=data["show_rank"],
+                                                 start_time=["start_time"], end_time=data["end_time"])
+                if data["password"]:
+                    contest.setpassword(data["password"])
+                contest.save()
+            except IntegrityError:
+                return error_response(u"比赛名已经存在")
+            return success_response(ContestSerializer(contest).data)
+        else:
+            serializer_invalid_response(serializer)
+
+    def put(self, request):
+        """
+        比赛编辑json api接口
+        ---
+        request_serializer: EditContestSerializer
+        response_serializer: ContestSerializer
+        """
+        serializer = EditContestSerializer(data=request.data)
+        if serializer.is_valid():
+            data = serializer.data
+            try:
+                contest = Contest.objects.get(id=data["id"])
+            except Contest.DoesNotExist:
+                error_response(u"该比赛不存在！")
+            try:
+                contest.title = data["contest"]
+                contest.description = data["description"]
+                contest.mode = data["mode"]
+                contest.show_rank = data["show_"]
+                if data["password"]:
+                    contest.set_password(data["password"])
+                contest.save()
+            except IntegrityError:
+                return error_response(u"比赛名已经存在")
+
+            return success_response(ContestSerializer(contest).data)
+        else:
+            serializer_invalid_response(serializer)
+
+    def get(self, request):
+        """
+        比赛分页json api接口
+        ---
+        response_serializer: ContestSerializer
+        """
+        contest = Contest.objects.all().order_by("-last_updated_time")
+        visible = request.GET.get("visible", None)
+        if visible:
+            contest = contest.filter(visible=(visible == "true"))
+        keyword = request.GET.get("keyword", None)
+        if keyword:
+            contest = contest.filter(Q(title__contains=keyword) |
+                                     Q(description__contains=keyword))
+        return paginate(request, contest, ContestSerializer)
+
+
+class ContestProblemAdminAPIView(APIView):
+    def post(self, request):
+        """
+        比赛题目发布json api接口
+        ---
+        request_serializer: CreateContestProblemSerializer
+        response_serializer: ContestProblemSerializer
+        """
+        serializer = CreateContestProblemSerializer(data=request.data)
+        if serializer.is_valid():
+            data = serializer.data
+            contest_problem = ContestProblem.objects.create(title=data["title"],
+                                                            description=data["description"],
+                                                            input_description=data["input_description"],
+                                                            output_description=data["output_description"],
+                                                            test_case_id=data["test_case_id"],
+                                                            samples=json.dumps(data["samples"]),
+                                                            time_limit=data["time_limit"],
+                                                            memory_limit=data["memory_limit"],
+                                                            difficulty=data["difficulty"],
+                                                            created_by=request.user,
+                                                            hint=data["hint"],
+                                                            contest=request.contest,
+                                                            sort_index=data["sort_index"])
+            return success_response(ContestProblemSerializer(contest_problem).data)
+        else:
+            serializer_invalid_response(serializer)
+
+    def put(self, request):
+        """
+        比赛题目编辑json api接口
+        ---
+        request_serializer: EditContestSerializer
+        response_serializer: ContestProblemSerializer
+        """
+        serializer = EditContestSerializer(data=request.data)
+        if serializer.is_valid():
+            data = serializer.data
+            try:
+                contest_problem = ContestProblem.objects.get(id=data["id"])
+            except ContestProblem.DoesNotExist:
+                return error_response(u"该比赛题目不存在！")
+            contest_problem.title = data["title"]
+            contest_problem.description = data["description"]
+            contest_problem.input_description = data["input_description"]
+            contest_problem.output_description = data["output_description"]
+            contest_problem.test_case_id = data["test_case_id"]
+            contest_problem.time_limit = data["time_limit"]
+            contest_problem.memory_limit = data["memory_limit"]
+            contest_problem.difficulty = data["difficulty"]
+            contest_problem.samples = json.dumps(data["samples"])
+            contest_problem.hint = data["hint"]
+            contest_problem.visible = data["visible"]
+            contest_problem.sort_index = data["sort_index"]
+            contest_problem.save()
+            return success_response(ContestProblem(contest_problem).data)
+        else:
+            return serializer_invalid_response(serializer)
+
+    def get(self, request):
+        """
+        比赛题目分页json api接口
+        ---
+        response_serializer: ProblemSerializer
+        """
+        contest_problem_id = request.GET.get("contest_problem_id", None)
+        if contest_problem_id:
+            try:
+                contest_problem = ContestProblemSerializer.objects.get(id=contest_problem_id)
+                return success_response(ContestProblemSerializer(contest_problem).data)
+            except contest_problem.DoesNotExist:
+                return error_response(u"比赛题目不存在")
+        contest_problem = ContestProblem.objects.all().order_by("sort_index")
+        visible = request.GET.get("visible", None)
+        if visible:
+            contest_problem = contest_problem.filter(visible=(visible == "true"))
+        keyword = request.GET.get("keyword", None)
+        if keyword:
+            contest_problem = contest_problem.filter(Q(title__contains=keyword) |
+                                                     Q(description__contains=keyword))
+
+        return paginate(request, contest_problem, ContestProblemSerializer)

problem/models.py

@@ -32,8 +32,6 @@ class AbstractProblem(models.Model):
     # last_update_time = models.DateTimeField(auto_now=True)
     # 这个题是谁创建的
     created_by = models.ForeignKey(User)
-    # 来源
-    source = models.CharField(max_length=30, blank=True, null=True)
     # 时间限制 单位是毫秒
     time_limit = models.IntegerField()
     # 内存限制 单位是MB
@@ -54,3 +52,5 @@ class Problem(AbstractProblem):
     difficulty = models.IntegerField()
     # 标签
     tags = models.ManyToManyField(ProblemTag)
+    # 来源
+    source = models.CharField(max_length=30, blank=True, null=True)

problem/tests.py

@@ -164,3 +164,8 @@ class ProblemTagAdminAPITest(APITestCase):
     def test_get_all_problem_tag_successfully(self):
         self.assertEqual(self.client.get(self.url).data["code"], 0)
 
+
+class ProblemListPageTest(TestCase):
+    def setUp(self):
+        pass
+