blob: f57f587aa35aac3d8686a77075283601e118e666 [file] [log] [blame]
Siobhan Tullycf04fb62014-01-11 11:25:57 -05001#views.py
Scott Bakerbd57a432014-04-14 16:12:15 -07002import functools
3import math
Scott Baker771819b2014-03-19 22:10:17 -07004import os
5import sys
Siobhan Tully06d2c032014-02-23 23:48:52 -05006from django.views.generic import TemplateView, View
7import datetime
Scott Baker771819b2014-03-19 22:10:17 -07008from pprint import pprint
9import json
Scott Baker168b7412014-05-16 14:25:41 -070010from syndicate.models import *
Scott Baker537d8232014-04-08 23:40:56 -070011from core.models import *
Scott Bakere9f4ec92014-04-29 11:29:45 -070012from hpc.models import ContentProvider
Scott Bakerbd57a432014-04-14 16:12:15 -070013from operator import attrgetter
Scott Baker186ae612014-05-19 10:42:51 -070014from django import template
Scott Bakerbd57a432014-04-14 16:12:15 -070015from django.views.decorators.csrf import csrf_exempt
Scott Baker60fd8cd2014-05-21 18:02:44 -070016from django.http import HttpResponse, HttpResponseServerError, HttpResponseForbidden
Scott Baker537d8232014-04-08 23:40:56 -070017from django.core import urlresolvers
Scott Bakerbd57a432014-04-14 16:12:15 -070018from django.contrib.gis.geoip import GeoIP
19from ipware.ip import get_ip
Scott Baker2c3cb642014-05-19 17:55:56 -070020from operator import itemgetter, attrgetter
Scott Baker771819b2014-03-19 22:10:17 -070021import traceback
Scott Baker6253dfc2014-04-14 09:29:30 -070022import socket
Siobhan Tully06d2c032014-02-23 23:48:52 -050023
Scott Bakera880bb52014-04-14 23:52:48 -070024BLESSED_SITES = ["Stanford", "Washington", "Princeton", "GeorgiaTech", "MaxPlanck"]
25
Scott Baker771819b2014-03-19 22:10:17 -070026if os.path.exists("/home/smbaker/projects/vicci/cdn/bigquery"):
27 sys.path.append("/home/smbaker/projects/vicci/cdn/bigquery")
28else:
29 sys.path.append("/opt/planetstack/hpc_wizard")
30import hpc_wizard
Scott Baker75095b62014-04-21 17:32:09 -070031from planetstack_analytics import DoPlanetStackAnalytics, PlanetStackAnalytics, RED_LOAD, BLUE_LOAD
Siobhan Tullycf04fb62014-01-11 11:25:57 -050032
33class DashboardWelcomeView(TemplateView):
34 template_name = 'admin/dashboard/welcome.html'
35
36 def get(self, request, *args, **kwargs):
37 context = self.get_context_data(**kwargs)
Scott Baker2c3cb642014-05-19 17:55:56 -070038 context = getDashboardContext(request.user, context)
Siobhan Tullycf04fb62014-01-11 11:25:57 -050039 return self.render_to_response(context=context)
Siobhan Tullye18b3442014-02-23 14:23:34 -050040
Scott Baker2c3cb642014-05-19 17:55:56 -070041class DashboardDynamicView(TemplateView):
Scott Baker186ae612014-05-19 10:42:51 -070042 head_template = r"""{% extends "admin/dashboard/dashboard_base.html" %}
43 {% load admin_static %}
44 {% block content %}
45 """
46
47 tail_template = r"{% endblock %}"
48
Scott Baker2c3cb642014-05-19 17:55:56 -070049 def get(self, request, name="root", *args, **kwargs):
Scott Baker186ae612014-05-19 10:42:51 -070050 context = self.get_context_data(**kwargs)
Scott Baker2c3cb642014-05-19 17:55:56 -070051 context = getDashboardContext(request.user, context)
Scott Baker186ae612014-05-19 10:42:51 -070052
Scott Baker2c3cb642014-05-19 17:55:56 -070053 if name=="root":
54 return self.multiDashboardView(request, context)
55 else:
56 return self.singleDashboardView(request, name, context)
57
58 def readDashboard(self, fn):
59 try:
60 template= open("/opt/planetstack/templates/admin/dashboard/%s.html" % fn, "r").read()
61 if (fn=="tenant"):
Scott Baker7237cdc2014-05-21 16:55:18 -070062 # fix for tenant view - it writes html to a div called tabs-5
Scott Baker2c3cb642014-05-19 17:55:56 -070063 template = '<div id="tabs-5"></div>' + template
Scott Baker7237cdc2014-05-21 16:55:18 -070064 if (fn=="slice_interactions"):
65 # fix for slice_interactions - it gives its container div a 40px
66 # margin, and then positions it's header at -40px
Scott Baker2de3f8f2014-05-21 16:56:58 -070067 template = '<div id="tabs-4">' + template + '</div>'
Scott Baker2c3cb642014-05-19 17:55:56 -070068 return template
69 except:
70 return "failed to open %s" % fn
71
72 def multiDashboardView(self, request, context):
Scott Bakerfc06e782014-05-19 10:52:10 -070073 head_template = self.head_template
74 tail_template = self.tail_template
75
Scott Baker2c3cb642014-05-19 17:55:56 -070076 body = """
77 <div id="hometabs" >
78 <ul id="suit_form_tabs" class="nav nav-tabs nav-tabs-suit" data-tab-prefix="suit-tab">
79 """
Scott Bakerfc06e782014-05-19 10:52:10 -070080
Scott Baker2c3cb642014-05-19 17:55:56 -070081 dashboards = request.user.get_dashboards()
Scott Bakerfc06e782014-05-19 10:52:10 -070082
Scott Baker2c3cb642014-05-19 17:55:56 -070083 # customize is a special dashboard they always get
84 customize = DashboardView.objects.filter(name="Customize")
85 if customize:
86 dashboards.append(customize[0])
Scott Baker186ae612014-05-19 10:42:51 -070087
Scott Baker2c3cb642014-05-19 17:55:56 -070088 for i,view in enumerate(dashboards):
89 body = body + '<li><a href="#dashtab-%d">%s</a></li>\n' % (i, view.name)
Scott Baker186ae612014-05-19 10:42:51 -070090
Scott Baker2c3cb642014-05-19 17:55:56 -070091 body = body + "</ul>\n"
92
93 for i,view in enumerate(dashboards):
94 url = view.url
95 body = body + '<div id="dashtab-%d">\n' % i
96 if url.startswith("template:"):
97 fn = url[9:]
98 body = body + self.readDashboard(fn)
99 body = body + '</div>\n'
100
101 body=body+"</div>\n"
102
103 t = template.Template(head_template + body + self.tail_template)
Scott Baker186ae612014-05-19 10:42:51 -0700104
105 response_kwargs = {}
106 response_kwargs.setdefault('content_type', self.content_type)
107 return self.response_class(
108 request = request,
109 template = t,
110 context = context,
111 **response_kwargs)
112
Scott Baker2c3cb642014-05-19 17:55:56 -0700113 def singleDashboardView(self, request, name, context):
114 head_template = self.head_template
115 tail_template = self.tail_template
116
117 t = template.Template(head_template + self.readDashboard(fn) + self.tail_template)
118
119 response_kwargs = {}
120 response_kwargs.setdefault('content_type', self.content_type)
121 return self.response_class(
122 request = request,
123 template = t,
124 context = context,
125 **response_kwargs)
126
127def getDashboardContext(user, context={}, tableFormat = False):
128 context = {}
Scott Baker771819b2014-03-19 22:10:17 -0700129
130 userSliceData = getSliceInfo(user)
131 if (tableFormat):
Scott Baker2c3cb642014-05-19 17:55:56 -0700132 context['userSliceInfo'] = userSliceTableFormatter(userSliceData)
Scott Baker771819b2014-03-19 22:10:17 -0700133 else:
Scott Baker2c3cb642014-05-19 17:55:56 -0700134 context['userSliceInfo'] = userSliceData
135 context['cdnData'] = getCDNOperatorData(wait=False)
136 context['cdnContentProviders'] = getCDNContentProviderData()
137
138 (dashboards, unusedDashboards)= getDashboards(user)
139 unusedDashboards=[x for x in unusedDashboards if x!="Customize"]
140 context['dashboards'] = dashboards
141 context['unusedDashboards'] = unusedDashboards
142
143 return context
144
145def getDashboards(user):
146 #dashboards = sorted(list(user.dashboardViews.all()), key=attrgetter('order'))
147 dashboards = user.get_dashboards()
148
149 dashboard_names = [d.name for d in dashboards]
150
151 unused_dashboard_names = []
152 for dashboardView in DashboardView.objects.all():
153 if not dashboardView.name in dashboard_names:
154 unused_dashboard_names.append(dashboardView.name)
155
156 return (dashboard_names, unused_dashboard_names)
Scott Baker771819b2014-03-19 22:10:17 -0700157
Scott Baker537d8232014-04-08 23:40:56 -0700158class TenantCreateSlice(View):
159 def post(self, request, *args, **kwargs):
Scott Baker94319c52014-05-26 11:45:29 -0700160 if request.user.isReadOnlyUser():
161 return HttpResponseForbidden("User is in read-only mode")
162
Scott Baker537d8232014-04-08 23:40:56 -0700163 sliceName = request.POST.get("sliceName", "0")
164 serviceClass = request.POST.get("serviceClass", "0")
165 imageName = request.POST.get("imageName", "0")
166 actionToDo = request.POST.get("actionToDo", "0")
Scott Baker9d9650d2014-05-23 11:50:54 -0700167 networkPorts = request.POST.get("network","0")
Scott Bakeredf2f672014-04-22 10:17:37 -0700168 mountDataSets = request.POST.get("mountDataSets","0")
Scott Baker9d9650d2014-05-23 11:50:54 -0700169 privateVolume = request.POST.get("privateVolume","0")
Scott Baker537d8232014-04-08 23:40:56 -0700170 if (actionToDo == "add"):
171 serviceClass = ServiceClass.objects.get(name=serviceClass)
172 site = request.user.site
Scott Baker4b0cedd2014-04-22 21:01:21 -0700173 image = Image.objects.get(name=imageName)
Scott Baker168b7412014-05-16 14:25:41 -0700174 newSlice = Slice(name=sliceName,serviceClass=serviceClass,site=site,imagePreference=image,mountDataSets=mountDataSets)
Scott Baker537d8232014-04-08 23:40:56 -0700175 newSlice.save()
Scott Baker9d9650d2014-05-23 11:50:54 -0700176 privateTemplate="Private"
177 publicTemplate="Public shared IPv4"
178 privateNetworkName = sliceName+"-"+privateTemplate
179 publicNetworkName = sliceName+"-"+publicTemplate
180 slice=Slice.objects.get(name=sliceName)
181 addNetwork(privateNetworkName,privateTemplate,slice)
182 addNetwork(publicNetworkName,publicTemplate,slice)
183 addOrModifyPorts(networkPorts,sliceName)
184 if privateVolume=="true":
185 privateVolForSlice(request.user,sliceName)
Scott Baker4b0cedd2014-04-22 21:01:21 -0700186 return HttpResponse("Slice created")
Scott Baker537d8232014-04-08 23:40:56 -0700187
Scott Baker9d9650d2014-05-23 11:50:54 -0700188def privateVolForSlice(user,sliceName):
189 if not hasPrivateVolume(sliceName):
190 volumeName=createPrivateVolume(user,sliceName)
191 readWrite="true"
192 mountVolume(sliceName,volumeName,readWrite)
193
Scott Bakeredf2f672014-04-22 10:17:37 -0700194class TenantUpdateSlice(View):
195 def post(self, request, *args, **kwargs):
Scott Baker94319c52014-05-26 11:45:29 -0700196 if request.user.isReadOnlyUser():
197 return HttpResponseForbidden("User is in read-only mode")
198
Scott Bakeredf2f672014-04-22 10:17:37 -0700199 sliceName = request.POST.get("sliceName", "0")
200 serviceClass = request.POST.get("serviceClass", "0")
201 imageName = request.POST.get("imageName", "0")
202 actionToDo = request.POST.get("actionToDo", "0")
Scott Baker9d9650d2014-05-23 11:50:54 -0700203 networkPorts = request.POST.get("networkPorts","0")
Scott Baker4b0cedd2014-04-22 21:01:21 -0700204 dataSet = request.POST.get("dataSet","0")
Scott Baker9d9650d2014-05-23 11:50:54 -0700205 privateVolume = request.POST.get("privateVolume","0")
Scott Baker4b0cedd2014-04-22 21:01:21 -0700206 slice = Slice.objects.all()
207 for entry in slice:
208 serviceClass = ServiceClass.objects.get(name=serviceClass)
209 if(entry.name==sliceName):
210 if (actionToDo == "update"):
211 setattr(entry,'serviceClass',serviceClass)
212 setattr(entry,'imagePreference',imageName)
Scott Baker4b0cedd2014-04-22 21:01:21 -0700213 setattr(entry,'mountDataSets',dataSet)
214 entry.save()
215 break
Scott Baker9d9650d2014-05-23 11:50:54 -0700216 addOrModifyPorts(networkPorts,sliceName)
217 if privateVolume=="true":
218 privateVolForSlice(request.user,sliceName)
Scott Baker4b0cedd2014-04-22 21:01:21 -0700219 return HttpResponse("Slice updated")
220
Scott Baker9d9650d2014-05-23 11:50:54 -0700221def addNetwork(name,template,sliceName):
222 networkTemplate=NetworkTemplate.objects.get(name=template)
223 newNetwork = Network(name = name,
224 template = networkTemplate,
225 owner = sliceName)
226 newNetwork.save()
227 addNetworkSlice(newNetwork,sliceName)
228
229def addNetworkSlice(networkSlice,sliceName):
230 newNetworkSlice=NetworkSlice(network =networkSlice,
231 slice=sliceName)
232 newNetworkSlice.save()
233
234def addOrModifyPorts(networkPorts,sliceName):
235 networkList = Network.objects.all()
236 networkInfo = []
237 if networkPorts:
238 for networkEntry in networkList:
239 networkSlices = networkEntry.slices.all()
240 for slice in networkSlices:
241 if slice.name==sliceName:
242 if networkEntry.template.name=="Public shared IPv4":
243 setattr(networkEntry,'ports',networkPorts)
244 networkEntry.save()
245
Scott Baker537d8232014-04-08 23:40:56 -0700246def getTenantSliceInfo(user, tableFormat = False):
247 tenantSliceDetails = {}
248 tenantSliceData = getTenantInfo(user)
249 tenantServiceClassData = getServiceClassInfo(user)
250 if (tableFormat):
251 tenantSliceDetails['userSliceInfo'] = userSliceTableFormatter(tenantSliceData)
252 tenantSliceDetails['sliceServiceClass']=userSliceTableFormatter(tenantServiceClassData)
253 else:
254 tenantSliceDetails['userSliceInfo'] = tenantSliceData
255 tenantSliceDetails['sliceServiceClass']=userSliceTableFormatter(tenantServiceClassData)
256 tenantSliceDetails['image']=userSliceTableFormatter(getImageInfo(user))
Scott Baker537d8232014-04-08 23:40:56 -0700257 tenantSliceDetails['deploymentSites']=userSliceTableFormatter(getDeploymentSites())
Scott Bakerb85c26b2014-04-18 11:25:32 -0700258 tenantSliceDetails['sites'] = userSliceTableFormatter(getTenantSitesInfo())
259 tenantSliceDetails['mountDataSets'] = userSliceTableFormatter(getMountDataSets())
Scott Baker168b7412014-05-16 14:25:41 -0700260 tenantSliceDetails['publicKey'] = getPublicKey(user)
Scott Baker537d8232014-04-08 23:40:56 -0700261 return tenantSliceDetails
262
Scott Baker537d8232014-04-08 23:40:56 -0700263def getTenantInfo(user):
264 slices =Slice.objects.all()
265 userSliceInfo = []
266 for entry in slices:
267 sliceName = Slice.objects.get(id=entry.id).name
268 slice = Slice.objects.get(name=Slice.objects.get(id=entry.id).name)
269 sliceServiceClass = entry.serviceClass.name
270 preferredImage = entry.imagePreference
Scott Baker9d9650d2014-05-23 11:50:54 -0700271 #sliceDataSet = entry.mountDataSets
272 sliceNetwork = {}
Scott Baker537d8232014-04-08 23:40:56 -0700273 numSliver = 0
274 sliceImage=""
275 sliceSite = {}
Scott Bakeredf2f672014-04-22 10:17:37 -0700276 sliceNode = {}
277 sliceInstance= {}
Scott Baker9d9650d2014-05-23 11:50:54 -0700278 #createPrivateVolume(user,sliceName)
Scott Baker537d8232014-04-08 23:40:56 -0700279 for sliver in slice.slivers.all():
Scott Bakera880bb52014-04-14 23:52:48 -0700280 if sliver.node.site.name in BLESSED_SITES:
Scott Bakera880bb52014-04-14 23:52:48 -0700281 sliceSite[sliver.node.site.name] = sliceSite.get(sliver.node.site.name,0) + 1
282 sliceImage = sliver.image.name
Scott Bakerb37ce332014-04-22 22:57:40 -0700283 sliceNode[str(sliver)] = sliver.node.name
Scott Bakeredf2f672014-04-22 10:17:37 -0700284 numSliver = sum(sliceSite.values())
Scott Bakerb85c26b2014-04-18 11:25:32 -0700285 numSites = len(sliceSite)
Scott Baker9d9650d2014-05-23 11:50:54 -0700286 userSliceInfo.append({'sliceName': sliceName,'sliceServiceClass': sliceServiceClass,'preferredImage':preferredImage,'numOfSites':numSites, 'sliceSite':sliceSite,'sliceImage':sliceImage,'numOfSlivers':numSliver,'instanceNodePair':sliceNode})
Scott Baker537d8232014-04-08 23:40:56 -0700287 return userSliceInfo
288
289def getTenantSitesInfo():
290 tenantSiteInfo=[]
291 for entry in Site.objects.all():
Scott Bakera880bb52014-04-14 23:52:48 -0700292 if entry.name in BLESSED_SITES:
293 tenantSiteInfo.append({'siteName':entry.name})
Scott Baker537d8232014-04-08 23:40:56 -0700294 return tenantSiteInfo
295
Scott Baker771819b2014-03-19 22:10:17 -0700296def userSliceTableFormatter(data):
297# pprint(data)
298 formattedData = {
299 'rows' : data
300 }
301 return formattedData
302
Scott Baker168b7412014-05-16 14:25:41 -0700303def getPublicKey(user):
304 users=User.objects.all()
305 for key in users:
306 if (str(key.email)==str(user)):
307 sshKey = key.public_key
308 return sshKey
309
Scott Baker537d8232014-04-08 23:40:56 -0700310def getServiceClassInfo(user):
311 serviceClassList = ServiceClass.objects.all()
312 sliceInfo = []
313 for entry in serviceClassList:
314 sliceInfo.append({'serviceClass':entry.name})
315 return sliceInfo
316
317def getImageInfo(user):
Scott Baker4b0cedd2014-04-22 21:01:21 -0700318 imageList = Image.objects.all()
319 #imageList = ['Fedora 16 LXC rev 1.3','Hadoop','MPI']
Scott Baker537d8232014-04-08 23:40:56 -0700320 imageInfo = []
321 for imageEntry in imageList:
Scott Baker4b0cedd2014-04-22 21:01:21 -0700322 imageInfo.append({'Image':imageEntry.name})
323 #imageInfo.append({'Image':imageEntry})
Scott Baker537d8232014-04-08 23:40:56 -0700324 return imageInfo
325
Scott Baker168b7412014-05-16 14:25:41 -0700326def createPrivateVolume(user, sliceName):
327 caps = Volume.CAP_READ_DATA | Volume.CAP_WRITE_DATA | Volume.CAP_HOST_DATA
328 getattr(Volume.default_gateway_caps,"read data") | \
329 getattr(Volume.default_gateway_caps,"write data") | \
330 getattr(Volume.default_gateway_caps,"host files")
331 v = Volume(name="private_" + sliceName, owner_id=user, description="private volume for %s" % sliceName, blocksize=61440, private=True, archive=False, default_gateway_caps = caps)
332 v.save()
Scott Baker9d9650d2014-05-23 11:50:54 -0700333 return v
Scott Baker168b7412014-05-16 14:25:41 -0700334
335SYNDICATE_REPLICATE_PORTNUM = 1025
336
337def get_free_port():
338 inuse={}
339 inuse[SYNDICATE_REPLICATE_PORTNUM] = True
340 for vs in VolumeSlice.objects.all():
341 inuse[vs.peer_portnum]=True
342 inuse[vs.replicate_portnum]=True
343 for network in Network.objects.all():
Scott Baker9d9650d2014-05-23 11:50:54 -0700344 if not network_ports:
345 continue
Scott Baker168b7412014-05-16 14:25:41 -0700346 network_ports = [x.strip() for x in network.ports.split(",")]
347 for network_port in network_ports:
348 try:
349 inuse[int(network_port)] = True
350 except:
351 # in case someone has put a malformed port number in the list
352 pass
353 for i in range(1025, 65535):
354 if not inuse.get(i,False):
355 return i
356 return False
357
358def mountVolume(sliceName, volumeName, readWrite):
359 slice = Slice.objects.get(name=sliceName)
360 volume = Volume.objects.get(name=volumeName)
361 # choose some unused port numbers
362 flags = Volume.CAP_READ_DATA
363 if readWrite:
364 flags = flags | Volume.CAP_WRITE_DATA
365 vs = VolumeSlice(volume_id = volume, slice_id = slice, gateway_caps=flags, peer_portnum = get_free_port(), replicate_portnum = SYNDICATE_REPLICATE_PORTNUM)
366 vs.save()
367
368def hasPrivateVolume(sliceName):
369 slice = Slice.objects.get(name=sliceName)
370 for vs in VolumeSlice.objects.filter(slice_id=slice):
371 if vs.volume_id.private:
372 return True
373 return False
374
Scott Bakerb85c26b2014-04-18 11:25:32 -0700375def getMountDataSets():
Scott Baker168b7412014-05-16 14:25:41 -0700376 dataSetInfo=[]
377 for volume in Volume.objects.all():
378 if not volume.private:
379 dataSetInfo.append({'DataSet': volume.name})
380
Scott Bakerb85c26b2014-04-18 11:25:32 -0700381 return dataSetInfo
382
Scott Baker537d8232014-04-08 23:40:56 -0700383def getDeploymentSites():
384 deploymentList = Deployment.objects.all()
385 deploymentInfo = []
386 for entry in deploymentList:
387 deploymentInfo.append({'DeploymentSite':entry.name})
388 return deploymentInfo
389
Scott Baker771819b2014-03-19 22:10:17 -0700390def getSliceInfo(user):
Siobhan Tully06d2c032014-02-23 23:48:52 -0500391 sliceList = Slice.objects.all()
Scott Baker771819b2014-03-19 22:10:17 -0700392 slicePrivs = SlicePrivilege.objects.filter(user=user)
Siobhan Tully06d2c032014-02-23 23:48:52 -0500393 userSliceInfo = []
394 for entry in slicePrivs:
Siobhan Tullye18b3442014-02-23 14:23:34 -0500395
Scott Baker771819b2014-03-19 22:10:17 -0700396 slicename = Slice.objects.get(id=entry.slice.id).name
Scott Baker537d8232014-04-08 23:40:56 -0700397 slice = Slice.objects.get(name=Slice.objects.get(id=entry.slice.id).name)
398 sliverList=Sliver.objects.all()
399 sites_used = {}
400 for sliver in slice.slivers.all():
401 #sites_used['deploymentSites'] = sliver.node.deployment.name
402 # sites_used[sliver.image.name] = sliver.image.name
403 sites_used[sliver.node.site.name] = sliver.numberCores
Scott Baker771819b2014-03-19 22:10:17 -0700404 sliceid = Slice.objects.get(id=entry.slice.id).id
Siobhan Tully06d2c032014-02-23 23:48:52 -0500405 try:
Scott Baker771819b2014-03-19 22:10:17 -0700406 sliverList = Sliver.objects.filter(slice=entry.slice.id)
407 siteList = {}
408 for x in sliverList:
409 if x.node.site not in siteList:
410 siteList[x.node.site] = 1
411 slivercount = len(sliverList)
412 sitecount = len(siteList)
Siobhan Tully06d2c032014-02-23 23:48:52 -0500413 except:
Scott Baker771819b2014-03-19 22:10:17 -0700414 traceback.print_exc()
415 slivercount = 0
416 sitecount = 0
Siobhan Tully06d2c032014-02-23 23:48:52 -0500417
Scott Baker771819b2014-03-19 22:10:17 -0700418 userSliceInfo.append({'slicename': slicename, 'sliceid':sliceid,
Scott Baker537d8232014-04-08 23:40:56 -0700419 'sitesUsed':sites_used,
420 'role': SliceRole.objects.get(id=entry.role.id).role,
421 'slivercount': slivercount,
422 'sitecount':sitecount})
Scott Baker771819b2014-03-19 22:10:17 -0700423
Siobhan Tully06d2c032014-02-23 23:48:52 -0500424 return userSliceInfo
425
Scott Bakere9f4ec92014-04-29 11:29:45 -0700426def getCDNContentProviderData():
427 cps = []
428 for dm_cp in ContentProvider.objects.all():
429 cp = {"name": dm_cp.name,
430 "account": dm_cp.account}
431 cps.append(cp)
432
433 return cps
434
Scott Baker75095b62014-04-21 17:32:09 -0700435def getCDNOperatorData(randomizeData = False, wait=True):
436 HPC_SLICE_NAME = "HyperCache"
437
Scott Baker0017c102014-04-18 10:48:31 -0700438 bq = PlanetStackAnalytics()
Siobhan Tully06d2c032014-02-23 23:48:52 -0500439
Scott Baker4a4472d2014-05-13 17:07:46 -0700440 rows = bq.get_cached_query_results(bq.compose_cached_query(), wait)
Siobhan Tully06d2c032014-02-23 23:48:52 -0500441
Scott Bakeredf2f672014-04-22 10:17:37 -0700442 # wait=False on the first time the Dashboard is opened. This means we might
443 # not have any rows yet. The dashboard code polls every 30 seconds, so it
444 # will eventually pick them up.
Scott Baker0017c102014-04-18 10:48:31 -0700445
Scott Baker75095b62014-04-21 17:32:09 -0700446 if rows:
Scott Baker696e2662014-04-24 17:03:29 -0700447 rows = bq.postprocess_results(rows, filter={"event": "hpc_heartbeat"}, maxi=["cpu"], count=["hostname"], computed=["bytes_sent/elapsed"], groupBy=["Time","site"], maxDeltaTime=80)
Scott Baker0017c102014-04-18 10:48:31 -0700448
Scott Baker75095b62014-04-21 17:32:09 -0700449 # dictionaryize the statistics rows by site name
450 stats_rows = {}
451 for row in rows:
452 stats_rows[row["site"]] = row
Scott Bakeredf2f672014-04-22 10:17:37 -0700453 else:
454 stats_rows = {}
Scott Baker0017c102014-04-18 10:48:31 -0700455
Scott Baker09fad022014-04-30 09:50:01 -0700456 slice = Slice.objects.filter(name=HPC_SLICE_NAME)
457 if slice:
458 slice_slivers = list(slice[0].slivers.all())
459 else:
460 slice_slivers = []
Scott Baker0017c102014-04-18 10:48:31 -0700461
462 new_rows = {}
Scott Baker75095b62014-04-21 17:32:09 -0700463 for site in Site.objects.all():
464 # compute number of slivers allocated in the data model
465 allocated_slivers = 0
466 for sliver in slice_slivers:
467 if sliver.node.site == site:
468 allocated_slivers = allocated_slivers + 1
469
470 stats_row = stats_rows.get(site.name,{})
471
472 max_cpu = stats_row.get("max_avg_cpu", stats_row.get("max_cpu",0))
473 cpu=float(max_cpu)/100.0
474 hotness = max(0.0, ((cpu*RED_LOAD) - BLUE_LOAD)/(RED_LOAD-BLUE_LOAD))
475
476 # format it to what that CDN Operations View is expecting
477 new_row = {"lat": float(site.location.longitude),
478 "long": float(site.location.longitude),
479 "lat": float(site.location.latitude),
Scott Baker0017c102014-04-18 10:48:31 -0700480 "health": 0,
Scott Baker75095b62014-04-21 17:32:09 -0700481 "numNodes": int(site.nodes.count()),
Scott Bakere9f4ec92014-04-29 11:29:45 -0700482 "activeHPCSlivers": int(stats_row.get("count_hostname", 0)), # measured number of slivers, from bigquery statistics
Scott Baker75095b62014-04-21 17:32:09 -0700483 "numHPCSlivers": allocated_slivers, # allocated number of slivers, from data model
484 "siteUrl": str(site.site_url),
485 "bandwidth": stats_row.get("sum_computed_bytes_sent_div_elapsed",0),
486 "load": max_cpu,
487 "hot": float(hotness)}
488 new_rows[str(site.name)] = new_row
489
490 # get rid of sites with 0 slivers that overlap other sites with >0 slivers
491 for (k,v) in new_rows.items():
492 bad=False
493 if v["numHPCSlivers"]==0:
494 for v2 in new_rows.values():
495 if (v!=v2) and (v2["numHPCSlivers"]>=0):
496 d = haversine(v["lat"],v["long"],v2["lat"],v2["long"])
497 if d<100:
498 bad=True
499 if bad:
500 del new_rows[k]
Scott Baker0017c102014-04-18 10:48:31 -0700501
502 return new_rows
Siobhan Tully06d2c032014-02-23 23:48:52 -0500503
Scott Baker168b7412014-05-16 14:25:41 -0700504def getPageSummary(request):
505 slice = request.GET.get('slice', None)
506 site = request.GET.get('site', None)
507 node = request.GET.get('node', None)
508
Scott Baker771819b2014-03-19 22:10:17 -0700509class SimulatorView(View):
510 def get(self, request, **kwargs):
511 sim = json.loads(file("/tmp/simulator.json","r").read())
512 text = "<html><head></head><body>"
513 text += "Iteration: %d<br>" % sim["iteration"]
514 text += "Elapsed since report %d<br><br>" % sim["elapsed_since_report"]
515 text += "<table border=1>"
516 text += "<tr><th>site</th><th>trend</th><th>weight</th><th>bytes_sent</th><th>hot</th></tr>"
517 for site in sim["site_load"].values():
518 text += "<tr>"
519 text += "<td>%s</td><td>%0.2f</td><td>%0.2f</td><td>%d</td><td>%0.2f</td>" % \
520 (site["name"], site["trend"], site["weight"], site["bytes_sent"], site["load_frac"])
521 text += "</tr>"
522 text += "</table>"
523 text += "</body></html>"
524 return HttpResponse(text)
525
526class DashboardUserSiteView(View):
527 def get(self, request, **kwargs):
Scott Baker2c3cb642014-05-19 17:55:56 -0700528 return HttpResponse(json.dumps(getDashboardContext(request.user, tableFormat=True)), mimetype='application/javascript')
Scott Baker771819b2014-03-19 22:10:17 -0700529
Scott Baker537d8232014-04-08 23:40:56 -0700530class TenantViewData(View):
531 def get(self, request, **kwargs):
532 return HttpResponse(json.dumps(getTenantSliceInfo(request.user, True)), mimetype='application/javascript')
533
Scott Baker75095b62014-04-21 17:32:09 -0700534def haversine(site_lat, site_lon, lat, lon):
Scott Baker75095b62014-04-21 17:32:09 -0700535 d=0
536 if lat and lon and site_lat and site_lon:
Scott Baker4b0cedd2014-04-22 21:01:21 -0700537 site_lat = float(site_lat)
538 site_lon = float(site_lon)
539 lat = float(lat)
540 lon = float(lon)
Scott Baker75095b62014-04-21 17:32:09 -0700541 R = 6378.1
542 a = math.sin( math.radians((lat - site_lat)/2.0) )**2 + math.cos( math.radians(lat) )*math.cos( math.radians(site_lat) )*(math.sin( math.radians((lon - site_lon)/2.0 ) )**2)
543 c = 2 * math.atan2( math.sqrt(a), math.sqrt(1 - a) )
544 d = R * c
545
546 return d
547
Scott Bakerbd57a432014-04-14 16:12:15 -0700548def siteSortKey(site, slice=None, count=None, lat=None, lon=None):
549 # try to pick a site we're already using
550 has_slivers_here=False
551 if slice:
552 for sliver in slice.slivers.all():
553 if sliver.node.site.name == site.name:
554 has_slivers_here=True
555
556 # Haversine method
Scott Baker75095b62014-04-21 17:32:09 -0700557 d = haversine(site.location.latitude, site.location.longitude, lat, lon)
Scott Bakerbd57a432014-04-14 16:12:15 -0700558
559 return (-has_slivers_here, d)
560
561def tenant_pick_sites(user, user_ip=None, slice=None, count=None):
562 """ Returns list of sites, sorted from most favorable to least favorable """
563 lat=None
564 lon=None
565 try:
566 client_geo = GeoIP().city(user_ip)
567 if client_geo:
568 lat=float(client_geo["latitude"])
569 lon=float(client_geo["longitude"])
570 except:
571 print "exception in geo code"
572 traceback.print_exc()
573
574 sites = Site.objects.all()
Scott Bakera880bb52014-04-14 23:52:48 -0700575 sites = [x for x in sites if x.name in BLESSED_SITES]
Scott Bakerbd57a432014-04-14 16:12:15 -0700576 sites = sorted(sites, key=functools.partial(siteSortKey, slice=slice, count=count, lat=lat, lon=lon))
577
578 return sites
579
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700580def slice_increase_slivers(user, user_ip, siteList, slice, count, noAct=False):
Scott Bakere4deddd2014-04-14 16:48:05 -0700581 sitesChanged = {}
582
Scott Bakerbd57a432014-04-14 16:12:15 -0700583 # let's compute how many slivers are in use in each node of each site
584 for site in siteList:
585 site.nodeList = list(site.nodes.all())
586 for node in site.nodeList:
587 node.sliverCount = 0
Scott Baker6253dfc2014-04-14 09:29:30 -0700588 for sliver in node.slivers.all():
Scott Baker21511942014-04-15 17:44:27 -0700589 if sliver.slice.id == slice.id:
Scott Baker0017c102014-04-18 10:48:31 -0700590 node.sliverCount = node.sliverCount + 1
Scott Baker6253dfc2014-04-14 09:29:30 -0700591
Scott Bakerbd57a432014-04-14 16:12:15 -0700592 # Allocate slivers to nodes
593 # for now, assume we want to allocate all slivers from the same site
594 nodes = siteList[0].nodeList
595 while (count>0):
596 # Sort the node list by number of slivers per node, then pick the
597 # node with the least number of slivers.
598 nodes = sorted(nodes, key=attrgetter("sliverCount"))
599 node = nodes[0]
Scott Baker6253dfc2014-04-14 09:29:30 -0700600
Scott Bakerbd57a432014-04-14 16:12:15 -0700601 print "adding sliver at node", node.name, "of site", node.site.name
Scott Baker6253dfc2014-04-14 09:29:30 -0700602
Scott Bakere4deddd2014-04-14 16:48:05 -0700603 if not noAct:
604 sliver = Sliver(name=node.name,
605 slice=slice,
606 node=node,
607 image = Image.objects.all()[0],
608 creator = User.objects.get(email=user),
609 deploymentNetwork=node.deployment,
610 numberCores =1 )
611 sliver.save()
Scott Bakerbd57a432014-04-14 16:12:15 -0700612
613 node.sliverCount = node.sliverCount + 1
614
615 count = count - 1
616
Scott Bakere4deddd2014-04-14 16:48:05 -0700617 sitesChanged[node.site.name] = sitesChanged.get(node.site.name,0) + 1
618
619 return sitesChanged
620
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700621def slice_decrease_slivers(user, siteList, slice, count, noAct=False):
Scott Bakere4deddd2014-04-14 16:48:05 -0700622 sitesChanged = {}
Scott Bakerb85c26b2014-04-18 11:25:32 -0700623 sliverList ={}
Scott Bakerbd57a432014-04-14 16:12:15 -0700624 if siteList:
625 siteNames = [site.name for site in siteList]
626 else:
627 siteNames = None
628
629 for sliver in slice.slivers.all():
Scott Bakerb85c26b2014-04-18 11:25:32 -0700630 if(not siteNames) or (sliver.node.site.name in siteNames):
631 node = sliver.node
632 sliverList[sliver.name]=node.name
Scott Bakerbd57a432014-04-14 16:12:15 -0700633
Scott Bakerb85c26b2014-04-18 11:25:32 -0700634 for key in sliverList:
635 if count>0:
Scott Bakerf1b65c02014-04-21 01:03:17 -0700636 sliver = Sliver.objects.filter(name=key)[0]
Scott Bakerb85c26b2014-04-18 11:25:32 -0700637 sliver.delete()
638 print "deleting sliver",sliverList[key],"at node",sliver.node.name
639 count=count-1
Scott Bakerf1b65c02014-04-21 01:03:17 -0700640 sitesChanged[sliver.node.site.name] = sitesChanged.get(sliver.node.site.name,0) - 1
Scott Bakerb85c26b2014-04-18 11:25:32 -0700641
Scott Bakere4deddd2014-04-14 16:48:05 -0700642 return sitesChanged
643
Scott Bakerb85c26b2014-04-18 11:25:32 -0700644class TenantDeleteSliceView(View):
645 def post(self,request):
Scott Baker94319c52014-05-26 11:45:29 -0700646 if request.user.isReadOnlyUser():
647 return HttpResponseForbidden("User is in read-only mode")
Scott Bakerb85c26b2014-04-18 11:25:32 -0700648 sliceName = request.POST.get("sliceName",None)
649 slice = Slice.objects.get(name=sliceName)
Scott Baker168b7412014-05-16 14:25:41 -0700650 #print slice, slice.id
Scott Bakerb85c26b2014-04-18 11:25:32 -0700651 sliceToDel=Slice(name=sliceName, id=slice.id)
652 sliceToDel.delete()
Scott Baker4b0cedd2014-04-22 21:01:21 -0700653 return HttpResponse("Slice deleted")
Scott Bakerb85c26b2014-04-18 11:25:32 -0700654
Scott Baker6253dfc2014-04-14 09:29:30 -0700655class TenantAddOrRemoveSliverView(View):
Scott Bakere4deddd2014-04-14 16:48:05 -0700656 """ Add or remove slivers from a Slice
657
658 Arguments:
659 siteName - name of site. If not specified, PlanetStack will pick the
660 best site.,
661 actionToDo - [add | rem]
662 count - number of slivers to add or remove
663 sliceName - name of slice
Scott Bakera880bb52014-04-14 23:52:48 -0700664 noAct - if set, no changes will be made to db, but result will still
665 show which sites would have been modified.
Scott Bakere4deddd2014-04-14 16:48:05 -0700666
667 Returns:
668 Dictionary of sites that were modified, and the count of nodes
669 that were added or removed at each site.
670 """
Scott Baker6253dfc2014-04-14 09:29:30 -0700671 def post(self, request, *args, **kwargs):
Scott Bakerbd57a432014-04-14 16:12:15 -0700672 siteName = request.POST.get("siteName", None)
673 actionToDo = request.POST.get("actionToDo", None)
674 count = int(request.POST.get("count","0"))
675 sliceName = request.POST.get("slice", None)
Scott Bakere4deddd2014-04-14 16:48:05 -0700676 noAct = request.POST.get("noAct", False)
Scott Bakerbd57a432014-04-14 16:12:15 -0700677
678 if not sliceName:
679 return HttpResponseServerError("No slice name given")
680
681 slice = Slice.objects.get(name=sliceName)
682
683 if siteName:
684 siteList = [Site.objects.get(name=siteName)]
685 else:
686 siteList = None
Scott Baker6253dfc2014-04-14 09:29:30 -0700687
688 if (actionToDo == "add"):
Scott Bakerbd57a432014-04-14 16:12:15 -0700689 user_ip = request.GET.get("ip", get_ip(request))
690 if (siteList is None):
691 siteList = tenant_pick_sites(user, user_ip, slice, count)
692
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700693 sitesChanged = slice_increase_slivers(request.user, user_ip, siteList, slice, count, noAct)
Scott Baker6253dfc2014-04-14 09:29:30 -0700694 elif (actionToDo == "rem"):
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700695 sitesChanged = slice_decrease_slivers(request.user, siteList, slice, count, noAct)
Scott Bakerbd57a432014-04-14 16:12:15 -0700696 else:
697 return HttpResponseServerError("Unknown actionToDo %s" % actionToDo)
698
Scott Bakere4deddd2014-04-14 16:48:05 -0700699 return HttpResponse(json.dumps(sitesChanged), mimetype='application/javascript')
Scott Bakerbd57a432014-04-14 16:12:15 -0700700
701 def get(self, request, *args, **kwargs):
702 request.POST = request.GET
703 return self.post(request, *args, **kwargs) # for testing REST in browser
704 #return HttpResponseServerError("GET is not supported")
705
706class TenantPickSitesView(View):
707 """ primarily just for testing purposes """
708 def get(self, request, *args, **kwargs):
709 count = request.GET.get("count","0")
710 slice = request.GET.get("slice",None)
711 if slice:
712 slice = Slice.objects.get(name=slice)
713 ip = request.GET.get("ip", get_ip(request))
714 sites = tenant_pick_sites(request.user, user_ip=ip, count=0, slice=slice)
715 sites = [x.name for x in sites]
716 return HttpResponse(json.dumps(sites), mimetype='application/javascript')
Scott Baker6253dfc2014-04-14 09:29:30 -0700717
Scott Baker771819b2014-03-19 22:10:17 -0700718class DashboardSummaryAjaxView(View):
719 def get(self, request, **kwargs):
Scott Baker0017c102014-04-18 10:48:31 -0700720 def avg(x):
Scott Baker111702b2014-05-15 16:17:16 -0700721 if len(x)==0:
722 return 0
Scott Baker0017c102014-04-18 10:48:31 -0700723 return float(sum(x))/len(x)
724
725 sites = getCDNOperatorData().values()
726
727 sites = [site for site in sites if site["numHPCSlivers"]>0]
728
729 total_slivers = sum( [site["numHPCSlivers"] for site in sites] )
730 total_bandwidth = sum( [site["bandwidth"] for site in sites] )
731 average_cpu = int(avg( [site["load"] for site in sites] ))
732
733 result= {"total_slivers": total_slivers,
734 "total_bandwidth": total_bandwidth,
735 "average_cpu": average_cpu}
736
737 return HttpResponse(json.dumps(result), mimetype='application/javascript')
Scott Baker771819b2014-03-19 22:10:17 -0700738
739class DashboardAddOrRemoveSliverView(View):
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700740 # TODO: deprecate this view in favor of using TenantAddOrRemoveSliverView
741
Scott Baker771819b2014-03-19 22:10:17 -0700742 def post(self, request, *args, **kwargs):
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700743 siteName = request.POST.get("site", None)
Scott Baker771819b2014-03-19 22:10:17 -0700744 actionToDo = request.POST.get("actionToDo", "0")
745
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700746 siteList = [Site.objects.get(name=siteName)]
747 slice = Slice.objects.get(name="HyperCache")
748
Scott Bakerd927d662014-05-22 09:29:06 -0700749 if request.user.isReadOnlyUser():
750 return HttpResponseForbidden("User is in read-only mode")
751
Scott Baker771819b2014-03-19 22:10:17 -0700752 if (actionToDo == "add"):
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700753 user_ip = request.GET.get("ip", get_ip(request))
754 slice_increase_slivers(request.user, user_ip, siteList, slice, 1)
Scott Baker771819b2014-03-19 22:10:17 -0700755 elif (actionToDo == "rem"):
Scott Bakerfe6a7e12014-04-15 10:58:31 -0700756 slice_decrease_slivers(request.user, siteList, slice, 1)
Scott Baker771819b2014-03-19 22:10:17 -0700757
758 print '*' * 50
759 print 'Ask for site: ' + siteName + ' to ' + actionToDo + ' another HPC Sliver'
Scott Bakerd927d662014-05-22 09:29:06 -0700760 return HttpResponse(json.dumps("Success"), mimetype='application/javascript')
Siobhan Tully06d2c032014-02-23 23:48:52 -0500761
762class DashboardAjaxView(View):
763 def get(self, request, **kwargs):
764 return HttpResponse(json.dumps(getCDNOperatorData(True)), mimetype='application/javascript')
Scott Baker771819b2014-03-19 22:10:17 -0700765
766class DashboardAnalyticsAjaxView(View):
767 def get(self, request, name="hello_world", **kwargs):
768 if (name == "hpcSummary"):
769 return HttpResponse(json.dumps(hpc_wizard.get_hpc_wizard().get_summary_for_view()), mimetype='application/javascript')
770 elif (name == "hpcUserSite"):
Scott Baker2c3cb642014-05-19 17:55:56 -0700771 return HttpResponse(json.dumps(getDashboardContext(request.user, tableFormat=True)), mimetype='application/javascript')
Scott Baker771819b2014-03-19 22:10:17 -0700772 elif (name == "hpcMap"):
773 return HttpResponse(json.dumps(getCDNOperatorData(True)), mimetype='application/javascript')
774 elif (name == "bigquery"):
775 (mimetype, data) = DoPlanetStackAnalytics(request)
776 return HttpResponse(data, mimetype=mimetype)
777 else:
778 return HttpResponse(json.dumps("Unknown"), mimetype='application/javascript')
779
Scott Baker2c3cb642014-05-19 17:55:56 -0700780class DashboardCustomize(View):
781 def post(self, request, *args, **kwargs):
Scott Baker60fd8cd2014-05-21 18:02:44 -0700782 if request.user.isReadOnlyUser():
783 return HttpResponseForbidden("User is in read-only mode")
784
Scott Baker2c3cb642014-05-19 17:55:56 -0700785 dashboards = request.POST.get("dashboards", None)
786 if not dashboards:
Scott Baker2870e272014-05-19 18:14:54 -0700787 dashboards=[]
788 else:
789 dashboards = [x.strip() for x in dashboards.split(",")]
790 dashboards = [DashboardView.objects.get(name=x) for x in dashboards]
Scott Baker2c3cb642014-05-19 17:55:56 -0700791
792 request.user.dashboardViews.all().delete()
793
794 for i,dashboard in enumerate(dashboards):
795 udbv = UserDashboardView(user=request.user, dashboardView=dashboard, order=i)
796 udbv.save()
797
Scott Baker60fd8cd2014-05-21 18:02:44 -0700798 return HttpResponse(json.dumps("Success"), mimetype='application/javascript')
Scott Baker2c3cb642014-05-19 17:55:56 -0700799