| #views.py |
| import functools |
| import math |
| import os |
| import sys |
| from django.views.generic import TemplateView, View |
| import datetime |
| from pprint import pprint |
| import json |
| from core.models import * |
| from operator import attrgetter |
| from django.views.decorators.csrf import csrf_exempt |
| from django.http import HttpResponse, HttpResponseServerError |
| from django.core import urlresolvers |
| from django.contrib.gis.geoip import GeoIP |
| from ipware.ip import get_ip |
| import traceback |
| import socket |
| |
| BLESSED_SITES = ["Stanford", "Washington", "Princeton", "GeorgiaTech", "MaxPlanck"] |
| |
| if os.path.exists("/home/smbaker/projects/vicci/cdn/bigquery"): |
| sys.path.append("/home/smbaker/projects/vicci/cdn/bigquery") |
| else: |
| sys.path.append("/opt/planetstack/hpc_wizard") |
| import hpc_wizard |
| from planetstack_analytics import DoPlanetStackAnalytics |
| |
| class DashboardWelcomeView(TemplateView): |
| template_name = 'admin/dashboard/welcome.html' |
| |
| def get(self, request, *args, **kwargs): |
| context = self.get_context_data(**kwargs) |
| userDetails = getUserSliceInfo(request.user) |
| #context['site'] = userDetails['site'] |
| |
| context['userSliceInfo'] = userDetails['userSliceInfo'] |
| context['cdnData'] = userDetails['cdnData'] |
| return self.render_to_response(context=context) |
| |
| def getUserSliceInfo(user, tableFormat = False): |
| userDetails = {} |
| # try: |
| # // site = Site.objects.filter(id=user.site.id) |
| # // except: |
| # // site = Site.objects.filter(name="Princeton") |
| # // userDetails['sitename'] = site[0].name |
| # // userDetails['siteid'] = site[0].id |
| |
| userSliceData = getSliceInfo(user) |
| if (tableFormat): |
| # pprint("******* GET USER SLICE INFO") |
| userDetails['userSliceInfo'] = userSliceTableFormatter(userSliceData) |
| else: |
| userDetails['userSliceInfo'] = userSliceData |
| userDetails['cdnData'] = getCDNOperatorData(); |
| # pprint( userDetails) |
| return userDetails |
| |
| class TenantCreateSlice(View): |
| def post(self, request, *args, **kwargs): |
| sliceName = request.POST.get("sliceName", "0") |
| serviceClass = request.POST.get("serviceClass", "0") |
| imageName = request.POST.get("imageName", "0") |
| actionToDo = request.POST.get("actionToDo", "0") |
| print sliceName |
| if (actionToDo == "add"): |
| serviceClass = ServiceClass.objects.get(name=serviceClass) |
| site = request.user.site |
| #image = Image.objects.get(name=imageName) |
| newSlice = Slice(name=sliceName,serviceClass=serviceClass,site=site,imagePreference=imageName) |
| newSlice.save() |
| return newSlice |
| |
| |
| def getTenantSliceInfo(user, tableFormat = False): |
| tenantSliceDetails = {} |
| tenantSliceData = getTenantInfo(user) |
| tenantServiceClassData = getServiceClassInfo(user) |
| if (tableFormat): |
| tenantSliceDetails['userSliceInfo'] = userSliceTableFormatter(tenantSliceData) |
| tenantSliceDetails['sliceServiceClass']=userSliceTableFormatter(tenantServiceClassData) |
| else: |
| tenantSliceDetails['userSliceInfo'] = tenantSliceData |
| tenantSliceDetails['sliceServiceClass']=userSliceTableFormatter(tenantServiceClassData) |
| tenantSliceDetails['image']=userSliceTableFormatter(getImageInfo(user)) |
| tenantSliceDetails['network']=userSliceTableFormatter(getNetworkInfo(user)) |
| tenantSliceDetails['deploymentSites']=userSliceTableFormatter(getDeploymentSites()) |
| tenantSliceDetails['sites'] = userSliceTableFormatter(getTenantSitesInfo()); |
| return tenantSliceDetails |
| |
| |
| def getTenantInfo(user): |
| slices =Slice.objects.all() |
| userSliceInfo = [] |
| for entry in slices: |
| sliceName = Slice.objects.get(id=entry.id).name |
| slice = Slice.objects.get(name=Slice.objects.get(id=entry.id).name) |
| sliceServiceClass = entry.serviceClass.name |
| preferredImage = entry.imagePreference |
| numSliver = 0 |
| sliceImage="" |
| sliceSite = {} |
| for sliver in slice.slivers.all(): |
| numSliver +=sliver.numberCores |
| # sliceSite[sliver.deploymentNetwork.name] =sliceSite.get(sliver.deploymentNetwork.name,0) + 1 |
| if sliver.node.site.name in BLESSED_SITES: |
| print "equal",sliver.node.site.name |
| sliceSite[sliver.node.site.name] = sliceSite.get(sliver.node.site.name,0) + 1 |
| sliceImage = sliver.image.name |
| userSliceInfo.append({'sliceName': sliceName,'sliceServiceClass': sliceServiceClass,'preferredImage':preferredImage, 'sliceSite':sliceSite,'sliceImage':sliceImage,'numOfSlivers':numSliver}) |
| return userSliceInfo |
| |
| def getTenantSitesInfo(): |
| tenantSiteInfo=[] |
| for entry in Site.objects.all(): |
| if entry.name in BLESSED_SITES: |
| tenantSiteInfo.append({'siteName':entry.name}) |
| return tenantSiteInfo |
| |
| def userSliceTableFormatter(data): |
| # pprint(data) |
| formattedData = { |
| 'rows' : data |
| } |
| return formattedData |
| |
| def getServiceClassInfo(user): |
| serviceClassList = ServiceClass.objects.all() |
| sliceInfo = [] |
| for entry in serviceClassList: |
| sliceInfo.append({'serviceClass':entry.name}) |
| return sliceInfo |
| |
| def getImageInfo(user): |
| imageList = Image.objects.all() |
| imageInfo = [] |
| for imageEntry in imageList: |
| imageInfo.append({'Image':imageEntry.name}) |
| return imageInfo |
| |
| def getNetworkInfo(user): |
| #networkList = Network.objects.all() |
| networkList = ['Private Only','Private and Publicly Routable'] |
| networkInfo = [] |
| for networkEntry in networkList: |
| #networkInfo.append({'Network':networkEntry.name}) |
| networkInfo.append({'Network':networkEntry}) |
| return networkInfo |
| |
| def getDeploymentSites(): |
| deploymentList = Deployment.objects.all() |
| deploymentInfo = [] |
| for entry in deploymentList: |
| deploymentInfo.append({'DeploymentSite':entry.name}) |
| return deploymentInfo |
| |
| def getSliceInfo(user): |
| sliceList = Slice.objects.all() |
| slicePrivs = SlicePrivilege.objects.filter(user=user) |
| userSliceInfo = [] |
| for entry in slicePrivs: |
| |
| slicename = Slice.objects.get(id=entry.slice.id).name |
| slice = Slice.objects.get(name=Slice.objects.get(id=entry.slice.id).name) |
| sliverList=Sliver.objects.all() |
| sites_used = {} |
| for sliver in slice.slivers.all(): |
| #sites_used['deploymentSites'] = sliver.node.deployment.name |
| # sites_used[sliver.image.name] = sliver.image.name |
| sites_used[sliver.node.site.name] = sliver.numberCores |
| sliceid = Slice.objects.get(id=entry.slice.id).id |
| try: |
| sliverList = Sliver.objects.filter(slice=entry.slice.id) |
| siteList = {} |
| for x in sliverList: |
| if x.node.site not in siteList: |
| siteList[x.node.site] = 1 |
| slivercount = len(sliverList) |
| sitecount = len(siteList) |
| except: |
| traceback.print_exc() |
| slivercount = 0 |
| sitecount = 0 |
| |
| userSliceInfo.append({'slicename': slicename, 'sliceid':sliceid, |
| 'sitesUsed':sites_used, |
| 'role': SliceRole.objects.get(id=entry.role.id).role, |
| 'slivercount': slivercount, |
| 'sitecount':sitecount}) |
| |
| return userSliceInfo |
| |
| def getCDNOperatorData(randomizeData = False): |
| return hpc_wizard.get_hpc_wizard().get_sites_for_view() |
| |
| def getPageSummary(request): |
| slice = request.GET.get('slice', None) |
| site = request.GET.get('site', None) |
| node = request.GET.get('node', None) |
| |
| |
| class SimulatorView(View): |
| def get(self, request, **kwargs): |
| sim = json.loads(file("/tmp/simulator.json","r").read()) |
| text = "<html><head></head><body>" |
| text += "Iteration: %d<br>" % sim["iteration"] |
| text += "Elapsed since report %d<br><br>" % sim["elapsed_since_report"] |
| text += "<table border=1>" |
| text += "<tr><th>site</th><th>trend</th><th>weight</th><th>bytes_sent</th><th>hot</th></tr>" |
| for site in sim["site_load"].values(): |
| text += "<tr>" |
| text += "<td>%s</td><td>%0.2f</td><td>%0.2f</td><td>%d</td><td>%0.2f</td>" % \ |
| (site["name"], site["trend"], site["weight"], site["bytes_sent"], site["load_frac"]) |
| text += "</tr>" |
| text += "</table>" |
| text += "</body></html>" |
| return HttpResponse(text) |
| |
| class DashboardUserSiteView(View): |
| def get(self, request, **kwargs): |
| return HttpResponse(json.dumps(getUserSliceInfo(request.user, True)), mimetype='application/javascript') |
| |
| class TenantViewData(View): |
| def get(self, request, **kwargs): |
| return HttpResponse(json.dumps(getTenantSliceInfo(request.user, True)), mimetype='application/javascript') |
| |
| def siteSortKey(site, slice=None, count=None, lat=None, lon=None): |
| # try to pick a site we're already using |
| has_slivers_here=False |
| if slice: |
| for sliver in slice.slivers.all(): |
| if sliver.node.site.name == site.name: |
| has_slivers_here=True |
| |
| # Haversine method |
| d = 0 |
| site_lat = site.location.latitude |
| site_lon = site.location.longitude |
| if lat and lon and site_lat and site_lon: |
| site_lat = float(site_lat) |
| site_lon = float(site_lon) |
| R = 6378.1 |
| 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) |
| c = 2 * math.atan2( math.sqrt(a), math.sqrt(1 - a) ) |
| d = R * c |
| |
| return (-has_slivers_here, d) |
| |
| def tenant_pick_sites(user, user_ip=None, slice=None, count=None): |
| """ Returns list of sites, sorted from most favorable to least favorable """ |
| lat=None |
| lon=None |
| try: |
| client_geo = GeoIP().city(user_ip) |
| if client_geo: |
| lat=float(client_geo["latitude"]) |
| lon=float(client_geo["longitude"]) |
| except: |
| print "exception in geo code" |
| traceback.print_exc() |
| |
| sites = Site.objects.all() |
| sites = [x for x in sites if x.name in BLESSED_SITES] |
| sites = sorted(sites, key=functools.partial(siteSortKey, slice=slice, count=count, lat=lat, lon=lon)) |
| |
| return sites |
| |
| def tenant_increase_slivers(user, user_ip, siteList, slice, count, noAct=False): |
| sitesChanged = {} |
| |
| # let's compute how many slivers are in use in each node of each site |
| for site in siteList: |
| site.nodeList = list(site.nodes.all()) |
| for node in site.nodeList: |
| node.sliverCount = 0 |
| for sliver in node.slivers.all(): |
| if sliver.slice.name == slice.name: |
| node.sliverCount = node.sliverCount +1 |
| |
| # Allocate slivers to nodes |
| # for now, assume we want to allocate all slivers from the same site |
| nodes = siteList[0].nodeList |
| while (count>0): |
| # Sort the node list by number of slivers per node, then pick the |
| # node with the least number of slivers. |
| nodes = sorted(nodes, key=attrgetter("sliverCount")) |
| node = nodes[0] |
| |
| print "adding sliver at node", node.name, "of site", node.site.name |
| |
| if not noAct: |
| sliver = Sliver(name=node.name, |
| slice=slice, |
| node=node, |
| image = Image.objects.all()[0], |
| creator = User.objects.get(email=user), |
| deploymentNetwork=node.deployment, |
| numberCores =1 ) |
| sliver.save() |
| |
| node.sliverCount = node.sliverCount + 1 |
| |
| count = count - 1 |
| |
| sitesChanged[node.site.name] = sitesChanged.get(node.site.name,0) + 1 |
| |
| return sitesChanged |
| |
| def tenant_decrease_slivers(user, siteList, slice, count, noAct=False): |
| sitesChanged = {} |
| |
| if siteList: |
| siteNames = [site.name for site in siteList] |
| else: |
| siteNames = None |
| |
| for sliver in slice.slivers.all(): |
| if (count <= 0): |
| break |
| |
| node = sliver.node |
| if (not siteNames) or (node.site.name in siteNames): |
| print "deleting sliver", sliver, "at node", node.name, "of site", node.site.name |
| if not noAct: |
| sliver.delete() |
| count = count -1 |
| |
| sitesChanged[node.site.name] = sitesChanged.get(node.site.name,0) - 1 |
| |
| return sitesChanged |
| |
| class TenantAddOrRemoveSliverView(View): |
| """ Add or remove slivers from a Slice |
| |
| Arguments: |
| siteName - name of site. If not specified, PlanetStack will pick the |
| best site., |
| actionToDo - [add | rem] |
| count - number of slivers to add or remove |
| sliceName - name of slice |
| noAct - if set, no changes will be made to db, but result will still |
| show which sites would have been modified. |
| |
| Returns: |
| Dictionary of sites that were modified, and the count of nodes |
| that were added or removed at each site. |
| """ |
| def post(self, request, *args, **kwargs): |
| siteName = request.POST.get("siteName", None) |
| actionToDo = request.POST.get("actionToDo", None) |
| count = int(request.POST.get("count","0")) |
| sliceName = request.POST.get("slice", None) |
| noAct = request.POST.get("noAct", False) |
| |
| if not sliceName: |
| return HttpResponseServerError("No slice name given") |
| |
| slice = Slice.objects.get(name=sliceName) |
| |
| if siteName: |
| siteList = [Site.objects.get(name=siteName)] |
| else: |
| siteList = None |
| |
| if (actionToDo == "add"): |
| user_ip = request.GET.get("ip", get_ip(request)) |
| if (siteList is None): |
| siteList = tenant_pick_sites(user, user_ip, slice, count) |
| |
| sitesChanged = tenant_increase_slivers(request.user, user_ip, siteList, slice, count, noAct) |
| elif (actionToDo == "rem"): |
| sitesChanged = tenant_decrease_slivers(request.user, siteList, slice, count, noAct) |
| else: |
| return HttpResponseServerError("Unknown actionToDo %s" % actionToDo) |
| |
| return HttpResponse(json.dumps(sitesChanged), mimetype='application/javascript') |
| |
| def get(self, request, *args, **kwargs): |
| request.POST = request.GET |
| return self.post(request, *args, **kwargs) # for testing REST in browser |
| #return HttpResponseServerError("GET is not supported") |
| |
| class TenantPickSitesView(View): |
| """ primarily just for testing purposes """ |
| def get(self, request, *args, **kwargs): |
| count = request.GET.get("count","0") |
| slice = request.GET.get("slice",None) |
| if slice: |
| slice = Slice.objects.get(name=slice) |
| ip = request.GET.get("ip", get_ip(request)) |
| sites = tenant_pick_sites(request.user, user_ip=ip, count=0, slice=slice) |
| sites = [x.name for x in sites] |
| return HttpResponse(json.dumps(sites), mimetype='application/javascript') |
| |
| class DashboardSummaryAjaxView(View): |
| def get(self, request, **kwargs): |
| return HttpResponse(json.dumps(hpc_wizard.get_hpc_wizard().get_summary_for_view()), mimetype='application/javascript') |
| |
| class DashboardAddOrRemoveSliverView(View): |
| def post(self, request, *args, **kwargs): |
| siteName = request.POST.get("site", "0") |
| actionToDo = request.POST.get("actionToDo", "0") |
| |
| if (actionToDo == "add"): |
| hpc_wizard.get_hpc_wizard().increase_slivers(siteName, 1) |
| elif (actionToDo == "rem"): |
| hpc_wizard.get_hpc_wizard().decrease_slivers(siteName, 1) |
| |
| print '*' * 50 |
| print 'Ask for site: ' + siteName + ' to ' + actionToDo + ' another HPC Sliver' |
| return HttpResponse('This is POST request ') |
| |
| class DashboardAjaxView(View): |
| def get(self, request, **kwargs): |
| return HttpResponse(json.dumps(getCDNOperatorData(True)), mimetype='application/javascript') |
| |
| class DashboardAnalyticsAjaxView(View): |
| def get(self, request, name="hello_world", **kwargs): |
| if (name == "hpcSummary"): |
| return HttpResponse(json.dumps(hpc_wizard.get_hpc_wizard().get_summary_for_view()), mimetype='application/javascript') |
| elif (name == "hpcUserSite"): |
| return HttpResponse(json.dumps(getUserSliceInfo(request.user, True)), mimetype='application/javascript') |
| elif (name == "hpcMap"): |
| return HttpResponse(json.dumps(getCDNOperatorData(True)), mimetype='application/javascript') |
| elif (name == "bigquery"): |
| (mimetype, data) = DoPlanetStackAnalytics(request) |
| return HttpResponse(data, mimetype=mimetype) |
| else: |
| return HttpResponse(json.dumps("Unknown"), mimetype='application/javascript') |
| |