Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [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 rest_framework.views import APIView
|
| 7 | from core.models import *
|
| 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 | |
Scott Baker | c94091a | 2015-01-12 12:57:44 -0800 | [diff] [blame] | 16 | def getTenantViewDict(user): |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 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 | |
Scott Baker | 2a11f25 | 2015-01-12 13:37:01 -0800 | [diff] [blame] | 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 | |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 44 | volumes=[] |
| 45 | for volume in Volume.objects.all(): |
| 46 | if not volume.private:
|
| 47 | volumes.append(volume) |
| 48 | |
Scott Baker | 2a11f25 | 2015-01-12 13:37:01 -0800 | [diff] [blame] | 49 | blessed_service_classes = [ServiceClass.objects.get(name="Best Effort")] |
| 50 | |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 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], |
Scott Baker | 2a11f25 | 2015-01-12 13:37:01 -0800 | [diff] [blame] | 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], |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 61 | "public_volume_names": [volume.name for volume in volumes], |
| 62 | "public_volumes": [volume.id for volume in volumes], |
Scott Baker | c94091a | 2015-01-12 12:57:44 -0800 | [diff] [blame] | 63 | "current_user_site_id": user.site.id, |
| 64 | "current_user_login_base": user.site.login_base, |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 65 | } |
| 66 | |
| 67 | class TenantList(APIView): |
| 68 | method_kind = "list" |
| 69 | method_name = "tenantview" |
| 70 | |
| 71 | def get(self, request, format=None): |
Scott Baker | c94091a | 2015-01-12 12:57:44 -0800 | [diff] [blame] | 72 | return Response( getTenantViewDict(request.user) ) |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 73 | |
| 74 | class TenantDetail(APIView): |
| 75 | method_kind = "detail" |
| 76 | method_name = "tenantview" |
| 77 | |
| 78 | def get(self, request, format=None, pk=0): |
Scott Baker | c94091a | 2015-01-12 12:57:44 -0800 | [diff] [blame] | 79 | return Response( [getTenantViewDict(request.user)] ) |
Scott Baker | 0332483 | 2015-01-08 22:38:17 -0800 | [diff] [blame] | 80 | |