OnlineJudge/account/decorators.py

# coding=utf-8
from functools import wraps
from django.http import HttpResponse, HttpResponseRedirect
from django.shortcuts import render

from utils.shortcuts import error_response, error_page
from .models import User, SUPER_ADMIN


def login_required(func):
    @wraps(func)
    def check(*args, **kwargs):
        # 在class based views 里面，args 有两个元素，一个是self, 第二个才是request，
        # 在function based views 里面，args 只有request 一个参数
        if len(args) == 2:
            request = args[-1]
        else:
            request = args[0]
        if request.user.is_authenticated():
            return func(*args, **kwargs)
        if request.is_ajax():
            return error_response(u"请先登录")
        else:
            return HttpResponseRedirect("/login/")
    return check


def admin_required(func):
    @wraps(func)
    def check(*args, **kwargs):
        if len(args) == 2:
            request = args[-1]
        else:
            request = args[0]
        if request.user.is_authenticated() and request.user.admin_type:
            return func(*args, **kwargs)
        if request.is_ajax():
            return error_response(u"需要管理员权限")
        else:
            return error_page(request, u"需要管理员权限，如果没有登录，请先登录")
    return check


def super_admin_required(func):
    @wraps(func)
    def check(*args, **kwargs):
        if len(args) == 2:
            request = args[-1]
        else:
            request = args[0]
        if request.user.is_authenticated() and request.user.admin_type == SUPER_ADMIN:
            return func(*args, **kwargs)
        if request.is_ajax():
            return error_response(u"需要超级管理员权限")
        else:
            return error_page(request, u"需要超级管理员权限")

    return check