planetstack/core/xoslib/methods/sliceplus.py

   1 from rest_framework.decorators import api_view
   2 from rest_framework.response import Response\r
   3 from rest_framework.reverse import reverse\r
   4 from rest_framework import serializers\r
   5 from rest_framework import generics\r
   6 from core.models import *\r
   7 from django.forms import widgets
   8 from core.xoslib.objects.sliceplus import SlicePlus
   9 from plus import PlusSerializerMixin
  10
  11 class SlicePlusIdSerializer(serializers.ModelSerializer, PlusSerializerMixin):
  12         id = serializers.Field()\r
  13 \r
  14         sliceInfo = serializers.SerializerMethodField("getSliceInfo")\r
  15 \r
  16         def getSliceInfo(self, slice):\r
  17             return slice.getSliceInfo(user=self.context['request'].user)\r
  18 \r
  19         networks = serializers.HyperlinkedRelatedField(many=True, read_only=True, view_name='network-detail')\r
  20         availableNetworks = serializers.HyperlinkedRelatedField(many=True, read_only=True, view_name='network-detail')\r
  21 \r
  22         class Meta:\r
  23             model = SlicePlus\r
  24             fields = ('id','created','updated','enacted','name','enabled','omf_friendly','description','slice_url','site','max_slivers','imagePreference','service','network','mountDataSets','serviceClass','creator','networks','availableNetworks','sliceInfo','backendIcon','backendHtml')
  25
  26 class SlicePlusList(generics.ListCreateAPIView):
  27     queryset = SlicePlus.objects.select_related().all()
  28     serializer_class = SlicePlusIdSerializer
  29
  30     method_kind = "list"
  31     method_name = "slicesplus"
  32
  33     def get_queryset(self):
  34         return SlicePlus.select_by_user(self.request.user)
  35
  36     def create(self, request, *args, **kwargs):
  37         obj = self.get_object()
  38         obj.caller = request.user
  39         if obj.can_update(request.user):
  40             return super(SliceList, self).create(request, *args, **kwargs)
  41         else:
  42             return Response(status=status.HTTP_400_BAD_REQUEST)
  43
  44 class SlicePlusDetail(generics.RetrieveUpdateDestroyAPIView):
  45     queryset = SlicePlus.objects.select_related().all()
  46     serializer_class = SlicePlusIdSerializer
  47
  48     method_kind = "detail"
  49     method_name = "slicesplus"
  50
  51     def get_queryset(self):
  52         return SlicePlus.select_by_user(self.request.user)
  53
  54     def update(self, request, *args, **kwargs):
  55         obj = self.get_object()
  56         if obj.can_update(request.user):
  57             return super(SlicePlusDetail, self).update(request, *args, **kwargs)
  58         else:
  59             return Response(status=status.HTTP_400_BAD_REQUEST)
  60
  61     def destroy(self, request, *args, **kwargs):
  62         obj = self.get_object()
  63         if obj.can_update(request.user):
  64             return super(SlicePlusDetail, self).destroy(request, *args, **kwargs)
  65         else:
  66             return Response(status=status.HTTP_400_BAD_REQUEST)