jumpserver/apps/perms/tree/app.py

from urllib.parse import urlencode, parse_qsl

from django.utils.translation import ugettext as _
from rest_framework.generics import get_object_or_404

from common.tree import TreeNode
from orgs.models import Organization
from assets.models import SystemUser
from applications.utils import KubernetesTree
from applications.models import Application
from perms.utils.application.permission import get_application_system_user_ids


class GrantedAppTreeUtil:
    @staticmethod
    def filter_organizations(applications):
        organization_ids = set(applications.values_list('org_id', flat=True))
        organizations = [Organization.get_instance(org_id) for org_id in organization_ids]
        organizations.sort(key=lambda x: x.name)
        return organizations

    @staticmethod
    def create_root_node():
        name = _('My applications')
        node = TreeNode(**{
            'id': 'applications',
            'name': name,
            'title': name,
            'pId': '',
            'open': True,
            'isParent': True,
            'meta': {
                'type': 'root'
            }
        })
        return node

    @staticmethod
    def get_children_nodes(tree_id, parent_info, user):
        tree_nodes = []
        parent_info = dict(parse_qsl(parent_info))
        pod_name = parent_info.get('pod')
        app_id = parent_info.get('app_id')
        namespace = parent_info.get('namespace')
        system_user_id = parent_info.get('system_user_id')

        if app_id and not any([pod_name, namespace, system_user_id]):
            app = get_object_or_404(Application, id=app_id)
            system_user_ids = get_application_system_user_ids(user, app)
            system_users = SystemUser.objects.filter(id__in=system_user_ids).order_by('priority')
            for system_user in system_users:
                system_user_node = KubernetesTree(tree_id).as_system_user_tree_node(
                    system_user, parent_info
                )
                tree_nodes.append(system_user_node)
            return tree_nodes
        tree_nodes = KubernetesTree(tree_id).async_tree_node(parent_info)
        return tree_nodes

    def create_tree_nodes(self, applications):
        tree_nodes = []
        if not applications:
            return tree_nodes

        root_node = self.create_root_node()
        tree_nodes.append(root_node)
        organizations = self.filter_organizations(applications)

        for i, org in enumerate(organizations):
            tree_id = urlencode({'org_id': str(org.id)})
            apps = applications.filter(org_id=org.id)

            # 组织节点
            org_node = org.as_tree_node(oid=tree_id, pid=root_node.id)
            org_node.name += '({})'.format(apps.count())
            tree_nodes.append(org_node)

            # 类别节点
            category_type_nodes = Application.create_category_type_tree_nodes(
                apps, tree_id, show_empty=False
            )
            tree_nodes += category_type_nodes

            for app in apps:
                app_node = app.as_tree_node(tree_id, k8s_as_tree=True)
                tree_nodes.append(app_node)
        return tree_nodes