From 631570b819bf1cc0ef3bfba5a2258b4385a31adb Mon Sep 17 00:00:00 2001 From: feng <1304903146@qq.com> Date: Tue, 15 Apr 2025 15:51:46 +0800 Subject: [PATCH] perf: Asset filter --- apps/assets/api/asset/asset.py | 3 ++- apps/assets/models/asset/common.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/apps/assets/api/asset/asset.py b/apps/assets/api/asset/asset.py index c9b5bd524..efb9683da 100644 --- a/apps/assets/api/asset/asset.py +++ b/apps/assets/api/asset/asset.py @@ -173,7 +173,8 @@ class AssetViewSet(SuggestionMixin, BaseAssetViewSet): @action(methods=["GET"], detail=True, url_path="accounts") def accounts(self, *args, **kwargs): - asset = super().get_object() + pk = self.kwargs.get("pk") + asset = get_object_or_404(self.model, pk=pk) queryset = asset.all_accounts.all() return self.get_paginated_response_from_queryset(queryset) diff --git a/apps/assets/models/asset/common.py b/apps/assets/models/asset/common.py index d59e86509..e38afb8d9 100644 --- a/apps/assets/models/asset/common.py +++ b/apps/assets/models/asset/common.py @@ -261,7 +261,7 @@ class Asset(NodesRelationMixin, LabeledMixin, AbsConnectivity, JSONFilterMixin, @lazyproperty def accounts_amount(self): - return self.accounts.count() + return self.all_accounts.count() def get_target_ip(self): return self.address