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, PlusRetrieveUpdateDestroyAPIView, PlusListCreateAPIView
  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 DictionaryField(serializers.WritableField):   # note: maybe just Field in rest_framework 3.x instead of WritableField
  26     def to_representation(self, obj):
  27         return json.dumps(obj)
  28
  29     def to_internal_value(self, data):
  30         return json.loads(data)
  31
  32 class ListField(serializers.WritableField):   # note: maybe just Field in rest_framework 3.x instead of WritableField
  33     def to_representation(self, obj):
  34         return json.dumps(obj)
  35
  36     def to_internal_value(self, data):
  37         return json.loads(data)
  38
  39 class SlicePlusIdSerializer(serializers.ModelSerializer, PlusSerializerMixin):
  40         id = IdField()
  41 \r
  42         sliceInfo = serializers.SerializerMethodField("getSliceInfo")\r
  43         humanReadableName = serializers.SerializerMethodField("getHumanReadableName")\r
  44         network_ports = NetworkPortsField(required=False)\r
  45         site_allocation = DictionaryField(required=False)\r
  46         site_ready = DictionaryField(required=False)\r
  47         users = ListField(required=False)\r
  48         user_names = ListField(required=False) # readonly = True ?\r
  49         current_user_can_see = serializers.SerializerMethodField("getCurrentUserCanSee")\r
  50 \r
  51         def getCurrentUserCanSee(self, slice):\r
  52             # user can 'see' the slice if he is the creator or he has a role\r
  53             current_user = self.context['request'].user\r
  54             if (slice.creator and slice.creator==current_user):\r
  55                 return True;\r
  56             return (len(slice.getSliceInfo(current_user)["roles"]) > 0)\r
  57 \r
  58         def getSliceInfo(self, slice):\r
  59             return slice.getSliceInfo(user=self.context['request'].user)\r
  60 \r
  61         def getHumanReadableName(self, obj):\r
  62             return str(obj)\r
  63 \r
  64         networks = serializers.PrimaryKeyRelatedField(many=True, read_only=True)\r
  65 \r
  66         class Meta:\r
  67             model = SlicePlus\r
  68             fields = ('humanReadableName', 'id','created','updated','enacted','name','enabled','omf_friendly','description','slice_url','site','max_slivers','service','network','mount_data_sets',
  69                       'default_image', 'default_flavor',
  70                       'serviceClass','creator','networks','sliceInfo','network_ports','backendIcon','backendHtml','site_allocation','site_ready','users',"user_names","current_user_can_see")
  71
  72 class SlicePlusList(PlusListCreateAPIView):
  73     queryset = SlicePlus.objects.select_related().all()
  74     serializer_class = SlicePlusIdSerializer
  75
  76     method_kind = "list"
  77     method_name = "slicesplus"
  78
  79     def get_queryset(self):
  80         current_user_can_see = self.request.QUERY_PARAMS.get('current_user_can_see', False)
  81
  82         slices = SlicePlus.select_by_user(self.request.user)
  83
  84         # If current_user_can_see is set, then filter the queryset to return
  85         # only those slices that the user is either creator or has privilege
  86         # on.
  87         if (current_user_can_see):
  88             slice_ids = []
  89             for slice in slices:
  90                 if (self.request.user == slice.creator) or (len(slice.getSliceInfo(self.request.user)["roles"]) > 0):
  91                     slice_ids.append(slice.id)
  92
  93             slices = SlicePlus.objects.filter(id__in=slice_ids)
  94
  95         return slices
  96
  97 class SlicePlusDetail(PlusRetrieveUpdateDestroyAPIView):
  98     queryset = SlicePlus.objects.select_related().all()
  99     serializer_class = SlicePlusIdSerializer
 100
 101     method_kind = "detail"
 102     method_name = "slicesplus"
 103
 104     def get_queryset(self):
 105         return SlicePlus.select_by_user(self.request.user)
 106
 107     def update(self, request, *args, **kwargs):
 108         obj = self.get_object()
 109         if obj.can_update(request.user):
 110             return super(SlicePlusDetail, self).update(request, *args, **kwargs)
 111         else:
 112             return Response(status=status.HTTP_400_BAD_REQUEST)
 113
 114     def destroy(self, request, *args, **kwargs):
 115         obj = self.get_object()
 116         if obj.can_update(request.user):
 117             return super(SlicePlusDetail, self).destroy(request, *args, **kwargs)
 118         else:
 119             return Response(status=status.HTTP_400_BAD_REQUEST)