mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2024-09-21 00:13:18 +00:00
增加创建提交的OpenAPI
This commit is contained in:
parent
823c6be0e1
commit
b4d31ed79b
@ -22,7 +22,7 @@ from admin.views import AdminTemplateView
|
||||
|
||||
from problem.views import TestCaseUploadAPIView, ProblemTagAdminAPIView, ProblemAdminAPIView, OpenAPIProblemAPI
|
||||
from submission.views import (SubmissionAPIView, SubmissionAdminAPIView, ContestSubmissionAPIView,
|
||||
SubmissionShareAPIView, SubmissionRejudgeAdminAPIView)
|
||||
SubmissionShareAPIView, SubmissionRejudgeAdminAPIView, OpenAPISubmitCodeAPI)
|
||||
from judge_dispatcher.views import AdminJudgeServerAPIView
|
||||
from utils.views import SimditorImageUploadAPIView
|
||||
|
||||
@ -100,6 +100,8 @@ urlpatterns = [
|
||||
url(r'^problem/(?P<problem_id>\d+)/submissions/$', "submission.views.problem_my_submissions_list_page",
|
||||
name="problem_my_submissions_page"),
|
||||
|
||||
url(r'^api/open/submission/$', OpenAPISubmitCodeAPI.as_view(), name="openapi_submit_code"),
|
||||
|
||||
url(r'^submission/(?P<submission_id>\w+)/$', "submission.views.my_submission", name="my_submission_page"),
|
||||
url(r'^submissions/$', "submission.views.my_submission_list_page", name="my_submission_list_page"),
|
||||
url(r'^submissions/(?P<page>\d+)/$', "submission.views.my_submission_list_page", name="my_submission_list_page"),
|
||||
|
@ -11,6 +11,13 @@ class CreateSubmissionSerializer(serializers.Serializer):
|
||||
code = serializers.CharField(max_length=20000)
|
||||
|
||||
|
||||
class OpenAPICreateSubmissionSerializer(serializers.Serializer):
|
||||
appkey = serializers.CharField(max_length=35)
|
||||
problem_id = serializers.IntegerField()
|
||||
language = serializers.IntegerField()
|
||||
code = serializers.CharField(max_length=20000)
|
||||
|
||||
|
||||
class SubmissionSerializer(serializers.ModelSerializer):
|
||||
user = serializers.SerializerMethodField("_get_submission_user")
|
||||
|
||||
|
@ -20,11 +20,57 @@ from .tasks import _judge
|
||||
from .models import Submission
|
||||
from .serializers import (CreateSubmissionSerializer, SubmissionSerializer,
|
||||
SubmissionhareSerializer, SubmissionRejudgeSerializer,
|
||||
CreateContestSubmissionSerializer)
|
||||
CreateContestSubmissionSerializer, OpenAPICreateSubmissionSerializer)
|
||||
|
||||
logger = logging.getLogger("app_info")
|
||||
|
||||
|
||||
def _submit_code(user, problem_id, language, code):
|
||||
controller = BucketController(user_id=user.id,
|
||||
redis_conn=redis.Redis(host=settings.REDIS_CACHE["host"],
|
||||
port=settings.REDIS_CACHE["port"],
|
||||
db=settings.REDIS_CACHE["db"]),
|
||||
default_capacity=settings.TOKEN_BUCKET_DEFAULT_CAPACITY)
|
||||
bucket = TokenBucket(fill_rate=settings.TOKEN_BUCKET_FILL_RATE,
|
||||
capacity=settings.TOKEN_BUCKET_DEFAULT_CAPACITY,
|
||||
last_capacity=controller.last_capacity,
|
||||
last_timestamp=controller.last_timestamp)
|
||||
if bucket.consume():
|
||||
controller.last_capacity -= 1
|
||||
else:
|
||||
return error_response(u"您提交的频率过快, 请等待%d秒" % int(bucket.expected_time() + 1))
|
||||
|
||||
try:
|
||||
problem = Problem.objects.get(id=problem_id)
|
||||
except Problem.DoesNotExist:
|
||||
return error_response(u"题目不存在")
|
||||
submission = Submission.objects.create(user_id=user.id,
|
||||
language=language,
|
||||
code=code,
|
||||
problem_id=problem.id)
|
||||
|
||||
try:
|
||||
_judge.delay(submission, problem.time_limit, problem.memory_limit, problem.test_case_id)
|
||||
except Exception as e:
|
||||
logger.error(e)
|
||||
return error_response(u"提交判题任务失败")
|
||||
return success_response({"submission_id": submission.id})
|
||||
|
||||
|
||||
class OpenAPISubmitCodeAPI(APIView):
|
||||
def post(self, request):
|
||||
serializer = OpenAPICreateSubmissionSerializer(data=request.data)
|
||||
if serializer.is_valid():
|
||||
data = serializer.data
|
||||
try:
|
||||
user = User.objects.get(openapi_appkey=data["appkey"])
|
||||
except User.DoesNotExist:
|
||||
return error_response(u"appkey无效")
|
||||
return _submit_code(user, data["problem_id"], data["language"], data["code"])
|
||||
else:
|
||||
return serializer_invalid_response(serializer)
|
||||
|
||||
|
||||
class SubmissionAPIView(APIView):
|
||||
@login_required
|
||||
def post(self, request):
|
||||
@ -33,38 +79,10 @@ class SubmissionAPIView(APIView):
|
||||
---
|
||||
request_serializer: CreateSubmissionSerializer
|
||||
"""
|
||||
controller = BucketController(user_id=request.user.id,
|
||||
redis_conn=redis.Redis(host=settings.REDIS_CACHE["host"],
|
||||
port=settings.REDIS_CACHE["port"],
|
||||
db=settings.REDIS_CACHE["db"]),
|
||||
default_capacity=settings.TOKEN_BUCKET_DEFAULT_CAPACITY)
|
||||
bucket = TokenBucket(fill_rate=settings.TOKEN_BUCKET_FILL_RATE,
|
||||
capacity=settings.TOKEN_BUCKET_DEFAULT_CAPACITY,
|
||||
last_capacity=controller.last_capacity,
|
||||
last_timestamp=controller.last_timestamp)
|
||||
if bucket.consume():
|
||||
controller.last_capacity -= 1
|
||||
else:
|
||||
return error_response(u"您提交的频率过快, 请等待%d秒" % int(bucket.expected_time() + 1))
|
||||
|
||||
serializer = CreateSubmissionSerializer(data=request.data)
|
||||
if serializer.is_valid():
|
||||
data = serializer.data
|
||||
try:
|
||||
problem = Problem.objects.get(id=data["problem_id"])
|
||||
except Problem.DoesNotExist:
|
||||
return error_response(u"题目不存在")
|
||||
submission = Submission.objects.create(user_id=request.user.id,
|
||||
language=int(data["language"]),
|
||||
code=data["code"],
|
||||
problem_id=problem.id)
|
||||
|
||||
try:
|
||||
_judge.delay(submission, problem.time_limit, problem.memory_limit, problem.test_case_id)
|
||||
except Exception as e:
|
||||
logger.error(e)
|
||||
return error_response(u"提交判题任务失败")
|
||||
return success_response({"submission_id": submission.id})
|
||||
return _submit_code(request.user, data["problem_id"], data["language"], data["code"])
|
||||
else:
|
||||
return serializer_invalid_response(serializer)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user