增加 @admin_required 修饰符代码和对应的测试

This commit is contained in:
virusdefender
2015-08-06 12:25:16 +08:00
parent 875e3b1671
commit 62f07e713f
4 changed files with 106 additions and 17 deletions

View File

@@ -20,5 +20,13 @@ def login_required(func):
return check
def admin_required():
pass
def admin_required(func):
def check(*args, **kwargs):
request = args[-1]
if request.user.is_authenticated() and request.user.admin_type:
return func(*args, **kwargs)
if request.is_ajax():
return error_response(u"需要管理员权限")
else:
return render(request, "utils/error.html", {"error": "需要管理员权限"})
return check