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:
links = {}
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:
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['network_urn'] = hrn_to_urn(self.api.hrn, 'authority+am')
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')
+ node['site_urn'] = hrn_to_urn(PlXrn.site_hrn(self.api.hrn, site['login_base']), 'authority+sa')
node['site'] = site
node['interfaces'] = interfaces
node['tags'] = tags
def get_rspec(self, slice_xrn=None, version = None):
self.prepare()
rspec = None
- if version:
- format = version.format
- if format == 'pg':
- rspec = PGRSpec()
- else:
- rspec = SfaRSpec()
- else:
- rspec = SfaRSpec()
+ rspec_version = RSpecVersion(version)
+ if slice_xrn:
+ type = 'manifest'
+ else:
+ type = 'advertisement'
+ if rspec_version['type'].lower() == 'protogeni':
+ rspec = PGRSpec(type=type)
+ elif rspec_version['type'].lower() == 'sfa':
+ rspec = SfaRSpec(type=type, user_options=self.user_options)
+ else:
+ rspec = SfaRSpec(type=type, user_options=self.user_options)
+
rspec.add_nodes(self.nodes.values())
rspec.add_interfaces(self.interfaces.values())
rspec.add_links(self.links.values())
if slice_xrn:
- # get slice details
+ # If slicename is specified then resulting rspec is a manifest.
+ # Add sliver details to rspec and remove 'advertisement' elements
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)
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']:
else:
tag_host = self.nodes[tag['node_id']]['hostname']
if tag_host == sliver['hostname']:
- sliver.tags.append(tag)
- rspec.add_slivers(slivers)
+ sliver['tags'].append(tag)
+ rspec.add_slivers(slivers, sliver_urn=slice_xrn)
- return rspec.toxml()
+ return rspec.toxml(cleanup=True)