github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: k8s tree

pull/9229/head
feng 2022-12-21 17:32:55 +08:00
parent c304a58c05
commit 510ca9a5b8
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/perms/api/user_permission/tree/node_with_asset.py
View File

@ -159,8 +159,8 @@ class UserGrantedK8sAsTreeApi(SelfOrPKUserMixin, ListAPIView):
else: else:
return account.secret return account.secret
def get_namespace_and_pod(self): @staticmethod
key = self.request.query_params.get('key') def get_namespace_and_pod(key):
namespace_and_pod = dict(parse_qsl(key)) namespace_and_pod = dict(parse_qsl(key))
pod = namespace_and_pod.get('pod') pod = namespace_and_pod.get('pod')
namespace = namespace_and_pod.get('namespace') namespace = namespace_and_pod.get('namespace')
@ -170,11 +170,12 @@ class UserGrantedK8sAsTreeApi(SelfOrPKUserMixin, ListAPIView):
token = self.get_token() token = self.get_token()
asset = token.asset asset = token.asset
secret = self.get_account_secret(token) secret = self.get_account_secret(token)
namespace, pod = self.get_namespace_and_pod() key = self.request.query_params.get('key')
namespace, pod = self.get_namespace_and_pod(key)
tree = [] tree = []
k8s_tree_instance = KubernetesTree(asset, secret) k8s_tree_instance = KubernetesTree(asset, secret)
if not any([namespace, pod]): if not any([namespace, pod]) and not key:
asset_node = k8s_tree_instance.as_asset_tree_node() asset_node = k8s_tree_instance.as_asset_tree_node()
tree.append(asset_node) tree.append(asset_node)
tree.extend(k8s_tree_instance.async_tree_node(namespace, pod)) tree.extend(k8s_tree_instance.async_tree_node(namespace, pod))