perf: 优化授权规则 user-permission 用户授权相关的 API; 包括 assets, nodes, tree-asset, tree-node, tree-node-with-asset;

apps/perms/api/user_permission/nodes.py

@@ -1,133 +1,49 @@
 # -*- 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 assets.models import Node
 from perms import serializers
-from perms.hands import User
 from perms.utils.user_permission import UserGrantedNodesQueryUtils
+from common.utils import get_logger, lazyproperty
 
-from .mixin import SelfOrPKUserMixin, RebuildTreeMixin
+from .mixin import SelfOrPKUserMixin
 
 logger = get_logger(__name__)
 
 __all__ = [
-    'UserPermedNodesApi',
-    'UserPermedNodesAsTreeApi',
+    'UserAllPermedNodesApi',
     'UserPermedNodeChildrenApi',
-    'UserPermedNodeChildrenAsTreeApi',
-    '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):
+class BaseUserPermedNodesApi(SelfOrPKUserMixin, ListAPIView):
     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_queryset(self):
+        if getattr(self, 'swagger_fake_view', False):
+            return Node.objects.none()
+        return self.get_nodes()
 
+    @abc.abstractmethod
     def get_nodes(self):
-        utils = UserGrantedNodesQueryUtils(self.user)
-        nodes = utils.get_whole_tree_nodes()
-        return nodes
+        return []
+
+    @lazyproperty
+    def query_node_util(self):
+        return UserGrantedNodesQueryUtils(self.user)
 
 
-# API
-
-
-class UserPermedNodeChildrenApi(
-    SelfOrPKUserMixin,
-    UserGrantedNodeChildrenMixin,
-    BaseNodeChildrenApi
-):
-    """ 用户授权的节点下的子节点"""
-    pass
-
-
-class UserPermedNodeChildrenAsTreeApi(
-    SelfOrPKUserMixin,
-    RebuildTreeMixin,
-    UserGrantedNodeChildrenMixin,
-    BaseNodeChildrenAsTreeApi
-):
-    """ 用户授权的节点下的子节点树"""
-    pass
-
-
-class UserPermedNodesApi(
-    SelfOrPKUserMixin,
-    UserGrantedNodesMixin,
-    BaseGrantedNodeApi
-):
+class UserAllPermedNodesApi(BaseUserPermedNodesApi):
     """ 用户授权的节点 """
-    pass
+    def get_nodes(self):
+        return self.query_node_util.get_whole_tree_nodes()
 
 
-class UserPermedNodesAsTreeApi(
-    SelfOrPKUserMixin,
-    RebuildTreeMixin,
-    UserGrantedNodesMixin,
-    BaseGrantedNodeAsTreeApi
-):
-    """ 用户授权的节点树 """
-    pass
+class UserPermedNodeChildrenApi(BaseUserPermedNodesApi):
+    """ 用户授权的节点下的子节点 """
+    def get_nodes(self):
+        key = self.request.query_params.get('key')
+        nodes = self.query_node_util.get_node_children(key)
+        return nodes