4 # import modules used here -- sys is a very standard one
10 from sfa.rspecs.version_manager import VersionManager
11 from sfa.senslab.OARrestapi import *
12 from sfa.senslab.slabdriver import SlabDriver
13 from sfa.util.config import Config
14 from sfa.util.xrn import hrn_to_urn, urn_to_hrn, urn_to_sliver_id
15 from sfa.util.plxrn import PlXrn, hostname_to_urn, hrn_to_pl_slicename
17 from sfa.rspecs.rspec import RSpec
18 from sfa.rspecs.elements.location import Location
19 from sfa.rspecs.elements.hardware_type import HardwareType
20 from sfa.rspecs.elements.node import Node
21 #from sfa.rspecs.elements.link import Link
22 #from sfa.rspecs.elements.sliver import Sliver
23 #from sfa.rspecs.elements.login import Login
24 #from sfa.rspecs.elements.location import Location
25 #from sfa.rspecs.elements.interface import Interface
26 #from sfa.rspecs.elements.services import Services
27 #from sfa.rspecs.elements.pltag import PLTag
28 from sfa.util.topology import Topology
29 from sfa.rspecs.version_manager import VersionManager
30 #from sfa.plc.vlink import get_tc_rate
31 from sfa.util.sfatime import epochparse
33 def hostname_to_hrn(root_auth,login_base,hostname):
34 return PlXrn(auth=root_auth,hostname=login_base+'_'+hostname).get_hrn()
50 def __init__(self ,api):
51 self.OARImporter = OARapi()
52 self.driver = SlabDriver(Config())
54 print >>sys.stderr,"\r\n \r\n \t\t_____________INIT Slabaggregate api : %s" %(api)
61 #if slice and 'node_ids' in slice and slice['node_ids']:
62 #filter['node_id'] = slice['node_ids']
63 #tags_filter=filter.copy()
65 #filter.update({'peer_id': None})
66 nodes = self.driver.GetNodes(filtre)
73 #site_ids.append(node['site_id'])
74 #interface_ids.extend(node['interface_ids'])
75 #tag_ids.extend(node['node_tag_ids'])
76 nodes_dict[node['node_id']] = node
79 #sites_dict = self.get_sites({'site_id': site_ids})
81 #interfaces = self.get_interfaces({'interface_id':interface_ids})
83 #node_tags = self.get_node_tags(tags_filter)
85 #pl_initscripts = self.get_pl_initscripts()
87 #links = self.get_links(sites_dict, nodes_dict, interfaces)
91 # skip whitelisted nodes
92 #if node['slice_ids_whitelist']:
93 #if not slice or slice['slice_id'] not in node['slice_ids_whitelist']:
95 node['hostname'] = hostname_to_hrn( self.driver.root_auth,node['site_login_base'], node['hostname'])
97 # xxx how to retrieve site['login_base']
98 #site_id=node['site_id']
99 #site=sites_dict[site_id]
101 rspec_node['component_id'] = hostname_to_urn(self.driver.root_auth, node['site_login_base'], node['hostname'])
102 rspec_node['component_name'] = node['hostname']
103 rspec_node['component_manager_id'] = hrn_to_urn(self.driver.root_auth, 'authority+sa')
104 rspec_node['authority_id'] = hrn_to_urn(PlXrn.site_hrn(self.driver.root_auth, node['site_login_base']), 'authority+sa')
105 rspec_node['boot_state'] = node['boot_state']
106 if node['posx'] and node['posy']:
107 location = Location({'longitude':node['posx'], 'latitude': node['posy']})
108 rspec_node['location'] = location
110 rspec_node['exclusive'] = 'True'
111 rspec_node['hardware_types']= [HardwareType({'name': 'senslab sensor node'})]
112 # only doing this because protogeni rspec needs
113 # to advertise available initscripts
114 #rspec_node['pl_initscripts'] = pl_initscripts.values()
115 # add site/interface info to nodes.
116 # assumes that sites, interfaces and tags have already been prepared.
117 #site = sites_dict[node['site_id']]
118 #if site['longitude'] and site['latitude']:
119 #location = Location({'longitude': site['longitude'], 'latitude': site['latitude']})
120 #rspec_node['location'] = location
121 rspec_node['interfaces'] = []
123 #for if_id in node['interface_ids']:
124 #interface = Interface(interfaces[if_id])
125 #interface['ipv4'] = interface['ip']
126 #interface['component_id'] = PlXrn(auth=self.api.hrn, interface='node%s:eth%s' % (node['node_id'], if_count)).get_urn()
127 #rspec_node['interfaces'].append(interface)
130 #tags = [PLTag(node_tags[tag_id]) for tag_id in node['node_tag_ids']]
131 rspec_node['tags'] = []
132 #if node['node_id'] in slivers:
134 #sliver = slivers[node['node_id']]
135 #rspec_node['sliver_id'] = sliver['sliver_id']
136 #rspec_node['client_id'] = node['hostname']
137 #rspec_node['slivers'] = [sliver]
139 ## slivers always provide the ssh service
140 #login = Login({'authentication': 'ssh-keys', 'hostname': node['hostname'], 'port':'22'})
141 #service = Services({'login': login})
142 #rspec_node['services'] = [service]
143 rspec_nodes.append(rspec_node)
146 #from plc/aggregate.py
147 def get_rspec(self, slice_xrn=None, version = None, options={}):
148 print>>sys.stderr, " \r\n SlabAggregate \t\t get_rspec **************\r\n"
152 version_manager = VersionManager()
153 version = version_manager.get_version(version)
157 rspec_version = version_manager._get_version(version.type, version.version, 'ad')
159 rspec_version = version_manager._get_version(version.type, version.version, 'manifest')
161 rspec = RSpec(version=rspec_version, user_options=options)
163 nodes = self.get_nodes()
164 rspec.version.add_nodes(nodes)
166 print >>sys.stderr, 'after add_nodes '