From e373a79d63b42146931fc6d381da94dea0e0ee51 Mon Sep 17 00:00:00 2001 From: feng <1304903146@qq.com> Date: Thu, 19 Sep 2024 10:44:46 +0800 Subject: [PATCH] perf: Gateway type asset filter --- apps/assets/api/asset/asset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/apps/assets/api/asset/asset.py b/apps/assets/api/asset/asset.py index bfac3ec1c..d3bcdb785 100644 --- a/apps/assets/api/asset/asset.py +++ b/apps/assets/api/asset/asset.py @@ -22,6 +22,7 @@ from common.drf.filters import BaseFilterSet, AttrRulesFilterBackend from common.utils import get_logger, is_uuid from orgs.mixins import generics from orgs.mixins.api import OrgBulkModelViewSet +from ...const import GATEWAY_NAME from ...notifications import BulkUpdatePlatformSkipAssetUserMsg logger = get_logger(__file__) @@ -71,6 +72,8 @@ class AssetFilterSet(BaseFilterSet): def filter_platform(queryset, name, value): if value.isdigit(): return queryset.filter(platform_id=value) + elif value == GATEWAY_NAME: + return queryset.filter(platform__name__istartswith=GATEWAY_NAME) else: return queryset.filter(platform__name=value)