Browse Source

Added device filter

Jeremy Stretch 9 years ago
parent
commit
b78caba1ca
1 changed files with 13 additions and 1 deletions
  1. 13 1
      netbox/secrets/filters.py

+ 13 - 1
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']