Updated Part API

- categories can now be filtered by parent
- Added FilterChildren func to invenTree.models

InvenTree/part/serializers.py

@@ -43,21 +43,13 @@ class PartCategoryBriefSerializer(serializers.ModelSerializer):
 
 class PartCategoryDetailSerializer(serializers.ModelSerializer):
 
-    # List of parts in this category
-    parts = PartSerializer(many=True, read_only=True)
-
-    # List of child categories under this one
-    children = PartCategoryBriefSerializer(many=True, read_only=True)
-
     class Meta:
         model = PartCategory
         fields = ('pk',
                   'name',
                   'description',
                   'parent',
-                  'path',
-                  'children',
-                  'parts')
+                  'path')
 
 
 class PartTemplateSerializer(serializers.ModelSerializer):

InvenTree/part/urls.py

@@ -10,10 +10,10 @@ from . import views
 categorypatterns = [
 
     # Part category detail
-    url(r'^category/(?P<pk>[0-9]+)/$', views.PartCategoryDetail.as_view()),
+    url(r'^(?P<pk>[0-9]+)/?$', views.PartCategoryDetail.as_view()),
 
     # List of top-level categories
-    url(r'^$', views.PartCategoryList.as_view())
+    url(r'^\?*[^/]*/?$', views.PartCategoryList.as_view())
 ]
 
 partparampatterns = [
@@ -47,13 +47,13 @@ urlpatterns = [
     url(r'^(?P<pk>[0-9]+)/$', views.PartDetail.as_view()),
 
     # Part categories
-    url(r'^category/', views.PartCategoryList.as_view()),
+    url(r'^category/?', include(categorypatterns)),
 
     # Part parameters
-    url(r'^parameters/', include(partparampatterns)),
+    url(r'^parameters/?', include(partparampatterns)),
 
     # Part templates
-    url(r'^templates/', include(parttemplatepatterns)),
+    url(r'^templates/?', include(parttemplatepatterns)),
 
     # List parts with optional filters
     url(r'^\?*[^/]*/?$', views.PartList.as_view()),

InvenTree/part/views.py

@@ -2,6 +2,7 @@
 
 from rest_framework import generics, permissions
 
+from InvenTree.models import FilterChildren
 from .models import PartCategory, Part, PartParameter, PartParameterTemplate
 from .serializers import PartSerializer
 from .serializers import PartCategoryDetailSerializer
@@ -91,6 +92,16 @@ class PartCategoryList(generics.ListCreateAPIView):
     """ Return a list of all top-level part categories.
     Categories are considered "top-level" if they do not have a parent
     """
+
+    def get_queryset(self):
+        params = self.request.query_params
+
+        categories = PartCategory.objects.all()
+
+        categories = FilterChildren(categories, params.get('parent', None))
+
+        return categories
+
     queryset = PartCategory.objects.filter(parent=None)
     serializer_class = PartCategoryDetailSerializer
     permission_classes = (permissions.IsAuthenticatedOrReadOnly,)