planetstack/core/xoslib/methods/tenantview.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 rest_framework.views import APIView\r
   7 from core.models import *\r
   8 from django.forms import widgets
   9 from syndicate_storage.models import Volume
  10
  11 # This REST API endpoint contains a bunch of misc information that the
  12 # tenant view needs to display
  13
  14 BLESSED_DEPLOYMENTS = ["ViCCI"] # ["US-MaxPlanck", "US-GeorgiaTech", "US-Princeton", "US-Washington", "US-Stanford"]
  15
  16 def getTenantViewDict(user):
  17     blessed_sites = []
  18     for site in Site.objects.all():
  19         good=False
  20         for deployment in site.deployments.all():
  21             if deployment.name in BLESSED_DEPLOYMENTS:
  22                 good=True
  23         if good:
  24             blessed_sites.append(site)
  25
  26     blessed_images=[]
  27     for image in Image.objects.all():
  28         good = False
  29         for deployment in image.deployments.all():
  30             if deployment.name in BLESSED_DEPLOYMENTS:
  31                  good=True
  32         if good:
  33             blessed_images.append(image)
  34
  35     blessed_flavors=[]
  36     for flavor in Flavor.objects.all():
  37         good = False
  38         for deployment in flavor.deployments.all():
  39             if deployment.name in BLESSED_DEPLOYMENTS:
  40                  good=True
  41         if good:
  42             blessed_flavors.append(flavor)
  43
  44     volumes=[]
  45     for volume in Volume.objects.all():
  46         if not volume.private:\r
  47             volumes.append(volume)
  48
  49     blessed_service_classes = [ServiceClass.objects.get(name="Best Effort")]
  50
  51     return {"id": 0,
  52             "blessed_deployment_names": BLESSED_DEPLOYMENTS,
  53             "blessed_site_names": [site.name for site in blessed_sites],
  54             "blessed_sites": [site.id for site in blessed_sites],
  55             "blessed_image_names": [image.name for image in blessed_images],
  56             "blessed_images": [image.id for image in blessed_images],
  57             "blessed_flavor_names": [flavor.name for flavor in blessed_flavors],
  58             "blessed_flavors": [flavor.id for flavor in blessed_flavors],
  59             "blessed_service_class_names": [serviceclass.name for serviceclass in blessed_service_classes],
  60             "blessed_service_classes": [serviceclass.id for serviceclass in blessed_service_classes],
  61             "public_volume_names": [volume.name for volume in volumes],
  62             "public_volumes": [volume.id for volume in volumes],
  63             "current_user_site_id": user.site.id,
  64             "current_user_login_base": user.site.login_base,
  65             }
  66
  67 class TenantList(APIView):
  68     method_kind = "list"
  69     method_name = "tenantview"
  70
  71     def get(self, request, format=None):
  72         return Response( getTenantViewDict(request.user) )
  73
  74 class TenantDetail(APIView):
  75     method_kind = "detail"
  76     method_name = "tenantview"
  77
  78     def get(self, request, format=None, pk=0):
  79         return Response( [getTenantViewDict(request.user)] )
  80