Scott Baker | b3fc6c0 | 2014-06-30 16:38:46 -0700 | [diff] [blame] | 1 | # /opt/planetstack/core/dashboard/views/helloworld.py |
| 2 | import datetime |
| 3 | import os |
| 4 | import sys |
| 5 | import time |
| 6 | import json |
| 7 | from django.http import HttpResponse, HttpResponseServerError, HttpResponseForbidden |
| 8 | from django.views.generic import TemplateView, View |
| 9 | from core.models import * |
| 10 | from django.forms.models import model_to_dict |
| 11 | |
| 12 | def ensure_serializable(d): |
| 13 | d2={} |
| 14 | for (k,v) in d.items(): |
| 15 | # datetime is not json serializable |
| 16 | if isinstance(v, datetime.datetime): |
| 17 | d2[k] = time.mktime(v.timetuple()) |
| 18 | elif v.__class__.__name__ == "Geoposition": |
| 19 | pass |
| 20 | else: |
| 21 | d2[k] = v |
| 22 | return d2 |
| 23 | |
| 24 | def sliver_to_dict(sliver): |
| 25 | d = model_to_dict(sliver) |
| 26 | d["slice_id"] = sliver.slice.id |
| 27 | d["node_id"] = sliver.node.id |
| 28 | return d |
| 29 | |
| 30 | def slice_to_dict(slice): |
| 31 | d = model_to_dict(slice) |
| 32 | d["slivers"] = [sliver_to_dict(x) for x in slice.slivers] |
| 33 | return d |
| 34 | |
| 35 | def node_to_dict(node): |
| 36 | d = model_to_dict(node) |
| 37 | d["slivers"] = [] |
| 38 | |
| 39 | |
| 40 | class OpenCloudData: |
| 41 | def __init__(self, user): |
| 42 | self.loadAll() |
| 43 | |
| 44 | def loadAll(self): |
| 45 | self.allNodes = list(Node.objects.all()) |
| 46 | self.allSlices = list(Slice.objects.all()) |
| 47 | self.allSlivers = list(Sliver.objects.all()) |
| 48 | self.allSites = list(Site.objects.all()) |
| 49 | |
| 50 | self.site_id = {} |
| 51 | for site in self.allSites: |
| 52 | d = model_to_dict(site) |
| 53 | d["node_ids"] = [] |
| 54 | d["slice_ids"] = [] |
| 55 | self.site_id[site.id] = ensure_serializable(d) |
| 56 | |
| 57 | self.node_id = {} |
| 58 | for node in self.allNodes: |
| 59 | d = model_to_dict(node) |
| 60 | d["sliver_ids"] = [] |
| 61 | self.node_id[node.id] = ensure_serializable(d) |
| 62 | self.site_id[node.site_id]["node_ids"].append(node.id) |
| 63 | |
| 64 | self.slice_id = {} |
| 65 | for slice in self.allSlices: |
| 66 | d = model_to_dict(slice) |
| 67 | d["sliver_ids"] = [] |
| 68 | self.slice_id[slice.id] = ensure_serializable(d) |
| 69 | self.site_id[slice.site_id]["slice_ids"].append(site.id) |
| 70 | |
| 71 | self.sliver_id = {} |
| 72 | for sliver in self.allSlivers: |
| 73 | self.sliver_id[sliver.id] = model_to_dict(sliver) |
| 74 | |
| 75 | self.slice_id[sliver.slice_id]["sliver_ids"].append(sliver.id) |
| 76 | self.node_id[sliver.node_id]["sliver_ids"].append(sliver.id) |
| 77 | |
| 78 | def get_opencloud_data(self): |
| 79 | return {"slices": self.slice_id.values(), |
| 80 | "slivers": self.sliver_id.values(), |
| 81 | "nodes": self.node_id.values(), |
| 82 | "sites": self.site_id.values()} |
| 83 | |
| 84 | class ShellDataView(View): |
| 85 | url = r'^shelldata/' |
| 86 | |
| 87 | def get(self, request, **kwargs): |
| 88 | result = OpenCloudData(request.user).get_opencloud_data() |
| 89 | |
| 90 | return HttpResponse(json.dumps(result), mimetype='application/json') |