diff --git a/netbox/secrets/filters.py b/netbox/secrets/filters.py index 0773562e6..5c8a770bc 100644 --- a/netbox/secrets/filters.py +++ b/netbox/secrets/filters.py @@ -1,9 +1,21 @@ import django_filters +from dcim.models import Device from .models import Secret, SecretRole class SecretFilter(django_filters.FilterSet): + device_id = django_filters.ModelMultipleChoiceFilter( + name='device', + queryset=Device.objects.all(), + label='Device (ID)', + ) + device = django_filters.ModelMultipleChoiceFilter( + name='device', + queryset=Device.objects.all(), + to_field_name='name', + label='Device (name)', + ) role_id = django_filters.ModelMultipleChoiceFilter( name='role', queryset=SecretRole.objects.all(), @@ -18,4 +30,4 @@ class SecretFilter(django_filters.FilterSet): class Meta: model = Secret - fields = ['name', 'role_id', 'role'] + fields = ['name', 'device_id', 'device', 'role_id', 'role']