Scott Baker | 0e8a2d7 | 2013-08-30 19:00:11 -0700 | [diff] [blame^] | 1 | import os |
| 2 | import json |
| 3 | import sys |
| 4 | import time |
| 5 | |
| 6 | sys.path.append("/opt/planetstack") |
| 7 | |
| 8 | os.environ.setdefault("DJANGO_SETTINGS_MODULE", "planetstack.settings") |
| 9 | from openstack.manager import OpenStackManager |
| 10 | from core.models import Slice, Sliver, ServiceClass, Reservation, Tag, Network, User, Node, Image, Deployment, Site, NetworkTemplate, NetworkSlice |
| 11 | |
| 12 | def ps_id_to_pl_id(x): |
| 13 | return 10000 + x |
| 14 | |
| 15 | def filter_fields(src, fields): |
| 16 | dest = {} |
| 17 | for (key,value) in src.items(): |
| 18 | if (not fields) or (key in fields): |
| 19 | dest[key] = value |
| 20 | return dest |
| 21 | |
| 22 | def GetSlices(filter={}): |
| 23 | ps_slices = Slice.objects.filter(**filter) |
| 24 | slices = [] |
| 25 | for ps_slice in ps_slices: |
| 26 | node_ids=[] |
| 27 | for ps_sliver in ps_slice.slivers.all(): |
| 28 | node_ids.append(ps_id_to_pl_id(ps_sliver.node.id)) |
| 29 | |
| 30 | slice = {"instantiation": "plc-instantiated", |
| 31 | "description": "planetstack slice", |
| 32 | "slice_id": ps_id_to_pl_id(ps_slice.id), |
| 33 | "node_ids": node_ids, |
| 34 | "url": "planetstack", |
| 35 | "max_nodes": 1000, |
| 36 | "peer_slice_id": None, |
| 37 | "slice_tag_ids": [], |
| 38 | "peer_id": None, |
| 39 | "site_id": ps_id_to_pl_id(ps.site_id), |
| 40 | "name": ps_slice.name} |
| 41 | |
| 42 | # creator_person_id, person_ids, expires, created |
| 43 | |
| 44 | slices.append(slice) |
| 45 | return slices |
| 46 | |
| 47 | def GetNodes(node_ids, fields=None) |
| 48 | ps_nodes = Node.objects.filter(id__in=node_ids) |
| 49 | nodes = [] |
| 50 | for ps_node in ps_nodes: |
| 51 | slice_ids=[] |
| 52 | for ps_sliver in ps_node.slivers.all(): |
| 53 | slice_ids.append(ps_id_to_pl_id(ps_sliver.slice.id)) |
| 54 | |
| 55 | node = {"node_id": ps_id_to_pl_id(ps_node.id), |
| 56 | "site_id": ps_id_to_pl_id(ps_node.site_id), |
| 57 | "node_type": "regular", |
| 58 | "peer_node_id": None, |
| 59 | "hostname": node.name, |
| 60 | "conf_file_ids": [], |
| 61 | "slice_ids": slice_ids, |
| 62 | "model": "planetstack", |
| 63 | "peer_id": None, |
| 64 | "node_tag_ids": []} |
| 65 | |
| 66 | # last_updated, key, boot_state, pcu_ids, node_type, session, last_boot, |
| 67 | # interface_ids, slice_ids_whitelist, run_level, ssh_rsa_key, last_pcu_reboot, |
| 68 | # nodegroup_ids, verified, last_contact, boot_nonce, version, |
| 69 | # last_pcu_configuration, last_download, date_created, ports |
| 70 | |
| 71 | nodes.append(node) |
| 72 | |
| 73 | nodes = [filter_fields(node, fields) for node in nodes] |
| 74 | |
| 75 | return nodes |
| 76 | |
| 77 | def GetTags(slicename,node_id): |
| 78 | return [] |
| 79 | |
| 80 | def GetSites(): |
| 81 | ps_sites = Site.objects.all() |
| 82 | sites = [] |
| 83 | for ps_site in ps_sites: |
| 84 | slice_ids=[] |
| 85 | for ps_slice in ps_site.slices.all(): |
| 86 | slice_ids.append(ps_id_to_pl_id(ps_slice.id)) |
| 87 | |
| 88 | node_ids=[] |
| 89 | for ps_node in ps_site.nodes.all(): |
| 90 | node_ids.append(ps_id_to_pl_id(ps_node.id)) |
| 91 | |
| 92 | |
| 93 | site = {"site_id": ps_id_to_pl_id(ps_site.id), |
| 94 | "node_ids": node_ids, |
| 95 | "pcu_ids": [], |
| 96 | "max_slices": 100, |
| 97 | "max_slivers": 1000, |
| 98 | "is_public": False, |
| 99 | "peer_site_id": None, |
| 100 | "abbrebiated_name": ps_site.abbreviated_name, |
| 101 | "address_ids": [], |
| 102 | "name": ps_site.name, |
| 103 | "url": None, |
| 104 | "site_tag_ids": [], |
| 105 | "enabled": True, |
| 106 | "longitude": float(ps_site.location.longitude), |
| 107 | "latitude": float(ps_site.location.latitude), |
| 108 | "slice_ids": slice_ids, |
| 109 | "login_base": ps_site.login_base, |
| 110 | "peer_id": None} |
| 111 | |
| 112 | # last_updated, ext_consortium_id, person_ids, date_created |
| 113 | |
| 114 | sites.append(site) |
| 115 | |
| 116 | return sites |
| 117 | |
| 118 | |
| 119 | def GetConfigurations(name): |
| 120 | slicename = name["name"] |
| 121 | if "node_id" in name: |
| 122 | node_id = name["node_id"] |
| 123 | else: |
| 124 | node_id = 0 |
| 125 | |
| 126 | slices = GetSlices({"name": slicename}) |
| 127 | perhost = {} |
| 128 | allinterfaces = {} |
| 129 | hostipmap = {} |
| 130 | nodes = [] |
| 131 | if len(slices)==1: |
| 132 | slice = slices[0] |
| 133 | node_ids = slice['node_ids'] |
| 134 | nodes = GetNodes(node_ids, ['hostname', 'node_id', 'site_id']) |
| 135 | nodemap = {} |
| 136 | for node in nodes: |
| 137 | nodemap[node['node_id']]=node['hostname'] |
| 138 | |
| 139 | # interfaces |
| 140 | |
| 141 | for nid in node_ids: |
| 142 | sliver_tags = GetTags(slicename,nid) |
| 143 | perhost[nodemap[nid]] = sliver_tags |
| 144 | |
| 145 | slivers = GetSlices() |
| 146 | if node_id != 0: |
| 147 | slivers = [slice for slice in slivers if (node_id in slice.node_ids)] |
| 148 | |
| 149 | sites = GetSites() |
| 150 | for site in sites: |
| 151 | site["site_tags"] = [] |
| 152 | |
| 153 | timestamp = int(time.time()) |
| 154 | return {'version': 3, |
| 155 | 'timestamp': timestamp, |
| 156 | 'configuration': node_sliver_tags, |
| 157 | 'allconfigurations':perhost, |
| 158 | 'hostipmap':hostipmap, |
| 159 | 'slivers': slivers, |
| 160 | 'interfaces': allinterfaces, |
| 161 | 'sites': sites, |
| 162 | 'nodes': nodes} |
| 163 | |
| 164 | |
| 165 | |