mirror of https://github.com/jumpserver/jumpserver
134 lines
3.2 KiB
Python
134 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
import abc
|
|
from rest_framework.request import Request
|
|
from rest_framework.response import Response
|
|
from rest_framework.generics import ListAPIView
|
|
|
|
from common.utils import get_logger
|
|
from assets.api.mixin import SerializeToTreeNodeMixin
|
|
from perms import serializers
|
|
from perms.hands import User
|
|
from perms.utils.user_permission import UserGrantedNodesQueryUtils
|
|
|
|
from .mixin import SelfOrPKUserMixin, RebuildTreeMixin
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
__all__ = [
|
|
'UserGrantedNodesApi',
|
|
'UserGrantedNodesAsTreeApi',
|
|
'UserGrantedNodeChildrenApi',
|
|
'UserGrantedNodeChildrenAsTreeApi',
|
|
'BaseGrantedNodeAsTreeApi',
|
|
'UserGrantedNodesMixin',
|
|
]
|
|
|
|
|
|
class _GrantedNodeStructApi(ListAPIView, metaclass=abc.ABCMeta):
|
|
@property
|
|
def user(self):
|
|
raise NotImplementedError
|
|
|
|
def get_nodes(self):
|
|
# 不使用 `get_queryset` 单独定义 `get_nodes` 的原因是
|
|
# `get_nodes` 返回的不一定是 `queryset`
|
|
raise NotImplementedError
|
|
|
|
|
|
class NodeChildrenMixin:
|
|
def get_children(self):
|
|
raise NotImplementedError
|
|
|
|
def get_nodes(self):
|
|
nodes = self.get_children()
|
|
return nodes
|
|
|
|
|
|
class BaseGrantedNodeApi(_GrantedNodeStructApi, metaclass=abc.ABCMeta):
|
|
serializer_class = serializers.NodeGrantedSerializer
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
nodes = self.get_nodes()
|
|
serializer = self.get_serializer(nodes, many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class BaseNodeChildrenApi(NodeChildrenMixin, BaseGrantedNodeApi, metaclass=abc.ABCMeta):
|
|
pass
|
|
|
|
|
|
class BaseGrantedNodeAsTreeApi(SerializeToTreeNodeMixin, _GrantedNodeStructApi, metaclass=abc.ABCMeta):
|
|
def list(self, request: Request, *args, **kwargs):
|
|
nodes = self.get_nodes()
|
|
nodes = self.serialize_nodes(nodes, with_asset_amount=True)
|
|
return Response(data=nodes)
|
|
|
|
|
|
class BaseNodeChildrenAsTreeApi(NodeChildrenMixin, BaseGrantedNodeAsTreeApi, metaclass=abc.ABCMeta):
|
|
pass
|
|
|
|
|
|
class UserGrantedNodeChildrenMixin:
|
|
user: User
|
|
request: Request
|
|
|
|
def get_children(self):
|
|
user = self.user
|
|
key = self.request.query_params.get('key')
|
|
nodes = UserGrantedNodesQueryUtils(user).get_node_children(key)
|
|
return nodes
|
|
|
|
|
|
class UserGrantedNodesMixin:
|
|
"""
|
|
查询用户授权的所有节点 直接授权节点 + 授权资产关联的节点
|
|
"""
|
|
user: User
|
|
|
|
def get_nodes(self):
|
|
utils = UserGrantedNodesQueryUtils(self.user)
|
|
nodes = utils.get_whole_tree_nodes()
|
|
return nodes
|
|
|
|
|
|
# API
|
|
|
|
|
|
class UserGrantedNodeChildrenApi(
|
|
SelfOrPKUserMixin,
|
|
UserGrantedNodeChildrenMixin,
|
|
BaseNodeChildrenApi
|
|
):
|
|
""" 用户授权的节点下的子节点"""
|
|
pass
|
|
|
|
|
|
class UserGrantedNodeChildrenAsTreeApi(
|
|
SelfOrPKUserMixin,
|
|
RebuildTreeMixin,
|
|
UserGrantedNodeChildrenMixin,
|
|
BaseNodeChildrenAsTreeApi
|
|
):
|
|
""" 用户授权的节点下的子节点树"""
|
|
pass
|
|
|
|
|
|
class UserGrantedNodesApi(
|
|
SelfOrPKUserMixin,
|
|
UserGrantedNodesMixin,
|
|
BaseGrantedNodeApi
|
|
):
|
|
""" 用户授权的节点 """
|
|
pass
|
|
|
|
|
|
class UserGrantedNodesAsTreeApi(
|
|
SelfOrPKUserMixin,
|
|
RebuildTreeMixin,
|
|
UserGrantedNodesMixin,
|
|
BaseGrantedNodeAsTreeApi
|
|
):
|
|
""" 用户授权的节点树 """
|
|
pass
|