sfa/rspecs/versions/sfav1.py

   1
   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
  16 class SFAv1(RSpecVersion):
  17     enabled = True
  18     type = 'SFA'
  19     content_type = '*'
  20     version = '1'
  21     schema = ''
  22     namespace = ''
  23     extensions = {}
  24     namespaces = None
  25     template = '<RSpec type="%s"></RSpec>' % type
  26
  27     # Network
  28     def get_networks(self):
  29         network_elems = self.xml.xpath('//network')
  30         networks = [network_elem.get_instance(fields=['name', 'slice']) for
  31                     network_elem in network_elems]
  32         return networks
  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     # Nodes
  43
  44     def get_nodes(self, filter=None):
  45         return SFAv1Node.get_nodes(self.xml, filter)
  46
  47     def get_nodes_with_slivers(self):
  48         return SFAv1Node.get_nodes_with_slivers(self.xml)
  49
  50     def add_nodes(self, nodes, network=None, no_dupes=False, rspec_content_type=None):
  51         SFAv1Node.add_nodes(self.xml, nodes, rspec_content_type)
  52
  53     def merge_node(self, source_node_tag, network, no_dupes=False):
  54         if no_dupes and self.get_node_element(node['hostname']):
  55             # node already exists
  56             return
  57
  58         network_tag = self.add_network(network)
  59         network_tag.append(deepcopy(source_node_tag))
  60
  61     # Slivers
  62
  63     def add_slivers(self, hostnames, attributes=None, sliver_urn=None, append=False):
  64         if attributes is None:
  65             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     def remove_slivers(self, slivers, network=None, no_dupes=False):
  92         SFAv1Node.remove_slivers(self.xml, slivers)
  93
  94     def get_slice_attributes(self, network=None):
  95         attributes = []
  96         nodes_with_slivers = self.get_nodes_with_slivers()
  97         for default_attribute in self.get_default_sliver_attributes(network):
  98             attribute = default_attribute.copy()
  99             attribute['node_id'] = None
 100             attributes.append(attribute)
 101         for node in nodes_with_slivers:
 102             nodename = node['component_name']
 103             sliver_attributes = self.get_sliver_attributes(nodename, network)
 104             for sliver_attribute in sliver_attributes:
 105                 sliver_attribute['node_id'] = nodename
 106                 attributes.append(sliver_attribute)
 107         return attributes
 108
 109     def add_sliver_attribute(self, component_id, name, value, network=None):
 110         nodes = self.get_nodes({'component_id': '*%s*' % component_id})
 111         if nodes is not None and isinstance(nodes, list) and len(nodes) > 0:
 112             node = nodes[0]
 113             slivers = SFAv1Sliver.get_slivers(node)
 114             if slivers:
 115                 sliver = slivers[0]
 116                 SFAv1Sliver.add_sliver_attribute(sliver, name, value)
 117         else:
 118             # should this be an assert / raise an exception?
 119             logger.error("WARNING: failed to find component_id %s" %
 120                          component_id)
 121
 122     def get_sliver_attributes(self, component_id, network=None):
 123         nodes = self.get_nodes({'component_id': '*%s*' % component_id})
 124         attribs = []
 125         if nodes is not None and isinstance(nodes, list) and len(nodes) > 0:
 126             node = nodes[0]
 127             slivers = SFAv1Sliver.get_slivers(node.element)
 128             if slivers is not None and isinstance(slivers, list) and len(slivers) > 0:
 129                 sliver = slivers[0]
 130                 attribs = SFAv1Sliver.get_sliver_attributes(sliver.element)
 131         return attribs
 132
 133     def remove_sliver_attribute(self, component_id, name, value, network=None):
 134         attribs = self.get_sliver_attributes(component_id)
 135         for attrib in attribs:
 136             if attrib['name'] == name and attrib['value'] == value:
 137                 # attrib.element.delete()
 138                 parent = attrib.element.getparent()
 139                 parent.remove(attrib.element)
 140
 141     def add_default_sliver_attribute(self, name, value, network=None):
 142         if network:
 143             defaults = self.xml.xpath(
 144                 "//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(
 162                 "//network[@name='%s']/sliver_defaults" % network)
 163         else:
 164             defaults = self.xml.xpath("//sliver_defaults")
 165         if not defaults:
 166             return []
 167         return SFAv1Sliver.get_sliver_attributes(defaults[0])
 168
 169     def remove_default_sliver_attribute(self, name, value, network=None):
 170         attribs = self.get_default_sliver_attributes(network)
 171         for attrib in attribs:
 172             if attrib['name'] == name and attrib['value'] == value:
 173                 # attrib.element.delete()
 174                 parent = attrib.element.getparent()
 175                 parent.remove(attrib.element)
 176
 177     # Links
 178
 179     def get_links(self, network=None):
 180         return PGv2Link.get_links(self.xml)
 181
 182     def get_link_requests(self):
 183         return PGv2Link.get_link_requests(self.xml)
 184
 185     def add_links(self, links):
 186         networks = self.get_networks()
 187         if len(networks) > 0:
 188             xml = networks[0].element
 189         else:
 190             xml = self.xml
 191         PGv2Link.add_links(xml, links)
 192
 193     def add_link_requests(self, links):
 194         PGv2Link.add_link_requests(self.xml, links)
 195
 196     # utility
 197
 198     def merge(self, in_rspec):
 199         """
 200         Merge contents for specified rspec with current rspec
 201         """
 202
 203         if not in_rspec:
 204             return
 205
 206         from sfa.rspecs.rspec import RSpec
 207         if isinstance(in_rspec, RSpec):
 208             rspec = in_rspec
 209         else:
 210             rspec = RSpec(in_rspec)
 211         if rspec.version.type.lower() == 'protogeni':
 212             from sfa.rspecs.rspec_converter import RSpecConverter
 213             in_rspec = RSpecConverter.to_sfa_rspec(rspec.toxml())
 214             rspec = RSpec(in_rspec)
 215
 216         # just copy over all networks
 217         current_networks = self.get_networks()
 218         networks = rspec.version.get_networks()
 219         for network in networks:
 220             current_network = network.get('name')
 221             if current_network and current_network not in current_networks:
 222                 self.xml.append(network.element)
 223                 current_networks.append(current_network)
 224
 225     # Leases
 226
 227     def get_leases(self, filter=None):
 228         return SFAv1Lease.get_leases(self.xml, filter)
 229
 230     def add_leases(self, leases, network=None, no_dupes=False):
 231         SFAv1Lease.add_leases(self.xml, leases)
 232
 233     # Spectrum
 234
 235     def get_channels(self, filter=None):
 236         return []
 237
 238     def add_channels(self, channels, network=None, no_dupes=False):
 239         pass
 240
 241 if __name__ == '__main__':
 242     from sfa.rspecs.rspec import RSpec
 243     from sfa.rspecs.rspec_elements import *
 244     r = RSpec('/tmp/resources.rspec')
 245     r.load_rspec_elements(SFAv1.elements)
 246     print(r.get(RSpecElements.NODE))