mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-08-28 11:01:13 +00:00
1f8ded49fa
* perf: 工作台受组织角色控制 * perf: workspace => workbench * perf: 修改 workspace codename Co-authored-by: ibuler <ibuler@qq.com>
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from rest_framework.request import Request
|
|
|
|
from common.http import is_true
|
|
from common.mixins.api import RoleAdminMixin as _RoleAdminMixin
|
|
from common.mixins.api import RoleUserMixin as _RoleUserMixin
|
|
from orgs.utils import tmp_to_root_org
|
|
from users.models import User
|
|
from perms.utils.asset.user_permission import UserGrantedTreeRefreshController
|
|
|
|
|
|
class PermBaseMixin:
|
|
user: User
|
|
|
|
def get(self, request: Request, *args, **kwargs):
|
|
force = is_true(request.query_params.get('rebuild_tree'))
|
|
controller = UserGrantedTreeRefreshController(self.user)
|
|
controller.refresh_if_need(force)
|
|
return super().get(request, *args, **kwargs)
|
|
|
|
|
|
class AssetRoleAdminMixin(PermBaseMixin, _RoleAdminMixin):
|
|
rbac_perms = (
|
|
('list', 'perms.view_userassets'),
|
|
('retrieve', 'perms.view_userassets'),
|
|
('get_tree', 'perms.view_userassets'),
|
|
('GET', 'perms.view_userassets'),
|
|
)
|
|
|
|
|
|
class AssetRoleUserMixin(PermBaseMixin, _RoleUserMixin):
|
|
rbac_perms = (
|
|
('list', 'perms.view_myassets'),
|
|
('retrieve', 'perms.view_myassets'),
|
|
('get_tree', 'perms.view_myassets'),
|
|
('GET', 'perms.view_myassets'),
|
|
)
|