GetConfiguration is part of legacyapi
diff --git a/planetstack/tests/GetConfiguration.py b/planetstack/tests/GetConfiguration.py
deleted file mode 100644
index 74d9844..0000000
--- a/planetstack/tests/GetConfiguration.py
+++ /dev/null
@@ -1,258 +0,0 @@
-import os
-import json
-import socket
-import sys
-import time
-
-sys.path.append("/opt/xos")
-
-os.environ.setdefault("DJANGO_SETTINGS_MODULE", "xos.settings")
-from openstack.manager import OpenStackManager
-from core.models import Slice, Sliver, ServiceClass, Reservation, Tag, Network, User, Node, Image, Deployment, Site, NetworkTemplate, NetworkSlice
-
-slice_name_map = {}
-
-def ps_id_to_pl_id(x):
- # Since we don't want the PlanetStack object IDs to conflict with existing
- # PlanetLab object IDs in the CMI, just add 100000 to the PlanetStack object
- # IDs.
- return 100000 + x
-
-def pl_id_to_ps_id(x):
- return x - 100000
-
-def pl_slice_id(slice):
- if slice.name.startswith("princeton_vcoblitz"):
- # 70 is the slice id of princeton_vcoblitz on vicci
- return 70
- else:
- return ps_id_to_pl_id(slice.id)
-
-def ps_slicename_to_pl_slicename(x):
- return slice_name_map.get(x,x)
-
-def filter_fields(src, fields):
- dest = {}
- for (key,value) in src.items():
- if (not fields) or (key in fields):
- dest[key] = value
- return dest
-
-def GetSlices(filter={}):
- ps_slices = Slice.objects.filter(**filter)
- slices = []
- for ps_slice in ps_slices:
- node_ids=[]
- for ps_sliver in ps_slice.slivers.all():
- node_ids.append(ps_id_to_pl_id(ps_sliver.node.id))
-
- slice = {"instantiation": "plc-instantiated",
- "description": "planetstack slice",
- "slice_id": pl_slice_id(ps_slice),
- "node_ids": node_ids,
- "url": "planetstack",
- "max_nodes": 1000,
- "peer_slice_id": None,
- "slice_tag_ids": [],
- "peer_id": None,
- "site_id": ps_id_to_pl_id(ps_slice.site_id),
- "name": ps_slicename_to_pl_slicename(ps_slice.name)}
-
- # creator_person_id, person_ids, expires, created
-
- slices.append(slice)
- return slices
-
-def GetNodes(node_ids=None, fields=None):
- if node_ids:
- ps_nodes = Node.objects.filter(id__in=[pl_id_to_ps_id(nid) for nid in node_ids])
- else:
- ps_nodes = Node.objects.all()
- nodes = []
- for ps_node in ps_nodes:
- slice_ids=[]
- for ps_sliver in ps_node.slivers.all():
- slice_ids.append(pl_slice_id(ps_sliver.slice))
-
- node = {"node_id": ps_id_to_pl_id(ps_node.id),
- "site_id": ps_id_to_pl_id(ps_node.site_id),
- "node_type": "regular",
- "peer_node_id": None,
- "hostname": ps_node.name,
- "conf_file_ids": [],
- "slice_ids": slice_ids,
- "model": "planetstack",
- "peer_id": None,
- "node_tag_ids": []}
-
- # last_updated, key, boot_state, pcu_ids, node_type, session, last_boot,
- # interface_ids, slice_ids_whitelist, run_level, ssh_rsa_key, last_pcu_reboot,
- # nodegroup_ids, verified, last_contact, boot_nonce, version,
- # last_pcu_configuration, last_download, date_created, ports
-
- nodes.append(node)
-
- nodes = [filter_fields(node, fields) for node in nodes]
-
- return nodes
-
-def GetTags(slicename,node_id):
- return {}
-
-def GetSites():
- ps_sites = Site.objects.all()
- sites = []
- for ps_site in ps_sites:
- slice_ids=[]
- for ps_slice in ps_site.slices.all():
- slice_ids.append(pl_slice_id(ps_slice))
-
- node_ids=[]
- for ps_node in ps_site.nodes.all():
- node_ids.append(ps_id_to_pl_id(ps_node.id))
-
-
- site = {"site_id": ps_id_to_pl_id(ps_site.id),
- "node_ids": node_ids,
- "pcu_ids": [],
- "max_slices": 100,
- "max_slivers": 1000,
- "is_public": False,
- "peer_site_id": None,
- "abbrebiated_name": ps_site.abbreviated_name,
- "address_ids": [],
- "name": ps_site.name,
- "url": None,
- "site_tag_ids": [],
- "enabled": True,
- "longitude": float(ps_site.location.longitude),
- "latitude": float(ps_site.location.latitude),
- "slice_ids": slice_ids,
- "login_base": ps_site.login_base,
- "peer_id": None}
-
- # last_updated, ext_consortium_id, person_ids, date_created
-
- sites.append(site)
-
- return sites
-
-def GetInterfaces(slicename, node_ids):
- interfaces = []
- ps_slices = Slice.objects.filter(name=slicename)
- for ps_slice in ps_slices:
- for ps_sliver in ps_slice.slivers.all():
- node_id = ps_id_to_pl_id(ps_sliver.node_id)
- if node_id in node_ids:
- ps_node = ps_sliver.node
- interface = {"node_id": node_id,
- "ip": socket.gethostbyname(ps_node.name),
- "broadcast": None,
- "mac": "11:22:33:44:55:66",
- "bwlimit": None,
- "network": None,
- "is_primary": True,
- "dns1": None,
- "hostname": None,
- "netmask": None,
- "interface_tag_ids": [],
- "interface_id": node_id, # assume each node has only one interface
- "gateway": None,
- "dns2": None,
- "type": "ipv4",
- "method": "dhcp"}
- interfaces.append(interface)
- return interfaces
-
-def find_multi_slicename(orig_slicename):
- """
- Because we sometimes have issues deleting a slice in planetstack and
- creating a new one, allow us to use a prefix match, that way someone
- can put a version number of the end of the slicename
- """
- global slice_name_map
- slices = Slice.objects.filter()
- for slice in slices:
- if slice.name.startswith(orig_slicename):
- slice_name_map[slice.name] = orig_slicename
- return slice.name
-
- return orig_slicename
-
-
-def GetConfiguration(name):
- slicename = name["name"]
- if "node_id" in name:
- node_id = name["node_id"]
- else:
- node_id = 0
-
- slicename = find_multi_slicename(slicename)
-
- node_sliver_tags = GetTags(slicename, node_id)
-
- slices = GetSlices({"name": slicename})
- perhost = {}
- allinterfaces = {}
- hostipmap = {}
- nodes = []
- if len(slices)==1:
- slice = slices[0]
- node_ids = slice['node_ids']
- nodes = GetNodes(node_ids, ['hostname', 'node_id', 'site_id'])
- nodemap = {}
- for node in nodes:
- nodemap[node['node_id']]=node['hostname']
-
- interfaces = GetInterfaces(slicename, node_ids)
- hostipmap = {}
- for interface in interfaces:
- if nodemap[interface['node_id']] not in allinterfaces:
- allinterfaces[nodemap[interface['node_id']]] = []
- interface['interface_tags'] = []
- allinterfaces[nodemap[interface['node_id']]].append(interface)
- if interface['is_primary']:
- hostipmap[nodemap[interface['node_id']]] = interface['ip']
-
- for nid in node_ids:
- sliver_tags = GetTags(slicename,nid)
- perhost[nodemap[nid]] = sliver_tags
-
- slivers = GetSlices()
- if node_id != 0:
- slivers = [slice for slice in slivers if (node_id in slice.node_ids)]
-
- sites = GetSites()
- for site in sites:
- site["site_tags"] = []
-
- timestamp = int(time.time())
- return {'version': 3,
- 'timestamp': timestamp,
- 'configuration': node_sliver_tags,
- 'allconfigurations':perhost,
- 'hostipmap':hostipmap,
- 'slivers': slivers,
- 'interfaces': allinterfaces,
- 'sites': sites,
- 'nodes': nodes}
-
-if __name__ == '__main__':
- find_multi_slicename("princeton_vcoblitz") # set up the mapping for princeton_vcoblitz2 -> princeton_vcoblitz
-
- slices = GetSlices()
- nodes = GetNodes()
-
- if ("-d" in sys.argv):
- config = GetConfiguration({"name": "princeton_vcoblitz"})
- print config
- print slices
- print nodes
- else:
- configs={}
- for slicename in ["princeton_vcoblitz"]:
- configs[slicename] = GetConfiguration({"name": slicename})
-
- file("xos.config","w").write(json.dumps(configs))
- file("planetstack_slices","w").write(json.dumps(slices))
- file("planetstack_nodes","w").write(json.dumps(nodes))