2015-08-23 06:31:53 +00:00
|
|
|
|
# coding=utf-8
|
2015-10-20 12:07:35 +00:00
|
|
|
|
import urllib
|
2015-08-23 06:31:53 +00:00
|
|
|
|
from functools import wraps
|
|
|
|
|
|
2015-08-23 12:45:02 +00:00
|
|
|
|
from django.http import HttpResponse, HttpResponseRedirect
|
2015-08-23 06:31:53 +00:00
|
|
|
|
from django.shortcuts import render
|
2015-11-03 15:55:14 +00:00
|
|
|
|
from django.core.urlresolvers import reverse
|
2015-08-23 06:31:53 +00:00
|
|
|
|
|
|
|
|
|
from utils.shortcuts import error_response, error_page
|
|
|
|
|
|
2015-12-09 12:02:47 +00:00
|
|
|
|
from account.models import SUPER_ADMIN, ADMIN
|
2015-12-05 07:46:54 +00:00
|
|
|
|
from .models import (Contest, PASSWORD_PROTECTED_CONTEST, PASSWORD_PROTECTED_GROUP_CONTEST, PUBLIC_CONTEST, GROUP_CONTEST,
|
2015-09-21 03:16:30 +00:00
|
|
|
|
CONTEST_ENDED, CONTEST_NOT_START, CONTEST_UNDERWAY)
|
2015-08-23 06:31:53 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def check_user_contest_permission(func):
|
|
|
|
|
@wraps(func)
|
|
|
|
|
def _check_user_contest_permission(*args, **kwargs):
|
|
|
|
|
"""
|
|
|
|
|
这个函数检查当前的这个比赛对于 request 的用户来说能不能参加
|
|
|
|
|
需要比较:比赛的开始和结束时间、比赛是否有密码、比赛是不是限定指定小组参加
|
|
|
|
|
如果是有密码或者限定指定小组参加的话,即使比赛已经结束,那么也是可以看到所有的题目和结果的
|
|
|
|
|
否则不能看到这个比赛的题目结果排名等等
|
|
|
|
|
"""
|
|
|
|
|
# CBV 的情况,第一个参数是self,第二个参数是request
|
|
|
|
|
if len(args) == 2:
|
|
|
|
|
request = args[-1]
|
|
|
|
|
else:
|
|
|
|
|
request = args[0]
|
|
|
|
|
|
|
|
|
|
if not request.user.is_authenticated():
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"请先登录")
|
|
|
|
|
else:
|
2015-10-20 12:07:35 +00:00
|
|
|
|
return HttpResponseRedirect("/login/?__from=" + urllib.quote(request.build_absolute_uri()))
|
2015-08-23 06:31:53 +00:00
|
|
|
|
|
2015-10-17 08:49:02 +00:00
|
|
|
|
# kwargs 就包含了 url 里面的参数
|
2015-08-23 06:31:53 +00:00
|
|
|
|
if "contest_id" in kwargs:
|
|
|
|
|
contest_id = kwargs["contest_id"]
|
|
|
|
|
elif "contest_id" in request.data:
|
|
|
|
|
contest_id = request.data["contest_id"]
|
2015-10-18 03:45:06 +00:00
|
|
|
|
elif "contest_id" in request.GET:
|
|
|
|
|
contest_id = request.GET["contest_id"]
|
2015-08-23 06:31:53 +00:00
|
|
|
|
else:
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"参数错误")
|
|
|
|
|
else:
|
|
|
|
|
return error_page(request, u"参数错误")
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
contest = Contest.objects.get(id=contest_id)
|
|
|
|
|
except Contest.DoesNotExist:
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"比赛不存在")
|
|
|
|
|
else:
|
|
|
|
|
return error_page(request, u"比赛不存在")
|
|
|
|
|
|
|
|
|
|
if request.user.admin_type == SUPER_ADMIN or request.user == contest.created_by:
|
|
|
|
|
return func(*args, **kwargs)
|
2015-12-09 12:02:47 +00:00
|
|
|
|
if request.user.admin_type == ADMIN:
|
|
|
|
|
contest_set = Contest.objects.filter(groups__in=request.user.managed_groups.all())
|
|
|
|
|
if contest in contest_set:
|
|
|
|
|
return func(*args, **kwargs)
|
2015-10-17 08:49:02 +00:00
|
|
|
|
# 管理员可见隐藏的比赛,已经先判断了身份
|
|
|
|
|
if not contest.visible:
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"比赛不存在")
|
|
|
|
|
else:
|
|
|
|
|
return error_page(request, u"比赛不存在")
|
|
|
|
|
|
2015-08-23 06:31:53 +00:00
|
|
|
|
# 有密码的公开赛
|
2015-12-05 08:19:22 +00:00
|
|
|
|
if contest.contest_type == PASSWORD_PROTECTED_CONTEST:
|
2015-08-23 06:31:53 +00:00
|
|
|
|
# 没有输入过密码
|
|
|
|
|
if contest.id not in request.session.get("contests", []):
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"请先输入密码")
|
|
|
|
|
else:
|
|
|
|
|
return render(request, "oj/contest/no_contest_permission.html",
|
|
|
|
|
{"reason": "password_protect", "show_tab": False, "contest": contest})
|
|
|
|
|
|
|
|
|
|
# 指定小组参加的
|
2015-09-21 03:16:30 +00:00
|
|
|
|
if contest.contest_type == GROUP_CONTEST:
|
2015-08-23 06:31:53 +00:00
|
|
|
|
if not contest.groups.filter(id__in=request.user.group_set.all()).exists():
|
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"只有指定小组的可以参加这场比赛")
|
|
|
|
|
else:
|
|
|
|
|
return render(request, "oj/contest/no_contest_permission.html",
|
|
|
|
|
{"reason": "group_limited", "show_tab": False, "contest": contest})
|
2015-12-05 08:19:22 +00:00
|
|
|
|
|
|
|
|
|
if contest.contest_type == PASSWORD_PROTECTED_GROUP_CONTEST:
|
|
|
|
|
if not contest.groups.filter(id__in=request.user.group_set.all()).exists():
|
|
|
|
|
if contest.id not in request.session.get("contests", []):
|
2015-12-05 08:29:07 +00:00
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"请先输入密码")
|
|
|
|
|
else:
|
|
|
|
|
return render(request, "oj/contest/no_contest_permission.html",
|
|
|
|
|
{"reason": "password_protect", "show_tab": False, "contest": contest})
|
|
|
|
|
|
2015-08-23 06:31:53 +00:00
|
|
|
|
# 比赛没有开始
|
2015-09-21 03:16:30 +00:00
|
|
|
|
if contest.status == CONTEST_NOT_START:
|
2015-08-23 06:31:53 +00:00
|
|
|
|
if request.is_ajax():
|
|
|
|
|
return error_response(u"比赛还没有开始")
|
|
|
|
|
else:
|
|
|
|
|
return render(request, "oj/contest/no_contest_permission.html",
|
|
|
|
|
{"reason": "contest_not_start", "show_tab": False, "contest": contest})
|
|
|
|
|
|
2015-11-02 13:03:16 +00:00
|
|
|
|
# 比赛已经结束了,只拦截 ajax 的答案提交
|
2015-11-03 15:55:14 +00:00
|
|
|
|
if contest.status == CONTEST_ENDED and request.path == reverse("contest_submission_api") and request.is_ajax():
|
2015-11-02 13:03:16 +00:00
|
|
|
|
return error_response(u"比赛已经结束")
|
|
|
|
|
|
2015-08-23 06:31:53 +00:00
|
|
|
|
return func(*args, **kwargs)
|
|
|
|
|
|
|
|
|
|
return _check_user_contest_permission
|