47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
"""
|
||
管理员模块权限
|
||
"""
|
||
from rest_framework.permissions import BasePermission
|
||
|
||
|
||
class IsAdminUser(BasePermission):
|
||
"""
|
||
验证是否为管理员用户（来自AdminUser模型）
|
||
"""
|
||
def has_permission(self, request, view):
|
||
from apps.admins.models import AdminUser
|
||
return (
|
||
request.user and
|
||
request.user.is_authenticated and
|
||
isinstance(request.user, AdminUser) and
|
||
request.user.is_active
|
||
)
|
||
|
||
|
||
class IsSuperAdmin(BasePermission):
|
||
"""
|
||
验证是否为超级管理员
|
||
"""
|
||
def has_permission(self, request, view):
|
||
from apps.admins.models import AdminUser
|
||
return (
|
||
request.user and
|
||
request.user.is_authenticated and
|
||
isinstance(request.user, AdminUser) and
|
||
request.user.role == 'super_admin'
|
||
)
|
||
|
||
|
||
class IsAdminOrOperator(BasePermission):
|
||
"""
|
||
验证是否为管理员或操作员
|
||
"""
|
||
def has_permission(self, request, view):
|
||
from apps.admins.models import AdminUser
|
||
return (
|
||
request.user and
|
||
request.user.is_authenticated and
|
||
isinstance(request.user, AdminUser) and
|
||
request.user.role in ['super_admin', 'admin', 'operator']
|
||
)
|