7 sys.path.append("/opt/planetstack")
9 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "planetstack.settings")
10 from openstack.manager import OpenStackManager
11 from core.models import Slice, Sliver, ServiceClass, Reservation, Tag, Network, User, Node, Image, Deployment, Site, NetworkTemplate, NetworkSlice
13 def ps_id_to_pl_id(x):
14 # Since we don't want the PlanetStack object IDs to conflict with existing
15 # PlanetLab object IDs in the CMI, just add 100000 to the PlanetStack object
19 def pl_id_to_ps_id(x):
22 def pl_slice_id(slice):
23 if slice.name == "princeton_vcoblitz":
24 # 70 is the slice id of princeton_vcoblitz on vicci
27 return ps_id_to_pl_id(slice.id)
29 def filter_fields(src, fields):
31 for (key,value) in src.items():
32 if (not fields) or (key in fields):
36 def GetSlices(filter={}):
37 ps_slices = Slice.objects.filter(**filter)
39 for ps_slice in ps_slices:
41 for ps_sliver in ps_slice.slivers.all():
42 node_ids.append(ps_id_to_pl_id(ps_sliver.node.id))
44 slice = {"instantiation": "plc-instantiated",
45 "description": "planetstack slice",
46 "slice_id": pl_slice_id(ps_slice),
50 "peer_slice_id": None,
53 "site_id": ps_id_to_pl_id(ps_slice.site_id),
54 "name": ps_slice.name}
56 # creator_person_id, person_ids, expires, created
61 def GetNodes(node_ids=None, fields=None):
63 ps_nodes = Node.objects.filter(id__in=[pl_id_to_ps_id(nid) for nid in node_ids])
65 ps_nodes = Node.objects.all()
67 for ps_node in ps_nodes:
69 for ps_sliver in ps_node.slivers.all():
70 slice_ids.append(pl_slice_id(ps_sliver.slice))
72 node = {"node_id": ps_id_to_pl_id(ps_node.id),
73 "site_id": ps_id_to_pl_id(ps_node.site_id),
74 "node_type": "regular",
76 "hostname": ps_node.name,
78 "slice_ids": slice_ids,
79 "model": "planetstack",
83 # last_updated, key, boot_state, pcu_ids, node_type, session, last_boot,
84 # interface_ids, slice_ids_whitelist, run_level, ssh_rsa_key, last_pcu_reboot,
85 # nodegroup_ids, verified, last_contact, boot_nonce, version,
86 # last_pcu_configuration, last_download, date_created, ports
90 nodes = [filter_fields(node, fields) for node in nodes]
94 def GetTags(slicename,node_id):
98 ps_sites = Site.objects.all()
100 for ps_site in ps_sites:
102 for ps_slice in ps_site.slices.all():
103 slice_ids.append(pl_slice_id(ps_slice))
106 for ps_node in ps_site.nodes.all():
107 node_ids.append(ps_id_to_pl_id(ps_node.id))
110 site = {"site_id": ps_id_to_pl_id(ps_site.id),
111 "node_ids": node_ids,
116 "peer_site_id": None,
117 "abbrebiated_name": ps_site.abbreviated_name,
119 "name": ps_site.name,
123 "longitude": float(ps_site.location.longitude),
124 "latitude": float(ps_site.location.latitude),
125 "slice_ids": slice_ids,
126 "login_base": ps_site.login_base,
129 # last_updated, ext_consortium_id, person_ids, date_created
135 def GetInterfaces(slicename, node_ids):
137 ps_slices = Slice.objects.filter(name=slicename)
138 for ps_slice in ps_slices:
139 for ps_sliver in ps_slice.slivers.all():
140 node_id = ps_id_to_pl_id(ps_sliver.node_id)
141 if node_id in node_ids:
142 ps_node = ps_sliver.node
143 interface = {"node_id": node_id,
144 "ip": socket.gethostbyname(ps_node.name),
146 "mac": "11:22:33:44:55:66",
153 "interface_tag_ids": [],
154 "interface_id": node_id, # assume each node has only one interface
159 interfaces.append(interface)
162 def GetConfiguration(name):
163 slicename = name["name"]
164 if "node_id" in name:
165 node_id = name["node_id"]
169 node_sliver_tags = GetTags(slicename, node_id)
171 slices = GetSlices({"name": slicename})
178 node_ids = slice['node_ids']
179 nodes = GetNodes(node_ids, ['hostname', 'node_id', 'site_id'])
182 nodemap[node['node_id']]=node['hostname']
184 interfaces = GetInterfaces(slicename, node_ids)
186 for interface in interfaces:
187 if nodemap[interface['node_id']] not in allinterfaces:
188 allinterfaces[nodemap[interface['node_id']]] = []
189 interface['interface_tags'] = []
190 allinterfaces[nodemap[interface['node_id']]].append(interface)
191 if interface['is_primary']:
192 hostipmap[nodemap[interface['node_id']]] = interface['ip']
195 sliver_tags = GetTags(slicename,nid)
196 perhost[nodemap[nid]] = sliver_tags
198 slivers = GetSlices()
200 slivers = [slice for slice in slivers if (node_id in slice.node_ids)]
204 site["site_tags"] = []
206 timestamp = int(time.time())
207 return {'version': 3,
208 'timestamp': timestamp,
209 'configuration': node_sliver_tags,
210 'allconfigurations':perhost,
211 'hostipmap':hostipmap,
213 'interfaces': allinterfaces,
217 if __name__ == '__main__':
221 if ("-d" in sys.argv):
222 config = GetConfiguration({"name": "princeton_coblitz"})
228 for slicename in ["princeton_vcoblitz"]:
229 configs[slicename] = GetConfiguration({"name": slicename})
231 file("planetstack_config","w").write(json.dumps(configs))
232 file("planetstack_slices","w").write(json.dumps(slices))
233 file("planetstack_nodes","w").write(json.dumps(nodes))