增加判断管理员权限的 decorator

This commit is contained in:
virusdefender 2015-09-22 16:18:32 +08:00
parent a57544db1d
commit 172d45eb78

View File

@ -4,7 +4,7 @@ from django.http import HttpResponse, HttpResponseRedirect
from django.shortcuts import render
from utils.shortcuts import error_response, error_page
from .models import User
from .models import User, SUPER_ADMIN
def login_required(func):
@ -12,7 +12,10 @@ def login_required(func):
def check(*args, **kwargs):
# 在class based views 里面args 有两个元素一个是self, 第二个才是request
# 在function based views 里面args 只有request 一个参数
request = args[-1]
if len(args) == 2:
request = args[-1]
else:
request = args[0]
if request.user.is_authenticated():
return func(*args, **kwargs)
if request.is_ajax():
@ -25,7 +28,10 @@ def login_required(func):
def admin_required(func):
@wraps(func)
def check(*args, **kwargs):
request = args[-1]
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():
@ -33,3 +39,20 @@ def admin_required(func):
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