修改部分命名风格

This commit is contained in:
virusdefender
2015-08-12 14:56:18 +08:00
parent 88be032a38
commit 2ba1fa5669
4 changed files with 21 additions and 12 deletions

View File

@@ -9,7 +9,9 @@ from rest_framework.views import APIView
from django.conf import settings
from judge.judger.result import result
from judge.controller.tasks import judge
from account.decorators import login_required
from problem.models import Problem
from utils.shortcuts import serializer_invalid_response, error_response, success_response
from .serializers import CreateSubmissionSerializer
@@ -32,10 +34,15 @@ class SubmissionnAPIView(APIView):
data = serializer.data
data["user_id"] = request.user.id
data["result"] = result["waiting"]
try:
problem = Problem.objects.get(id=data["problem_id"])
except Problem.DoesNotExist:
return error_response(u"题目不存在")
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)
judge.deply(submission_id, problem.max_cpu_time, problem_)
return success_response({"submission_id": submission_id})
else:
return serializer_invalid_response(serializer)