github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-07-05 02:56:31 +00:00
Code Issues Projects Releases Wiki Activity

perf: gather account node_id filter

Browse Source
This commit is contained in:
feng 2025-03-12 18:04:22 +08:00 committed by feng626
parent 94a8122eac
commit 042c3d1ba8
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
apps/accounts/api/automations/gather_account.py
View File

@ -9,7 +9,7 @@ from rest_framework.response import Response
from accounts import serializers from accounts import serializers
from accounts.const import AutomationTypes from accounts.const import AutomationTypes
from accounts.filters import GatheredAccountFilterSet from accounts.filters import GatheredAccountFilterSet, NodeFilterBackend
from accounts.models import GatherAccountsAutomation, AutomationExecution, Account from accounts.models import GatherAccountsAutomation, AutomationExecution, Account
from accounts.models import GatheredAccount from accounts.models import GatheredAccount
from assets.models import Asset from assets.models import Asset
@ -76,6 +76,7 @@ class GatheredAccountViewSet(OrgBulkModelViewSet):
model = GatheredAccount model = GatheredAccount
search_fields = ("username",) search_fields = ("username",)
filterset_class = GatheredAccountFilterSet filterset_class = GatheredAccountFilterSet
extra_filter_backends = [NodeFilterBackend]
ordering = ("status",) ordering = ("status",)
serializer_classes = { serializer_classes = {
"default": serializers.DiscoverAccountSerializer, "default": serializers.DiscoverAccountSerializer,

5
apps/accounts/filters.py
View File

@ -143,17 +143,12 @@ class AccountFilterSet(BaseFilterSet):
class GatheredAccountFilterSet(BaseFilterSet): class GatheredAccountFilterSet(BaseFilterSet):
node_id = drf_filters.CharFilter(method="filter_nodes")
asset_id = drf_filters.CharFilter(field_name="asset_id", lookup_expr="exact") asset_id = drf_filters.CharFilter(field_name="asset_id", lookup_expr="exact")
asset_name = drf_filters.CharFilter( asset_name = drf_filters.CharFilter(
field_name="asset__name", lookup_expr="icontains" field_name="asset__name", lookup_expr="icontains"
) )
status = drf_filters.CharFilter(field_name="status", lookup_expr="exact") status = drf_filters.CharFilter(field_name="status", lookup_expr="exact")
@staticmethod
def filter_nodes(queryset, name, value):
return AccountFilterSet.filter_nodes(queryset, name, value)
class Meta: class Meta:
model = GatheredAccount model = GatheredAccount
fields = ["id", "username"] fields = ["id", "username"]