修改判题设置；增加web 和 mongodb 数据库的操作

submission/views.py

@@ -0,0 +1,52 @@
+# coding=utf-8
+import pymongo
+from bson.objectid import ObjectId
+
+from django.shortcuts import render
+
+from rest_framework.views import APIView
+
+from django.conf import settings
+
+from judge.judger.result import result
+from account.decorators import login_required
+from utils.shortcuts import serializer_invalid_response, error_response, success_response
+from .serializers import  CreateSubmissionSerializer
+
+
+class SubmissionnAPIView(APIView):
+    def _create_mondodb_connection(self):
+        mongodb_setting = settings.DATABASES["mongodb"]
+        connection = pymongo.MongoClient(host=mongodb_setting["HOST"], port=mongodb_setting["PORT"])
+        return connection["oj"]["oj_submission"]
+
+    # @login_required
+    def post(self, request):
+        """
+        提交代码
+        ---
+        request_serializer: CreateSubmissionSerializer
+        """
+        serializer = CreateSubmissionSerializer(data=request.data)
+        if serializer.is_valid():
+            data = serializer.data
+            data["user_id"] = request.user.id
+            data["result"] = result["waiting"]
+            mongodb_setting = settings.DATABASES["mongodb"]
+            connection = pymongo.MongoClient(host=mongodb_setting["HOST"], port=mongodb_setting["PORT"])
+            collection = connection["oj"]["oj_submission"]
+            submission_id = str(collection.insert_one(data).inserted_id)
+            return success_response({"submission_id": submission_id})
+        else:
+            return serializer_invalid_response(serializer)
+
+    # @login_required
+    def get(self, request):
+        submission_id = request.GET.get("submission_id", None)
+        if not submission_id:
+            return error_response(u"参数错误")
+        submission = self._create_mondodb_connection().find_one({"_id": ObjectId(submission_id), "user_id": result.user.id})
+        if submission:
+            return success_response({"result": submission["result"]})
+        else:
+            return error_response(u"提交不存在")