OnlineJudge/contest_submission/views.py

# coding=utf-8
import json
from datetime import datetime
import redis
import pytz
from django.shortcuts import render
from django.core.paginator import Paginator
from django.utils import timezone
from rest_framework.views import APIView

from judge.judger_controller.tasks import judge

from account.decorators import login_required
from account.models import SUPER_ADMIN

from contest.decorators import check_user_contest_permission
from contest.models import Contest, ContestProblem

from utils.shortcuts import serializer_invalid_response, error_response, success_response, error_page, paginate
from utils.cache import get_cache_redis
from submission.models import Submission
from .serializers import CreateContestSubmissionSerializer
from submission.serializers import SubmissionSerializer


class ContestSubmissionAPIView(APIView):
    @check_user_contest_permission
    def post(self, request):
        """
        创建比赛的提交
        ---
        request_serializer: CreateContestSubmissionSerializer
        """
        serializer = CreateContestSubmissionSerializer(data=request.data)
        if serializer.is_valid():
            data = serializer.data
            contest = Contest.objects.get(id=data["contest_id"])
            try:
                problem = ContestProblem.objects.get(contest=contest, id=data["problem_id"])
            except ContestProblem.DoesNotExist:
                return error_response(u"题目不存在")
            submission = Submission.objects.create(user_id=request.user.id, language=int(data["language"]),
                                                   contest_id=contest.id, code=data["code"], problem_id=problem.id)
            try:
                judge.delay(submission.id, problem.time_limit, problem.memory_limit, problem.test_case_id)
            except Exception:
                return error_response(u"提交判题任务失败")
            # 修改用户解题状态
            problems_status = request.user.problems_status
            if "contest_problems" not in problems_status:
                problems_status["contest_problems"] = {}
            problems_status["contest_problems"][str(data["problem_id"])] = 2
            request.user.problems_status = problems_status
            request.user.save()
            # 增加redis 中判题队列长度的计数器
            r = get_cache_redis()
            r.incr("judge_queue_length")
            return success_response({"submission_id": submission.id})
        else:
            return serializer_invalid_response(serializer)


@login_required
def contest_problem_my_submissions_list_page(request, contest_id, contest_problem_id):
    """
    我比赛单个题目的所有提交列表
    """
    try:
        Contest.objects.get(id=contest_id)
    except Contest.DoesNotExist:
        return error_page(request, u"比赛不存在")
    try:
        contest_problem = ContestProblem.objects.get(id=contest_problem_id, visible=True)
    except ContestProblem.DoesNotExist:
        return error_page(request, u"比赛问题不存在")
    submissions = Submission.objects.filter(user_id=request.user.id, problem_id=contest_problem.id).order_by(
        "-create_time"). \
        values("id", "result", "create_time", "accepted_answer_time", "language")
    return render(request, "oj/contest/my_submissions_list.html",
                  {"submissions": submissions, "problem": contest_problem})


@check_user_contest_permission
def contest_problem_submissions_list_page(request, contest_id, page=1):
    """
    单个比赛中的所有提交（包含自己和别人，自己可查提交结果，其他人不可查）
    """
    try:
        contest = Contest.objects.get(id=contest_id)
    except Contest.DoesNotExist:
        return error_page(request, u"比赛不存在")

    submissions = Submission.objects.filter(contest_id=contest_id).\
        values("id", "contest_id", "problem_id", "result", "create_time",
               "accepted_answer_time", "language", "user_id").order_by("-create_time")

    user_id = request.GET.get("user_id", None)
    if user_id:
        submissions = submissions.filter(user_id=request.GET.get("user_id"))

    # 封榜的时候只能看到自己的提交
    if not contest.real_time_rank:
        if not (request.user.admin_type == SUPER_ADMIN or request.user == contest.created_by):
            submissions = submissions.filter(user_id=request.user.id)

    language = request.GET.get("language", None)
    filter = None
    if language:
        submissions = submissions.filter(language=int(language))
        filter = {"name": "language", "content": language}
    result = request.GET.get("result", None)
    if result:
        submissions = submissions.filter(result=int(result))
        filter = {"name": "result", "content": result}
    paginator = Paginator(submissions, 20)

    # 为查询题目标题创建新字典
    title = {}
    contest_problems = ContestProblem.objects.filter(contest=contest)
    for item in contest_problems:
        title[item.id] = item.title
    for item in submissions:
        item['title'] = title[item['problem_id']]

    try:
        current_page = paginator.page(int(page))
    except Exception:
        return error_page(request, u"不存在的页码")
    previous_page = next_page = None
    try:
        previous_page = current_page.previous_page_number()
    except Exception:
        pass
    try:
        next_page = current_page.next_page_number()
    except Exception:
        pass

    for item in current_page:
        # 自己提交的 管理员和创建比赛的可以看到所有的提交链接
        if item["user_id"] == request.user.id or request.user.admin_type == SUPER_ADMIN or \
                        request.user == contest.created_by:
            item["show_link"] = True
        else:
            item["show_link"] = False

    return render(request, "oj/contest/submissions_list.html",
                  {"submissions": current_page, "page": int(page),
                   "previous_page": previous_page, "next_page": next_page, "start_id": int(page) * 20 - 20,
                   "contest": contest, "filter": filter, "user_id": user_id})


class ContestSubmissionAdminAPIView(APIView):
    def get(self, request):
        """
        查询比赛提交,单个比赛题目提交的adminAPI
        ---
        response_serializer: SubmissionSerializer
        """
        problem_id = request.GET.get("problem_id", None)
        contest_id = request.GET.get("contest_id", None)
        if contest_id:
            try:
                contest = Contest.objects.get(pk=contest_id)
            except Contest.DoesNotExist:
                return error_response(u"比赛不存在!")
            if request.user.admin_type != SUPER_ADMIN and contest.created_by != request.user:
                return error_response(u"您无权查看该信息!")
            submissions = Submission.objects.filter(contest_id=contest_id).order_by("-create_time")
        else:
            if problem_id:
                try:
                    contest_problem = ContestProblem.objects.get(pk=problem_id)
                except ContestProblem.DoesNotExist:
                    return error_response(u"问题不存在!")
                if request.user.admin_type != SUPER_ADMIN and contest_problem.contest.created_by != request.user:
                    return error_response(u"您无权查看该信息!")
                submissions = Submission.objects.filter(contest_id=contest_problem.contest_id).order_by("-create_time")
            else:
                return error_response(u"参数错误!")
        if problem_id:
            submissions = submissions.filter(problem_id=problem_id)
        return paginate(request, submissions, SubmissionSerializer)