fix create image over api

coco-project · Aug 31, 2015 · 47453b7 · 47453b7
1 parent 981ca02
commit 47453b7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/ipynbsrv/api/serializer.py b/ipynbsrv/api/serializer.py
@@ -206,6 +206,17 @@ class Meta:
         model = ContainerImage
 
 
+class FlatContainerImageSerializer(serializers.ModelSerializer):
+    """
+    Todo: write doc.
+    """
+    friendly_name = serializers.CharField(read_only=True, source='get_friendly_name')
+    access_groups = FlatCollaborationGroupSerializer(many=True, read_only=True)
+
+    class Meta:
+        model = ContainerImage
+
+
 class ContainerSnapshotSerializer(serializers.ModelSerializer):
     """
     Todo: write doc.

diff --git a/ipynbsrv/api/views.py b/ipynbsrv/api/views.py
@@ -712,7 +712,11 @@ class ContainerImageList(generics.ListCreateAPIView):
     """
     Get a list of all the container images.
     """
-    serializer_class = ContainerImageSerializer
+
+    def get_serializer_class(self, *args, **kwargs):
+        if self.request.method in ['PATCH', 'POST', 'PUT']:
+            return FlatContainerImageSerializer
+        return ContainerImageSerializer
 
     def get_queryset(self):
         if self.request.user.is_superuser: