Improve API for Project app

InvenTree/project/models.py

@@ -1,5 +1,5 @@
 from __future__ import unicode_literals
-from django.utils.translation import ugettext as _
+# from django.utils.translation import ugettext as _
 
 from django.db import models
 
@@ -47,6 +47,13 @@ class ProjectPart(models.Model):
     The overage is the number of extra parts that are generally used for a single run.
     """
 
+    part = models.ForeignKey(Part, on_delete=models.CASCADE)
+    project = models.ForeignKey(Project, on_delete=models.CASCADE)
+    quantity = models.PositiveIntegerField(default=1)
+
+    """
+    # TODO - Add overage model fields
+
     # Overage types
     OVERAGE_PERCENT = 0
     OVERAGE_ABSOLUTE = 1
@@ -56,13 +63,11 @@ class ProjectPart(models.Model):
         OVERAGE_ABSOLUTE: _("Absolute")
     }
 
-    part = models.ForeignKey(Part, on_delete=models.CASCADE)
-    project = models.ForeignKey(Project, on_delete=models.CASCADE)
-    quantity = models.PositiveIntegerField(default=1)
     overage = models.FloatField(default=0)
     overage_type = models.PositiveIntegerField(
         default=OVERAGE_ABSOLUTE,
         choices=OVARAGE_CODES.items())
+    """
 
     # Set if the part is generated by the project,
     # rather than being consumed by the project

InvenTree/project/serializers.py

@@ -11,12 +11,10 @@ class ProjectPartSerializer(serializers.ModelSerializer):
                   'part',
                   'project',
                   'quantity',
-                  'overage',
-                  'overage_type',
                   'output')
 
 
-class ProjectBriefSerializer(serializers.ModelSerializer):
+class ProjectSerializer(serializers.ModelSerializer):
     """ Serializer for displaying brief overview of a project
     """
 
@@ -28,18 +26,6 @@ class ProjectBriefSerializer(serializers.ModelSerializer):
                   'category')
 
 
-class ProjectDetailSerializer(serializers.ModelSerializer):
-    """ Serializer for detailed project information
-    """
-
-    class Meta:
-        model = Project
-        fields = ('pk',
-                  'name',
-                  'description',
-                  'category')
-
-
 class ProjectCategoryBriefSerializer(serializers.ModelSerializer):
 
     class Meta:
@@ -49,9 +35,9 @@ class ProjectCategoryBriefSerializer(serializers.ModelSerializer):
 
 class ProjectCategoryDetailSerializer(serializers.ModelSerializer):
 
-    projects = ProjectBriefSerializer(many=True)
+    projects = ProjectSerializer(many=True, read_only=True)
 
-    children = ProjectCategoryBriefSerializer(many=True)
+    children = ProjectCategoryBriefSerializer(many=True, read_only=True)
 
     class Meta:
         model = ProjectCategory

InvenTree/project/urls.py

@@ -1,20 +1,39 @@
-from django.conf.urls import url
+from django.conf.urls import url, include
 
 from . import views
 
-urlpatterns = [
+""" URL patterns associated with project
+/project/<pk>       -> Detail view of single project
+/project/<pk>/parts -> Detail all parts associated with project
+"""
+projectdetailpatterns = [
     # Single project detail
-    url(r'^(?P<pk>[0-9]+)/$', views.ProjectDetail.as_view()),
+    url(r'^$', views.ProjectDetail.as_view()),
 
     # Parts associated with a project
-    url(r'^(?P<pk>[0-9]+)/parts$', views.ProjectPartsList.as_view()),
+    url(r'^parts/$', views.ProjectPartsList.as_view()),
+]
+
+projectcategorypatterns = [
+    # List of top-level project categories
+    url(r'^$', views.ProjectCategoryList.as_view()),
+
+    # Detail of a single project category
+    url(r'^(?P<pk>[0-9]+)/$', views.ProjectCategoryDetail.as_view()),
+
+    # Create a new category
+    url(r'^new/$', views.NewProjectCategory.as_view())
+
+]
+
+urlpatterns = [
+
+    # Individual project URL
+    url(r'^(?P<pk>[0-9]+)/', include(projectdetailpatterns)),
 
     # List of all projects
     url(r'^$', views.ProjectList.as_view()),
 
-    # List of top-level project categories
+    # Project categories
-    url(r'^category/$', views.ProjectCategoryList.as_view()),
+    url(r'^category/', include(projectcategorypatterns)),
-
-    # Detail of a single project category
-    url(r'^category/(?P<pk>[0-9]+)/$', views.ProjectCategoryDetail.as_view())
 ]

InvenTree/project/views.py

@@ -1,38 +1,50 @@
-from rest_framework import generics
+from rest_framework import generics, permissions
 
 from .models import ProjectCategory, Project, ProjectPart
-from .serializers import ProjectBriefSerializer, ProjectDetailSerializer
+from .serializers import ProjectSerializer
 from .serializers import ProjectCategoryDetailSerializer
 from .serializers import ProjectPartSerializer
 
 
-class ProjectDetail(generics.RetrieveAPIView):
+class ProjectDetail(generics.RetrieveUpdateAPIView):
 
     queryset = Project.objects.all()
-    serializer_class = ProjectDetailSerializer
+    serializer_class = ProjectSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
 
 
-class ProjectList(generics.ListAPIView):
+class ProjectList(generics.ListCreateAPIView):
 
     queryset = Project.objects.all()
-    serializer_class = ProjectBriefSerializer
+    serializer_class = ProjectSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
 
 
-class ProjectCategoryDetail(generics.RetrieveAPIView):
+class NewProjectCategory(generics.CreateAPIView):
+    """ Create a new Project Category
+    """
+    serializer_class = ProjectCategoryDetailSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
+
+
+class ProjectCategoryDetail(generics.RetrieveUpdateAPIView):
 
     queryset = ProjectCategory.objects.all()
     serializer_class = ProjectCategoryDetailSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
 
 
-class ProjectCategoryList(generics.ListAPIView):
+class ProjectCategoryList(generics.ListCreateAPIView):
 
     queryset = ProjectCategory.objects.filter(parent=None)
     serializer_class = ProjectCategoryDetailSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
 
 
-class ProjectPartsList(generics.ListAPIView):
+class ProjectPartsList(generics.ListCreateAPIView):
 
     serializer_class = ProjectPartSerializer
+    permission_classes = (permissions.IsAuthenticatedOrReadOnly,)
 
     def get_queryset(self):
         project_id = self.kwargs['pk']