Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 1 | from rest_framework.decorators import api_view |
| 2 | from rest_framework.response import Response
|
| 3 | from rest_framework.reverse import reverse
|
| 4 | from rest_framework import serializers
|
| 5 | from rest_framework import generics
|
| 6 | from core.models import *
|
| 7 | from django.forms import widgets |
| 8 | from core.xoslib.objects.sliceplus import SlicePlus |
Scott Baker | bbc8a5b | 2015-01-12 12:57:22 -0800 | [diff] [blame] | 9 | from plus import PlusSerializerMixin, PlusRetrieveUpdateDestroyAPIView, PlusListCreateAPIView |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 10 | |
Scott Baker | c8af575 | 2014-12-23 11:12:07 -0800 | [diff] [blame] | 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 | |
Scott Baker | 8b89d30 | 2015-01-08 22:34:51 -0800 | [diff] [blame] | 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 | |
Scott Baker | dcf6fbf | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 25 | class SiteAllocationField(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 | |
Scott Baker | e791dc6 | 2014-08-28 14:02:54 -0700 | [diff] [blame] | 32 | class SlicePlusIdSerializer(serializers.ModelSerializer, PlusSerializerMixin): |
Scott Baker | c8af575 | 2014-12-23 11:12:07 -0800 | [diff] [blame] | 33 | id = IdField() |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 34 |
|
| 35 | sliceInfo = serializers.SerializerMethodField("getSliceInfo")
|
Scott Baker | 196ffef | 2014-11-21 11:32:24 -0800 | [diff] [blame] | 36 | humanReadableName = serializers.SerializerMethodField("getHumanReadableName")
|
Scott Baker | dcf6fbf | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 37 | network_ports = NetworkPortsField(required=False)
|
| 38 | site_allocation = SiteAllocationField(required=False)
|
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 39 |
|
| 40 | def getSliceInfo(self, slice):
|
Scott Baker | db236c3 | 2014-07-13 17:36:19 -0700 | [diff] [blame] | 41 | return slice.getSliceInfo(user=self.context['request'].user)
|
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 42 |
|
Scott Baker | 196ffef | 2014-11-21 11:32:24 -0800 | [diff] [blame] | 43 | def getHumanReadableName(self, obj):
|
| 44 | return str(obj)
|
| 45 |
|
Scott Baker | 8b89d30 | 2015-01-08 22:34:51 -0800 | [diff] [blame] | 46 | networks = serializers.PrimaryKeyRelatedField(many=True, read_only=True)
|
| 47 | # availableNetworks = serializers.PrimaryKeyRelatedField(many=True, read_only=True, view_name='network-detail')
|
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 48 |
|
| 49 | class Meta:
|
| 50 | model = SlicePlus
|
Scott Baker | dcf6fbf | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 51 | fields = ('humanReadableName', 'id','created','updated','enacted','name','enabled','omf_friendly','description','slice_url','site','max_slivers','image_preference','service','network','mount_data_sets', |
| 52 | 'serviceClass','creator','networks','sliceInfo','network_ports','backendIcon','backendHtml','site_allocation') |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 53 | |
Scott Baker | bbc8a5b | 2015-01-12 12:57:22 -0800 | [diff] [blame] | 54 | class SlicePlusList(PlusListCreateAPIView): #generics.ListCreateAPIView): |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 55 | queryset = SlicePlus.objects.select_related().all() |
| 56 | serializer_class = SlicePlusIdSerializer |
| 57 | |
Scott Baker | db236c3 | 2014-07-13 17:36:19 -0700 | [diff] [blame] | 58 | method_kind = "list" |
| 59 | method_name = "slicesplus" |
| 60 | |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 61 | def get_queryset(self): |
| 62 | return SlicePlus.select_by_user(self.request.user) |
| 63 | |
Scott Baker | dcf6fbf | 2015-01-11 13:45:19 -0800 | [diff] [blame] | 64 | class SlicePlusDetail(PlusRetrieveUpdateDestroyAPIView): |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 65 | queryset = SlicePlus.objects.select_related().all() |
| 66 | serializer_class = SlicePlusIdSerializer |
| 67 | |
Scott Baker | db236c3 | 2014-07-13 17:36:19 -0700 | [diff] [blame] | 68 | method_kind = "detail" |
| 69 | method_name = "slicesplus" |
| 70 | |
Scott Baker | 88e3437 | 2014-07-13 11:46:36 -0700 | [diff] [blame] | 71 | def get_queryset(self): |
| 72 | return SlicePlus.select_by_user(self.request.user) |
| 73 | |
| 74 | def update(self, request, *args, **kwargs): |
| 75 | obj = self.get_object() |
| 76 | if obj.can_update(request.user): |
| 77 | return super(SlicePlusDetail, self).update(request, *args, **kwargs) |
| 78 | else: |
| 79 | return Response(status=status.HTTP_400_BAD_REQUEST) |
| 80 | |
| 81 | def destroy(self, request, *args, **kwargs): |
| 82 | obj = self.get_object() |
| 83 | if obj.can_update(request.user): |
| 84 | return super(SlicePlusDetail, self).destroy(request, *args, **kwargs) |
| 85 | else: |
| 86 | return Response(status=status.HTTP_400_BAD_REQUEST) |