sliceplus support for site_allocation and network_ports for tenant view

[plstackapi.git] / planetstack / core / xoslib / methods / plus.py

diff --git a/planetstack/core/xoslib/methods/plus.py b/planetstack/core/xoslib/methods/plus.py
index 9ace688..ca89c79 100644 (file)
--- a/planetstack/core/xoslib/methods/plus.py
+++ b/planetstack/core/xoslib/methods/plus.py
@@ -1,4 +1,7 @@
+from rest_framework import generics
 from rest_framework import serializers
+from rest_framework.response import Response
+from rest_framework import status
 
 """ PlusSerializerMixin
 
@@ -22,3 +25,53 @@ class PlusSerializerMixin():
     def getBackendHtml(self, obj):
         return obj.getBackendHtml()
 
+# XXX this is taken from genapi.py
+# XXX find a better way to re-use the code
+class PlusRetrieveUpdateDestroyAPIView(generics.RetrieveUpdateDestroyAPIView):
+
+    # To handle fine-grained field permissions, we have to check can_update
+    # the object has been updated but before it has been saved.
+
+    def update(self, request, *args, **kwargs):\r
+        partial = kwargs.pop('partial', False)\r
+        self.object = self.get_object_or_none()\r
+\r
+        serializer = self.get_serializer(self.object, data=request.DATA,\r
+                                         files=request.FILES, partial=partial)\r
+\r
+        if not serializer.is_valid():\r
+            response = {"error": "validation",\r
+                        "specific_error": "not serializer.is_valid()",\r
+                        "reasons": serializer.errors}\r
+            return Response(response, status=status.HTTP_400_BAD_REQUEST)\r
+\r
+        try:\r
+            self.pre_save(serializer.object)\r
+        except ValidationError as err:\r
+            # full_clean on model instance may be called in pre_save,\r
+            # so we have to handle eventual errors.\r
+            response = {"error": "validation",\r
+                         "specific_error": "ValidationError in pre_save",\r
+                         "reasons": err.message_dict}\r
+            return Response(response, status=status.HTTP_400_BAD_REQUEST)\r
+\r
+        if serializer.object is not None:\r
+            if not serializer.object.can_update(request.user):\r
+                return Response(status=status.HTTP_400_BAD_REQUEST)\r
+\r
+        if self.object is None:\r
+            self.object = serializer.save(force_insert=True)\r
+            self.post_save(self.object, created=True)\r
+            return Response(serializer.data, status=status.HTTP_201_CREATED)\r
+\r
+        self.object = serializer.save(force_update=True)\r
+        self.post_save(self.object, created=False)\r
+        return Response(serializer.data, status=status.HTTP_200_OK)
+
+    def destroy(self, request, *args, **kwargs):
+        obj = self.get_object()
+        if obj.can_update(request.user):
+            return super(generics.RetrieveUpdateDestroyAPIView, self).destroy(request, *args, **kwargs)
+        else:
+            return Response(status=status.HTTP_400_BAD_REQUEST)
+