增加禁用用户的功能
This commit is contained in:
@@ -5,7 +5,7 @@ from functools import wraps
|
||||
|
||||
from django.http import HttpResponseRedirect
|
||||
|
||||
from utils.shortcuts import error_response
|
||||
from utils.shortcuts import error_response, error_page
|
||||
from .models import SUPER_ADMIN, ADMIN
|
||||
|
||||
|
||||
@@ -23,6 +23,8 @@ class BasePermissionDecorator(object):
|
||||
self.request = args[0]
|
||||
|
||||
if self.check_permission():
|
||||
if self.request.user.is_forbidden is True:
|
||||
return error_page(self.request, u"用户被禁用,请联系管理员")
|
||||
return self.func(*args, **kwargs)
|
||||
else:
|
||||
if self.request.is_ajax():
|
||||
@@ -46,4 +48,4 @@ class super_admin_required(BasePermissionDecorator):
|
||||
|
||||
class admin_required(BasePermissionDecorator):
|
||||
def check_permission(self):
|
||||
return self.request.user.is_authenticated() and self.request.user.admin_type in [SUPER_ADMIN, ADMIN]
|
||||
return self.request.user.is_authenticated() and self.request.user.admin_type in [SUPER_ADMIN, ADMIN]
|
||||
|
||||
Reference in New Issue
Block a user