feat: 限制超级权限

pull/11559/head
ibuler 2023-09-14 10:42:16 +08:00
parent 54d0a1b871
commit 82f96d6ed2
3 changed files with 17 additions and 4 deletions

View File

@ -578,7 +578,9 @@ class Config(dict):
'FTP_FILE_MAX_STORE': 100, 'FTP_FILE_MAX_STORE': 100,
# API 请求次数限制 # API 请求次数限制
'MAX_LIMIT_PER_PAGE': 100 'MAX_LIMIT_PER_PAGE': 100,
'LIMIT_SUPER_PRIV': False,
} }
old_config_map = { old_config_map = {

View File

@ -203,3 +203,4 @@ MAX_LIMIT_PER_PAGE = CONFIG.MAX_LIMIT_PER_PAGE
# Magnus DB Port # Magnus DB Port
MAGNUS_ORACLE_PORTS = CONFIG.MAGNUS_ORACLE_PORTS MAGNUS_ORACLE_PORTS = CONFIG.MAGNUS_ORACLE_PORTS
LIMIT_SUPER_PRIV = CONFIG.LIMIT_SUPER_PRIV

View File

@ -400,10 +400,17 @@ class RoleMixin:
data = cache.get(key) data = cache.get(key)
if data: if data:
return data return data
console_orgs = RoleBinding.get_user_has_the_perm_orgs('rbac.view_console', self)
audit_orgs = RoleBinding.get_user_has_the_perm_orgs('rbac.view_audit', self)
workbench_orgs = RoleBinding.get_user_has_the_perm_orgs('rbac.view_workbench', self)
if settings.LIMIT_SUPER_PRIV:
audit_orgs = list(set(audit_orgs) - set(console_orgs))
data = { data = {
'console_orgs': RoleBinding.get_user_has_the_perm_orgs('rbac.view_console', self), 'console_orgs': console_orgs,
'audit_orgs': RoleBinding.get_user_has_the_perm_orgs('rbac.view_audit', self), 'audit_orgs': audit_orgs,
'workbench_orgs': RoleBinding.get_user_has_the_perm_orgs('rbac.view_workbench', self), 'workbench_orgs': workbench_orgs,
} }
cache.set(key, data, 60 * 60) cache.set(key, data, 60 * 60)
return data return data
@ -541,6 +548,9 @@ class RoleMixin:
def get_all_permissions(self): def get_all_permissions(self):
from rbac.models import RoleBinding from rbac.models import RoleBinding
perms = RoleBinding.get_user_perms(self) perms = RoleBinding.get_user_perms(self)
if settings.LIMIT_SUPER_PRIV and 'view_console' in perms:
perms = [p for p in perms if p != "view_audit"]
return perms return perms