feat: 只有系统管理员才能更新或删除系统管理员 (#10306)

Co-authored-by: feng <1304903146@qq.com>
This commit is contained in:
fit2bot
2023-04-21 17:31:39 +08:00
committed by GitHub
parent a105748a55
commit a1f65bccc5
2 changed files with 21 additions and 1 deletions

View File

@@ -1,5 +1,6 @@
from rest_framework import permissions
from rbac.builtin import BuiltinRole
from .utils import is_auth_password_time_valid
@@ -7,4 +8,20 @@ class IsAuthPasswdTimeValid(permissions.IsAuthenticated):
def has_permission(self, request, view):
return super().has_permission(request, view) \
and is_auth_password_time_valid(request.session)
and is_auth_password_time_valid(request.session)
class UserObjectPermission(permissions.BasePermission):
def has_object_permission(self, request, view, obj):
if view.action not in ['update', 'partial_update', 'destroy']:
return True
user = request.user
if user.is_superuser:
return True
system_admin_id = BuiltinRole.system_admin.id
return system_admin_id not in [
str(r.id) for r in obj.system_roles.all()
]