Parcourir la source

Closes #1910: Added filters for cluter group and cluster type

Jeremy Stretch il y a 7 ans
Parent
commit
6b62720daf
2 fichiers modifiés avec 16 ajouts et 0 suppressions
  1. 2 0
      netbox/virtualization/api/views.py
  2. 14 0
      netbox/virtualization/filters.py

+ 2 - 0
netbox/virtualization/api/views.py

@@ -25,11 +25,13 @@ class VirtualizationFieldChoicesViewSet(FieldChoicesViewSet):
 class ClusterTypeViewSet(ModelViewSet):
     queryset = ClusterType.objects.all()
     serializer_class = serializers.ClusterTypeSerializer
+    filter_class = filters.ClusterTypeFilter
 
 
 class ClusterGroupViewSet(ModelViewSet):
     queryset = ClusterGroup.objects.all()
     serializer_class = serializers.ClusterGroupSerializer
+    filter_class = filters.ClusterGroupFilter
 
 
 class ClusterViewSet(CustomFieldModelViewSet):

+ 14 - 0
netbox/virtualization/filters.py

@@ -13,6 +13,20 @@ from .constants import VM_STATUS_CHOICES
 from .models import Cluster, ClusterGroup, ClusterType, VirtualMachine
 
 
+class ClusterTypeFilter(django_filters.FilterSet):
+
+    class Meta:
+        model = ClusterType
+        fields = ['name', 'slug']
+
+
+class ClusterGroupFilter(django_filters.FilterSet):
+
+    class Meta:
+        model = ClusterGroup
+        fields = ['name', 'slug']
+
+
 class ClusterFilter(CustomFieldFilterSet):
     id__in = NumericInFilter(name='id', lookup_expr='in')
     q = django_filters.CharFilter(