2015-08-18 06:59:00 +00:00
|
|
|
|
# coding=utf-8
|
|
|
|
|
import json
|
2015-06-26 07:59:53 +00:00
|
|
|
|
from django.shortcuts import render
|
2015-08-18 06:59:00 +00:00
|
|
|
|
from django.db import IntegrityError
|
|
|
|
|
from django.db.models import Q
|
|
|
|
|
from rest_framework.views import APIView
|
2015-08-18 12:12:27 +00:00
|
|
|
|
from utils.shortcuts import (serializer_invalid_response, error_response,
|
|
|
|
|
success_response, paginate, rand_str, error_page)
|
|
|
|
|
|
|
|
|
|
from account.models import REGULAR_USER, ADMIN, SUPER_ADMIN
|
|
|
|
|
from group.models import Group
|
2015-06-26 07:59:53 +00:00
|
|
|
|
|
2015-08-18 06:59:00 +00:00
|
|
|
|
from .models import Contest, ContestProblem
|
|
|
|
|
from .serializers import (CreateContestSerializer, ContestSerializer, EditContestSerializer,
|
|
|
|
|
CreateContestProblemSerializer, ContestProblemSerializer, EditContestProblemSerializer)
|
2015-08-18 12:12:27 +00:00
|
|
|
|
|
2015-08-18 06:59:00 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|
2015-08-18 12:12:27 +00:00
|
|
|
|
groups = []
|
|
|
|
|
# 首先判断比赛的类型: 0 即为是小组赛,1 即为是无密码的公开赛,2 即为是有密码的公开赛
|
|
|
|
|
# 此时为有密码的公开赛,并且此时只能超级管理员才有权限此创建比赛
|
2015-08-19 09:53:43 +00:00
|
|
|
|
if data["contest_type"] in [1, 2]:
|
|
|
|
|
if request.user.admin_type != SUPER_ADMIN:
|
|
|
|
|
return error_response(u"只有超级管理员才可创建公开赛")
|
2015-08-18 12:12:27 +00:00
|
|
|
|
if data["contest_type"] == 2:
|
|
|
|
|
if not data["password"]:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
return error_response(u"此比赛为有密码的公开赛,密码不可为空")
|
|
|
|
|
|
|
|
|
|
# 没有密码的公开赛 没有密码的小组赛
|
|
|
|
|
elif data["contest_type"] == 0:
|
|
|
|
|
if request.user.admin_type == SUPER_ADMIN:
|
|
|
|
|
groups = Group.objects.filter(id__in=data["groups"])
|
2015-08-18 12:12:27 +00:00
|
|
|
|
else:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
groups = Group.objects.filter(id__in=data["groups"], admin=request.user)
|
|
|
|
|
if not groups.count():
|
|
|
|
|
return error_response(u"请至少选择一个小组")
|
2015-08-18 06:59:00 +00:00
|
|
|
|
try:
|
|
|
|
|
contest = Contest.objects.create(title=data["title"], description=data["description"],
|
|
|
|
|
mode=data["mode"], show_rank=data["show_rank"],
|
2015-08-18 12:12:27 +00:00
|
|
|
|
show_user_submission=data["show_user_submission"],
|
2015-08-19 09:53:43 +00:00
|
|
|
|
start_time=["start_time"], end_time=data["end_time"],
|
|
|
|
|
password=data["password"])
|
2015-08-18 06:59:00 +00:00
|
|
|
|
except IntegrityError:
|
|
|
|
|
return error_response(u"比赛名已经存在")
|
2015-08-18 12:12:27 +00:00
|
|
|
|
contest.groups.add(*groups)
|
2015-08-18 06:59:00 +00:00
|
|
|
|
return success_response(ContestSerializer(contest).data)
|
|
|
|
|
else:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-08-18 06:59:00 +00:00
|
|
|
|
|
|
|
|
|
def put(self, request):
|
|
|
|
|
"""
|
|
|
|
|
比赛编辑json api接口
|
|
|
|
|
---
|
|
|
|
|
request_serializer: EditContestSerializer
|
|
|
|
|
response_serializer: ContestSerializer
|
|
|
|
|
"""
|
|
|
|
|
serializer = EditContestSerializer(data=request.data)
|
|
|
|
|
if serializer.is_valid():
|
|
|
|
|
data = serializer.data
|
2015-08-19 09:53:43 +00:00
|
|
|
|
groups = []
|
2015-08-18 06:59:00 +00:00
|
|
|
|
try:
|
|
|
|
|
contest = Contest.objects.get(id=data["id"])
|
|
|
|
|
except Contest.DoesNotExist:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
return error_response(u"该比赛不存在!")
|
2015-08-18 06:59:00 +00:00
|
|
|
|
try:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
contest = Contest.objects.get(title=data["title"])
|
|
|
|
|
if contest.id != data["id"]:
|
|
|
|
|
return error_response(u"该比赛名称已经存在")
|
|
|
|
|
except Contest.DoesNotExist:
|
|
|
|
|
pass
|
|
|
|
|
if data["contest_type"] in [1, 2]:
|
|
|
|
|
if request.user.admin_type != SUPER_ADMIN:
|
|
|
|
|
return error_response(u"只有超级管理员才可创建公开赛")
|
|
|
|
|
if data["contest_type"] == 2:
|
|
|
|
|
if not data["password"]:
|
|
|
|
|
return error_response(u"此比赛为有密码的公开赛,密码不可为空")
|
|
|
|
|
elif data["contest_type"] == 0:
|
|
|
|
|
if request.user.admin_type == SUPER_ADMIN:
|
|
|
|
|
groups = Group.objects.filter(id__in=data["groups"])
|
|
|
|
|
else:
|
|
|
|
|
groups = Group.objects.filter(id__in=data["groups"], admin=request.user)
|
|
|
|
|
if not groups.count():
|
|
|
|
|
return error_response(u"请至少选择一个小组")
|
|
|
|
|
contest.title = data["title"]
|
|
|
|
|
contest.description = data["description"]
|
|
|
|
|
contest.mode = data["mode"]
|
|
|
|
|
contest.show_rank = data["show_rank"]
|
|
|
|
|
contest.show_user_submission = data["show_user_submission"]
|
|
|
|
|
contest.start_time = ["start_time"]
|
|
|
|
|
contest.end_time = data["end_time"]
|
|
|
|
|
contest.password = data["password"]
|
|
|
|
|
contest.save()
|
2015-08-18 06:59:00 +00:00
|
|
|
|
|
2015-08-19 09:53:43 +00:00
|
|
|
|
contest.groups.clear()
|
|
|
|
|
contest.groups.add(*groups)
|
2015-08-18 06:59:00 +00:00
|
|
|
|
return success_response(ContestSerializer(contest).data)
|
|
|
|
|
else:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-08-18 06:59:00 +00:00
|
|
|
|
|
|
|
|
|
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:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
return serializer_invalid_response(serializer)
|
2015-08-18 06:59:00 +00:00
|
|
|
|
|
|
|
|
|
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:
|
2015-08-19 09:53:43 +00:00
|
|
|
|
contest_problem = ContestProblem.objects.get(id=contest_problem_id)
|
2015-08-18 06:59:00 +00:00
|
|
|
|
return success_response(ContestProblemSerializer(contest_problem).data)
|
2015-08-19 09:53:43 +00:00
|
|
|
|
except ContestProblem.DoesNotExist:
|
2015-08-18 06:59:00 +00:00
|
|
|
|
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)
|