--- /dev/null
+from rest_framework.decorators import api_view
+from rest_framework.response import Response\r
+from rest_framework.reverse import reverse\r
+from rest_framework import serializers\r
+from rest_framework import generics\r
+from core.models import *\r
+from django.forms import widgets
+from core.xoslib.objects.sliceplus import SlicePlus
+
+class SlicePlusIdSerializer(serializers.ModelSerializer):
+ id = serializers.Field()\r
+\r
+ sliceInfo = serializers.SerializerMethodField("getSliceInfo")\r
+\r
+ def getSliceInfo(self, slice):\r
+ return slice.getSliceInfo()\r
+\r
+ networks = serializers.HyperlinkedRelatedField(many=True, read_only=True, view_name='network-detail')\r
+ availableNetworks = serializers.HyperlinkedRelatedField(many=True, read_only=True, view_name='network-detail')\r
+\r
+ class Meta:\r
+ model = SlicePlus\r
+ fields = ('id','created','updated','enacted','name','enabled','omf_friendly','description','slice_url','site','max_slivers','imagePreference','service','network','mountDataSets','serviceClass','creator','networks','availableNetworks','sliceInfo')
+
+class SlicePlusList(generics.ListCreateAPIView):
+ queryset = SlicePlus.objects.select_related().all()
+ serializer_class = SlicePlusIdSerializer
+
+ def get_queryset(self):
+ return SlicePlus.select_by_user(self.request.user)
+
+ def create(self, request, *args, **kwargs):
+ obj = self.get_object()
+ obj.caller = request.user
+ if obj.can_update(request.user):
+ return super(SliceList, self).create(request, *args, **kwargs)
+ else:
+ return Response(status=status.HTTP_400_BAD_REQUEST)
+
+class SlicePlusDetail(generics.RetrieveUpdateDestroyAPIView):
+ queryset = SlicePlus.objects.select_related().all()
+ serializer_class = SlicePlusIdSerializer
+
+ def get_queryset(self):
+ return SlicePlus.select_by_user(self.request.user)
+
+ def update(self, request, *args, **kwargs):
+ obj = self.get_object()
+ if obj.can_update(request.user):
+ return super(SlicePlusDetail, self).update(request, *args, **kwargs)
+ else:
+ return Response(status=status.HTTP_400_BAD_REQUEST)
+
+ def destroy(self, request, *args, **kwargs):
+ obj = self.get_object()
+ if obj.can_update(request.user):
+ return super(SlicePlusDetail, self).destroy(request, *args, **kwargs)
+ else:
+ return Response(status=status.HTTP_400_BAD_REQUEST)
from core.models.slice import Slice
class SlicePlus(Slice):
+ class Meta:
+ proxy = True
+
def getSliceInfo(self, user=None):
- used_sites = []
- used_deployments = []
+ used_sites = {}
+ used_deployments = {}
sliverCount = 0
for sliver in self.slivers.all():
site = sliver.node.site
deployment = sliver.node.deployment
- used_sites[site.name] = used.sites.get(site.name, 0) + 1
+ used_sites[site.name] = used_sites.get(site.name, 0) + 1
used_deployments[deployment.name] = used_deployments.get(deployment.name, 0) + 1
+ sliverCount = sliverCount + 1
roles = []
if (user!=None):
return {"sitesUsed": used_sites,
"deploymentsUsed": used_deployments,
- "sliverCount": sliceCount,
+ "sliverCount": sliverCount,
"siteCount": len(used_sites.keys()),
"roles": roles}
+
+ @staticmethod
+ def select_by_user(user):
+ print "XX"
+ if user.is_admin:
+ qs = SlicePlus.objects.all()
+ else:
+ slice_ids = [sp.slice.id for sp in SlicePrivilege.objects.filter(user=user)]
+ qs = SlicePlus.objects.filter(id__in=slice_ids)
+ print qs
+ print qs.all()
+ return qs