From b79aaff4a0d87d7cd8a6d03ea4e570fe3c2fa443 Mon Sep 17 00:00:00 2001 From: Eric Date: Mon, 12 Jun 2023 14:40:50 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E8=B5=84=E4=BA=A7=E6=A0=B9=E6=8D=AE=20?= =?UTF-8?q?ssh=20=E5=8D=8F=E8=AE=AE=E8=BF=87=E6=BB=A4=E6=98=AF=E5=90=A6?= =?UTF-8?q?=E5=90=AF=E7=94=A8=20sftp?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/assets/api/mixin.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/apps/assets/api/mixin.py b/apps/assets/api/mixin.py index 39d51197b..2cd928605 100644 --- a/apps/assets/api/mixin.py +++ b/apps/assets/api/mixin.py @@ -2,7 +2,7 @@ from typing import List from rest_framework.request import Request -from assets.models import Node, PlatformProtocol +from assets.models import Node, PlatformProtocol, Protocol from assets.utils import get_node_from_request, is_query_node_all_assets from common.utils import lazyproperty, timeit @@ -78,7 +78,10 @@ class SerializeToTreeNodeMixin: get_pid = lambda asset: getattr(asset, 'parent_key', '') else: get_pid = lambda asset: node_key - + ssh_asset_ids = [ + str(i) for i in + Protocol.objects.filter(name='ssh').values_list('asset_id', flat=True) + ] data = [ { 'id': str(asset.id), @@ -96,7 +99,8 @@ class SerializeToTreeNodeMixin: 'data': { 'platform_type': asset.platform.type, 'org_name': asset.org_name, - 'sftp': asset.platform_id in sftp_enabled_platform, + 'sftp': (asset.platform_id in sftp_enabled_platform) \ + and (str(asset.id) in ssh_asset_ids), 'name': asset.name, 'address': asset.address },