from sfa.util.plxrn import *
from sfa.rspecs.sfa_rspec import SfaRSpec
from sfa.rspecs.pg_rspec import PGRSpec
+from sfa.rspecs.rspec_version import RSpecVersion
class Aggregate:
nodes = {}
interfaces = {}
links = {}
- tag_types = {}
+ node_tags = {}
prepared=False
+ #panos new user options variable
+ user_options = {}
- def __init__(self, api):
+ def __init__(self, api, user_options={}):
self.api = api
+ self.user_options = user_options
def prepare_sites(self, force=False):
if not self.sites or force:
if not self.links or force:
pass
- def prepare_tagtypes(self, force=False):
- if not self.tag_types or force:
- for tag_type in self.api.plshell.GetTagTypes(self.api.plauth):
- self.tag_types[tag_type['tag_type_id']] = tag_type
+ def prepare_node_tags(self, force=False):
+ if not self.node_tags or force:
+ for node_tag in self.api.plshell.GetNodeTags(self.api.plauth):
+ self.node_tags[node_tag['node_tag_id']] = node_tag
def prepare(self, force=False):
if not self.prepared or force:
self.prepare_nodes(force)
self.prepare_interfaces(force)
self.prepare_links(force)
- self.prepare_tagtypes(force)
+ self.prepare_node_tags(force)
# add site/interface info to nodes
- for node in self.nodes:
+ for node_id in self.nodes:
+ node = self.nodes[node_id]
site = self.sites[node['site_id']]
interfaces = [self.interfaces[interface_id] for interface_id in node['interface_ids']]
+ tags = [self.node_tags[tag_id] for tag_id in node['node_tag_ids']]
+ node['network'] = self.api.hrn
+ node['network_urn'] = hrn_to_urn(self.api.hrn, 'authority+sa')
+ node['urn'] = hostname_to_urn(self.api.hrn, site['login_base'], node['hostname'])
+ node['site_urn'] = hrn_to_urn(PlXrn.site_hrn(self.api.hrn, site['login_base']), 'authority+am')
+ node['site'] = site
+ node['interfaces'] = interfaces
+ node['tags'] = tags
self.prepared = True
- def get_rspec(self, slice_xrn=None, format='sfa'):
+ def get_rspec(self, slice_xrn=None, version = None, type=None):
self.prepare()
rspec = None
- if format == ['pg']:
- rspec = PGRSpec()
+ rspec_version = RSpecVersion(version)
+ if rspec_version['type'].lower() == 'protogeni':
+ rspec = PGRSpec(type=type)
+ #panos pass user options to SfaRSpec
+ elif rspec_version['type'].lower() == 'sfa':
+ rspec = SfaRSpec(type=type, user_options=self.user_options)
else:
- rspec = SfaRSpec()
+ rspec = SfaRSpec(type=type, user_options=self.user_options)
- rspec.add_nodes(self.nodes)
- rspec.add_interfaces(self.interfaces)
- rspec.add_links(self.links)
+
+ rspec.add_nodes(self.nodes.values())
+ rspec.add_interfaces(self.interfaces.values())
+ rspec.add_links(self.links.values())
if slice_xrn:
+ # get slice details
slice_hrn, _ = urn_to_hrn(slice_xrn)
slice_name = hrn_to_pl_slicename(slice_hrn)
slices = self.api.plshell.GetSlices(self.api.plauth, slice_name)
if slices:
- slice = slices[0]
- hostnames = [self.nodes[node_id]['hostname'] for node_id in slice['node_ids']]
- rspec.add_slivers(hostnames)
+ slice = slices[0]
+ slivers = []
+ tags = self.api.plshell.GetSliceTags(self.api.plauth, slice['slice_tag_ids'])
+ for node_id in slice['node_ids']:
+ sliver = {}
+ sliver['hostname'] = self.nodes[node_id]['hostname']
+ sliver['tags'] = []
+ slivers.append(sliver)
+ for tag in tags:
+ # if tag isn't bound to a node then it applies to all slivers
+ if not tag['node_id']:
+ sliver['tags'].append(tag)
+ else:
+ tag_host = self.nodes[tag['node_id']]['hostname']
+ if tag_host == sliver['hostname']:
+ sliver.tags.append(tag)
+ rspec.add_slivers(slivers)
+
+ return rspec.toxml()