blob: 0a8852ab5c5e2c37af8eae9a177b04dc2a846b22 [file] [log] [blame]
Scott Baker88e34372014-07-13 11:46:36 -07001from rest_framework.decorators import api_view
2from rest_framework.response import Response
3from rest_framework.reverse import reverse
4from rest_framework import serializers
5from rest_framework import generics
6from core.models import *
7from django.forms import widgets
8from core.xoslib.objects.sliceplus import SlicePlus
Scott Bakerbbc8a5b2015-01-12 12:57:22 -08009from plus import PlusSerializerMixin, PlusRetrieveUpdateDestroyAPIView, PlusListCreateAPIView
Scott Baker88e34372014-07-13 11:46:36 -070010
Scott Bakerc8af5752014-12-23 11:12:07 -080011if hasattr(serializers, "ReadOnlyField"):
12 # rest_framework 3.x
13 IdField = serializers.ReadOnlyField
14else:
15 # rest_framework 2.x
16 IdField = serializers.Field
17
Scott Baker8b89d302015-01-08 22:34:51 -080018class 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
Scott Baker97acad92015-01-12 19:45:40 -080025class 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
32class ListField(serializers.WritableField): # note: maybe just Field in rest_framework 3.x instead of WritableField
Scott Bakerdcf6fbf2015-01-11 13:45:19 -080033 def to_representation(self, obj):
34 return json.dumps(obj)
35
36 def to_internal_value(self, data):
37 return json.loads(data)
38
Scott Bakere791dc62014-08-28 14:02:54 -070039class SlicePlusIdSerializer(serializers.ModelSerializer, PlusSerializerMixin):
Scott Bakerc8af5752014-12-23 11:12:07 -080040 id = IdField()
Scott Baker88e34372014-07-13 11:46:36 -070041
42 sliceInfo = serializers.SerializerMethodField("getSliceInfo")
Scott Baker196ffef2014-11-21 11:32:24 -080043 humanReadableName = serializers.SerializerMethodField("getHumanReadableName")
Scott Bakerdcf6fbf2015-01-11 13:45:19 -080044 network_ports = NetworkPortsField(required=False)
Scott Baker97acad92015-01-12 19:45:40 -080045 site_allocation = DictionaryField(required=False)
Scott Baker21909342015-01-22 15:21:24 -080046 site_ready = DictionaryField(required=False)
Scott Baker97acad92015-01-12 19:45:40 -080047 users = ListField(required=False)
Scott Baker7a76f322015-01-16 19:07:36 -080048 user_names = ListField(required=False) # readonly = True ?
Scott Bakere0e407a2015-01-20 17:02:53 -080049 current_user_can_see = serializers.SerializerMethodField("getCurrentUserCanSee")
50
51 def getCurrentUserCanSee(self, slice):
52 # user can 'see' the slice if he is the creator or he has a role
53 current_user = self.context['request'].user
54 if (slice.creator and slice.creator==current_user):
55 return True;
56 return (len(slice.getSliceInfo(current_user)["roles"]) > 0)
Scott Baker88e34372014-07-13 11:46:36 -070057
58 def getSliceInfo(self, slice):
Scott Bakerdb236c32014-07-13 17:36:19 -070059 return slice.getSliceInfo(user=self.context['request'].user)
Scott Baker88e34372014-07-13 11:46:36 -070060
Scott Baker196ffef2014-11-21 11:32:24 -080061 def getHumanReadableName(self, obj):
62 return str(obj)
63
Scott Baker8b89d302015-01-08 22:34:51 -080064 networks = serializers.PrimaryKeyRelatedField(many=True, read_only=True)
Scott Baker88e34372014-07-13 11:46:36 -070065
66 class Meta:
67 model = SlicePlus
Scott Baker339d5122015-01-12 13:36:40 -080068 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',
Scott Baker21909342015-01-22 15:21:24 -080070 'serviceClass','creator','networks','sliceInfo','network_ports','backendIcon','backendHtml','site_allocation','site_ready','users',"user_names","current_user_can_see")
Scott Baker88e34372014-07-13 11:46:36 -070071
Scott Bakere0e407a2015-01-20 17:02:53 -080072class SlicePlusList(PlusListCreateAPIView):
Scott Baker88e34372014-07-13 11:46:36 -070073 queryset = SlicePlus.objects.select_related().all()
74 serializer_class = SlicePlusIdSerializer
75
Scott Bakerdb236c32014-07-13 17:36:19 -070076 method_kind = "list"
77 method_name = "slicesplus"
78
Scott Baker88e34372014-07-13 11:46:36 -070079 def get_queryset(self):
Scott Bakere0e407a2015-01-20 17:02:53 -080080 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
Scott Baker88e34372014-07-13 11:46:36 -070096
Scott Bakerdcf6fbf2015-01-11 13:45:19 -080097class SlicePlusDetail(PlusRetrieveUpdateDestroyAPIView):
Scott Baker88e34372014-07-13 11:46:36 -070098 queryset = SlicePlus.objects.select_related().all()
99 serializer_class = SlicePlusIdSerializer
100
Scott Bakerdb236c32014-07-13 17:36:19 -0700101 method_kind = "detail"
102 method_name = "slicesplus"
103
Scott Baker88e34372014-07-13 11:46:36 -0700104 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)