From 500124f4858a13f87c2ed317da77ecd0daf2cf31 Mon Sep 17 00:00:00 2001 From: fit2bot <68588906+fit2bot@users.noreply.github.com> Date: Mon, 6 Feb 2023 13:37:03 +0800 Subject: [PATCH] perf: asset automation filter (#9440) Co-authored-by: feng <1304903146@qq.com> --- apps/assets/api/asset/asset.py | 25 +++++++++++++++++++++++-- 1 file changed, 23 insertions(+), 2 deletions(-) diff --git a/apps/assets/api/asset/asset.py b/apps/assets/api/asset/asset.py index 1838a4f1a..13f4b86ac 100644 --- a/apps/assets/api/asset/asset.py +++ b/apps/assets/api/asset/asset.py @@ -29,10 +29,31 @@ __all__ = [ class AssetFilterSet(BaseFilterSet): + labels = django_filters.CharFilter(method='filter_labels') + platform = django_filters.CharFilter(method='filter_platform') type = django_filters.CharFilter(field_name="platform__type", lookup_expr="exact") category = django_filters.CharFilter(field_name="platform__category", lookup_expr="exact") - platform = django_filters.CharFilter(method='filter_platform') - labels = django_filters.CharFilter(method='filter_labels') + domain_enabled = django_filters.BooleanFilter( + field_name="platform__domain_enabled", lookup_expr="exact" + ) + ping_enabled = django_filters.BooleanFilter( + field_name="platform__automation__ping_enabled", lookup_expr="exact" + ) + gather_facts_enabled = django_filters.BooleanFilter( + field_name="platform__automation__gather_facts_enabled", lookup_expr="exact" + ) + change_secret_enabled = django_filters.BooleanFilter( + field_name="platform__automation__change_secret_enabled", lookup_expr="exact" + ) + push_account_enabled = django_filters.BooleanFilter( + field_name="platform__automation__push_account_enabled", lookup_expr="exact" + ) + verify_account_enabled = django_filters.BooleanFilter( + field_name="platform__automation__verify_account_enabled", lookup_expr="exact" + ) + gather_accounts_enabled = django_filters.BooleanFilter( + field_name="platform__automation__gather_accounts_enabled", lookup_expr="exact" + ) class Meta: model = Asset