Further API improvements

InvenTree/part/views.py

@@ -1,3 +1,5 @@
+# import django_filters
+
 from rest_framework import generics, permissions
 
 from .models import PartCategory, Part, PartParameter, PartParameterTemplate
@@ -46,9 +48,33 @@ class PartParamDetail(generics.RetrieveUpdateDestroyAPIView):
     permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
 
 
-class PartList(generics.ListCreateAPIView):
+"""
+class PartFilter(django_filters.rest_framework.FilterSet):
+    min_stock = django_filters.NumberFilter(name="stock", lookup_expr="gte")
+    max_stock = django_filters.NumberFilter(name="stock", lookup_expr="lte")
+
+    class Meta:
+        model = Part
+        fields = ['stock']
+"""
+
+
+class PartList(generics.ListCreateAPIView):
+    """ Display a list of parts, with optional filters
+    Filters are specified in the url, e.g.
+    /part/?category=127
+    /part/?min_stock=100
+    """
+
+    def get_queryset(self):
+        parts = Part.objects.all()
+
+        cat_id = self.request.query_params.get('category', None)
+        if cat_id:
+            parts = parts.filter(category=cat_id)
+
+        return parts
 
-    queryset = Part.objects.all()
     serializer_class = PartSerializer
     permission_classes = (permissions.IsAuthenticatedOrReadOnly,)