sfa/rspecs/versions/sfav1.py

   1 from __future__ import print_function
   2
   3 from copy import deepcopy
   4 from lxml import etree
   5
   6 from sfa.util.sfalogging import logger
   7 from sfa.util.xrn import hrn_to_urn, urn_to_hrn
   8 from sfa.rspecs.version import RSpecVersion
   9 from sfa.rspecs.elements.element import Element
  10 from sfa.rspecs.elements.versions.pgv2Link import PGv2Link
  11 from sfa.rspecs.elements.versions.sfav1Node import SFAv1Node
  12 from sfa.rspecs.elements.versions.sfav1Sliver import SFAv1Sliver
  13 from sfa.rspecs.elements.versions.sfav1Lease import SFAv1Lease
  14
  15 class SFAv1(RSpecVersion):
  16     enabled = True
  17     type = 'SFA'
  18     content_type = '*'
  19     version = '1'
  20     schema = ''
  21     namespace = ''
  22     extensions = {}
  23     namespaces = None
  24     template = '<RSpec type="%s"></RSpec>' % type
  25
  26     # Network
  27     def get_networks(self):
  28         network_elems = self.xml.xpath('//network')
  29         networks = [network_elem.get_instance(fields=['name', 'slice']) for \
  30                     network_elem in network_elems]
  31         return networks
  32
  33
  34     def add_network(self, network):
  35         network_tags = self.xml.xpath('//network[@name="%s"]' % network)
  36         if not network_tags:
  37             network_tag = self.xml.add_element('network', name=network)
  38         else:
  39             network_tag = network_tags[0]
  40         return network_tag
  41
  42
  43     # Nodes
  44
  45     def get_nodes(self, filter=None):
  46         return SFAv1Node.get_nodes(self.xml, filter)
  47
  48     def get_nodes_with_slivers(self):
  49         return SFAv1Node.get_nodes_with_slivers(self.xml)
  50
  51     def add_nodes(self, nodes, network = None, no_dupes=False, rspec_content_type=None):
  52         SFAv1Node.add_nodes(self.xml, nodes, rspec_content_type)
  53
  54     def merge_node(self, source_node_tag, network, no_dupes=False):
  55         if no_dupes and self.get_node_element(node['hostname']):
  56             # node already exists
  57             return
  58
  59         network_tag = self.add_network(network)
  60         network_tag.append(deepcopy(source_node_tag))
  61
  62     # Slivers
  63
  64     def add_slivers(self, hostnames, attributes=None, sliver_urn=None, append=False):
  65         if attributes is None: attributes=[]
  66         # add slice name to network tag
  67         network_tags = self.xml.xpath('//network')
  68         if network_tags:
  69             network_tag = network_tags[0]
  70             network_tag.set('slice', urn_to_hrn(sliver_urn)[0])
  71
  72         # add slivers
  73         sliver = {'name':sliver_urn,
  74                   'pl_tags': attributes}
  75         for hostname in hostnames:
  76             if sliver_urn:
  77                 sliver['name'] = sliver_urn
  78             node_elems = self.get_nodes({'component_id': '*%s*' % hostname})
  79             if not node_elems:
  80                 continue
  81             node_elem = node_elems[0]
  82             SFAv1Sliver.add_slivers(node_elem.element, sliver)
  83
  84         # remove all nodes without slivers
  85         if not append:
  86             for node_elem in self.get_nodes():
  87                 if not node_elem['slivers']:
  88                     parent = node_elem.element.getparent()
  89                     parent.remove(node_elem.element)
  90
  91
  92     def remove_slivers(self, slivers, network=None, no_dupes=False):
  93         SFAv1Node.remove_slivers(self.xml, slivers)
  94
  95     def get_slice_attributes(self, network=None):
  96         attributes = []
  97         nodes_with_slivers = self.get_nodes_with_slivers()
  98         for default_attribute in self.get_default_sliver_attributes(network):
  99             attribute = default_attribute.copy()
 100             attribute['node_id'] = None
 101             attributes.append(attribute)
 102         for node in nodes_with_slivers:
 103             nodename=node['component_name']
 104             sliver_attributes = self.get_sliver_attributes(nodename, network)
 105             for sliver_attribute in sliver_attributes:
 106                 sliver_attribute['node_id'] = nodename
 107                 attributes.append(sliver_attribute)
 108         return attributes
 109
 110
 111     def add_sliver_attribute(self, component_id, name, value, network=None):
 112         nodes = self.get_nodes({'component_id': '*%s*' % component_id})
 113         if nodes is not None and isinstance(nodes, list) and len(nodes) > 0:
 114             node = nodes[0]
 115             slivers = SFAv1Sliver.get_slivers(node)
 116             if slivers:
 117                 sliver = slivers[0]
 118                 SFAv1Sliver.add_sliver_attribute(sliver, name, value)
 119         else:
 120             # should this be an assert / raise an exception?
 121             logger.error("WARNING: failed to find component_id %s" % component_id)
 122
 123     def get_sliver_attributes(self, component_id, network=None):
 124         nodes = self.get_nodes({'component_id': '*%s*' % component_id})
 125         attribs = []
 126         if nodes is not None and isinstance(nodes, list) and len(nodes) > 0:
 127             node = nodes[0]
 128             slivers = SFAv1Sliver.get_slivers(node.element)
 129             if slivers is not None and isinstance(slivers, list) and len(slivers) > 0:
 130                 sliver = slivers[0]
 131                 attribs = SFAv1Sliver.get_sliver_attributes(sliver.element)
 132         return attribs
 133
 134     def remove_sliver_attribute(self, component_id, name, value, network=None):
 135         attribs = self.get_sliver_attributes(component_id)
 136         for attrib in attribs:
 137             if attrib['name'] == name and attrib['value'] == value:
 138                 #attrib.element.delete()
 139                 parent = attrib.element.getparent()
 140                 parent.remove(attrib.element)
 141
 142     def add_default_sliver_attribute(self, name, value, network=None):
 143         if network:
 144             defaults = self.xml.xpath("//network[@name='%s']/sliver_defaults" % network)
 145         else:
 146             defaults = self.xml.xpath("//sliver_defaults")
 147         if not defaults:
 148             if network:
 149                 network_tag = self.xml.xpath("//network[@name='%s']" % network)
 150             else:
 151                 network_tag = self.xml.xpath("//network")
 152             if isinstance(network_tag, list):
 153                 network_tag = network_tag[0]
 154             defaults = network_tag.add_element('sliver_defaults')
 155         elif isinstance(defaults, list):
 156             defaults = defaults[0]
 157         SFAv1Sliver.add_sliver_attribute(defaults, name, value)
 158
 159     def get_default_sliver_attributes(self, network=None):
 160         if network:
 161             defaults = self.xml.xpath("//network[@name='%s']/sliver_defaults" % network)
 162         else:
 163             defaults = self.xml.xpath("//sliver_defaults")
 164         if not defaults: return []
 165         return SFAv1Sliver.get_sliver_attributes(defaults[0])
 166
 167     def remove_default_sliver_attribute(self, name, value, network=None):
 168         attribs = self.get_default_sliver_attributes(network)
 169         for attrib in attribs:
 170             if attrib['name'] == name and attrib['value'] == value:
 171                 #attrib.element.delete()
 172                 parent = attrib.element.getparent()
 173                 parent.remove(attrib.element)
 174
 175     # Links
 176
 177     def get_links(self, network=None):
 178         return PGv2Link.get_links(self.xml)
 179
 180     def get_link_requests(self):
 181         return PGv2Link.get_link_requests(self.xml)
 182
 183     def add_links(self, links):
 184         networks = self.get_networks()
 185         if len(networks) > 0:
 186             xml = networks[0].element
 187         else:
 188             xml = self.xml
 189         PGv2Link.add_links(xml, links)
 190
 191     def add_link_requests(self, links):
 192         PGv2Link.add_link_requests(self.xml, links)
 193
 194     # utility
 195
 196     def merge(self, in_rspec):
 197         """
 198         Merge contents for specified rspec with current rspec
 199         """
 200
 201         if not in_rspec:
 202             return
 203
 204         from sfa.rspecs.rspec import RSpec
 205         if isinstance(in_rspec, RSpec):
 206             rspec = in_rspec
 207         else:
 208             rspec = RSpec(in_rspec)
 209         if rspec.version.type.lower() == 'protogeni':
 210             from sfa.rspecs.rspec_converter import RSpecConverter
 211             in_rspec = RSpecConverter.to_sfa_rspec(rspec.toxml())
 212             rspec = RSpec(in_rspec)
 213
 214         # just copy over all networks
 215         current_networks = self.get_networks()
 216         networks = rspec.version.get_networks()
 217         for network in networks:
 218             current_network = network.get('name')
 219             if current_network and current_network not in current_networks:
 220                 self.xml.append(network.element)
 221                 current_networks.append(current_network)
 222
 223     # Leases
 224
 225     def get_leases(self, filter=None):
 226         return SFAv1Lease.get_leases(self.xml, filter)
 227
 228     def add_leases(self, leases, network = None, no_dupes=False):
 229         SFAv1Lease.add_leases(self.xml, leases)
 230
 231     # Spectrum
 232
 233     def get_channels(self, filter=None):
 234         return []
 235
 236     def add_channels(self, channels, network = None, no_dupes=False):
 237         pass
 238
 239 if __name__ == '__main__':
 240     from sfa.rspecs.rspec import RSpec
 241     from sfa.rspecs.rspec_elements import *
 242     r = RSpec('/tmp/resources.rspec')
 243     r.load_rspec_elements(SFAv1.elements)
 244     print(r.get(RSpecElements.NODE))