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 if hasattr(serializers, "ReadOnlyField"):
  12     # rest_framework 3.x
  13     IdField = serializers.ReadOnlyField
  14 else:
  15     # rest_framework 2.x
  16     IdField = serializers.Field
  17
  18 class NetworkPortsField(serializers.WritableField):   # note: maybe just Field in rest_framework 3.x instead of WritableField
  19     def to_representation(self, obj):
  20         return obj
  21
  22     def to_internal_value(self, data):
  23         return data
  24
  25 class SlicePlusIdSerializer(serializers.ModelSerializer, PlusSerializerMixin):
  26         id = IdField()
  27 \r
  28         sliceInfo = serializers.SerializerMethodField("getSliceInfo")\r
  29         humanReadableName = serializers.SerializerMethodField("getHumanReadableName")\r
  30         networkPorts = NetworkPortsField()\r
  31 \r
  32         def getSliceInfo(self, slice):\r
  33             return slice.getSliceInfo(user=self.context['request'].user)\r
  34 \r
  35         def getHumanReadableName(self, obj):\r
  36             return str(obj)\r
  37 \r
  38         networks = serializers.PrimaryKeyRelatedField(many=True, read_only=True)\r
  39 #        availableNetworks = serializers.PrimaryKeyRelatedField(many=True, read_only=True, view_name='network-detail')\r
  40 \r
  41         class Meta:\r
  42             model = SlicePlus\r
  43             fields = ('humanReadableName', 'id','created','updated','enacted','name','enabled','omf_friendly','description','slice_url','site','max_slivers','image_preference','service','network','mount_data_sets','serviceClass','creator','networks','sliceInfo','networkPorts','backendIcon','backendHtml')
  44
  45 class SlicePlusList(generics.ListCreateAPIView):
  46     queryset = SlicePlus.objects.select_related().all()
  47     serializer_class = SlicePlusIdSerializer
  48
  49     method_kind = "list"
  50     method_name = "slicesplus"
  51
  52     def get_queryset(self):
  53         return SlicePlus.select_by_user(self.request.user)
  54
  55     def create(self, request, *args, **kwargs):
  56         obj = self.get_object()
  57         obj.caller = request.user
  58         if obj.can_update(request.user):
  59             return super(SliceList, self).create(request, *args, **kwargs)
  60         else:
  61             return Response(status=status.HTTP_400_BAD_REQUEST)
  62
  63 class SlicePlusDetail(generics.RetrieveUpdateDestroyAPIView):
  64     queryset = SlicePlus.objects.select_related().all()
  65     serializer_class = SlicePlusIdSerializer
  66
  67     method_kind = "detail"
  68     method_name = "slicesplus"
  69
  70     def get_queryset(self):
  71         return SlicePlus.select_by_user(self.request.user)
  72
  73     def update(self, request, *args, **kwargs):
  74         obj = self.get_object()
  75         if obj.can_update(request.user):
  76             return super(SlicePlusDetail, self).update(request, *args, **kwargs)
  77         else:
  78             return Response(status=status.HTTP_400_BAD_REQUEST)
  79
  80     def destroy(self, request, *args, **kwargs):
  81         obj = self.get_object()
  82         if obj.can_update(request.user):
  83             return super(SlicePlusDetail, self).destroy(request, *args, **kwargs)
  84         else:
  85             return Response(status=status.HTTP_400_BAD_REQUEST)