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
11 if hasattr(serializers, "ReadOnlyField"):
13 IdField = serializers.ReadOnlyField
16 IdField = serializers.Field
18 class NetworkPortsField(serializers.WritableField): # note: maybe just Field in rest_framework 3.x instead of WritableField
19 def to_representation(self, obj):
22 def to_internal_value(self, data):
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)
29 def to_internal_value(self, data):
30 return json.loads(data)
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)
36 def to_internal_value(self, data):
37 return json.loads(data)
39 class SlicePlusIdSerializer(serializers.ModelSerializer, PlusSerializerMixin):
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
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
56 return (len(slice.getSliceInfo(current_user)["roles"]) > 0)
\r
58 def getSliceInfo(self, slice):
\r
59 return slice.getSliceInfo(user=self.context['request'].user)
\r
61 def getHumanReadableName(self, obj):
\r
64 networks = serializers.PrimaryKeyRelatedField(many=True, read_only=True)
\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")
72 class SlicePlusList(PlusListCreateAPIView):
73 queryset = SlicePlus.objects.select_related().all()
74 serializer_class = SlicePlusIdSerializer
77 method_name = "slicesplus"
79 def get_queryset(self):
80 current_user_can_see = self.request.QUERY_PARAMS.get('current_user_can_see', False)
82 slices = SlicePlus.select_by_user(self.request.user)
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
87 if (current_user_can_see):
90 if (self.request.user == slice.creator) or (len(slice.getSliceInfo(self.request.user)["roles"]) > 0):
91 slice_ids.append(slice.id)
93 slices = SlicePlus.objects.filter(id__in=slice_ids)
97 class SlicePlusDetail(PlusRetrieveUpdateDestroyAPIView):
98 queryset = SlicePlus.objects.select_related().all()
99 serializer_class = SlicePlusIdSerializer
101 method_kind = "detail"
102 method_name = "slicesplus"
104 def get_queryset(self):
105 return SlicePlus.select_by_user(self.request.user)
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)
112 return Response(status=status.HTTP_400_BAD_REQUEST)
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)
119 return Response(status=status.HTTP_400_BAD_REQUEST)