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

perf: Gateway type asset filter

pull/14198/head
feng 2024-09-19 10:44:46 +08:00 committed by feng626
parent 744a5cd0e3
commit e373a79d63
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/assets/api/asset/asset.py
View File

@ -22,6 +22,7 @@ from common.drf.filters import BaseFilterSet, AttrRulesFilterBackend
from common.utils import get_logger, is_uuid from common.utils import get_logger, is_uuid
from orgs.mixins import generics from orgs.mixins import generics
from orgs.mixins.api import OrgBulkModelViewSet from orgs.mixins.api import OrgBulkModelViewSet
from ...const import GATEWAY_NAME
from ...notifications import BulkUpdatePlatformSkipAssetUserMsg from ...notifications import BulkUpdatePlatformSkipAssetUserMsg
logger = get_logger(__file__) logger = get_logger(__file__)
@ -71,6 +72,8 @@ class AssetFilterSet(BaseFilterSet):
def filter_platform(queryset, name, value): def filter_platform(queryset, name, value):
if value.isdigit(): if value.isdigit():
return queryset.filter(platform_id=value) return queryset.filter(platform_id=value)
elif value == GATEWAY_NAME:
return queryset.filter(platform__name__istartswith=GATEWAY_NAME)
else: else:
return queryset.filter(platform__name=value) return queryset.filter(platform__name=value)