sfa/rspecs/elements/versions/sfav1Node.py

   1 from sfa.util.sfalogging import logger
   2 from sfa.util.xml import XpathFilter
   3 from sfa.util.plxrn import PlXrn, xrn_to_hostname
   4 from sfa.util.xrn import Xrn
   5 from sfa.rspecs.elements.element import Element
   6 from sfa.rspecs.elements.node import Node
   7 from sfa.rspecs.elements.sliver import Sliver
   8 from sfa.rspecs.elements.location import Location
   9 from sfa.rspecs.elements.hardware_type import HardwareType
  10 from sfa.rspecs.elements.disk_image import DiskImage
  11 from sfa.rspecs.elements.interface import Interface
  12 from sfa.rspecs.elements.bwlimit import BWlimit
  13 from sfa.rspecs.elements.pltag import PLTag
  14 from sfa.rspecs.elements.versions.sfav1Sliver import SFAv1Sliver
  15 from sfa.rspecs.elements.versions.sfav1PLTag import SFAv1PLTag
  16 from sfa.rspecs.elements.versions.pgv2Services import PGv2Services
  17
  18 class SFAv1Node:
  19
  20     @staticmethod
  21     def add_nodes(xml, nodes):
  22         network_elems = xml.xpath('//network')
  23         if len(network_elems) > 0:
  24             network_elem = network_elems[0]
  25         elif len(nodes) > 0 and nodes[0].get('component_manager_id'):
  26             network_urn = nodes[0]['component_manager_id']
  27             network_elem = xml.add_element('network', name = Xrn(network_urn).get_hrn()[0])
  28         else:
  29             network_elem = xml
  30
  31         node_elems = []
  32         for node in nodes:
  33             node_fields = ['component_manager_id', 'component_id', 'boot_state']
  34             node_elem = network_elem.add_instance('node', node, node_fields)
  35             node_elems.append(node_elem)
  36
  37             # determine network hrn
  38             network_hrn = None
  39             if 'component_manager_id' in node and node['component_manager_id']:
  40                 network_hrn = Xrn(node['component_manager_id']).get_hrn()
  41
  42             # set component_name attribute and  hostname element
  43             if 'component_id' in node and node['component_id']:
  44                 component_name = xrn_to_hostname(node['component_id'])
  45                 node_elem.set('component_name', component_name)
  46                 hostname_elem = node_elem.add_element('hostname')
  47                 hostname_elem.set_text(component_name)
  48
  49             # set site id
  50             if 'authority_id' in node and node['authority_id']:
  51                 node_elem.set('site_id', node['authority_id'])
  52
  53             # add locaiton
  54             location = node.get('location')
  55             if location:
  56                 node_elem.add_instance('location', location, Location.fields)
  57
  58             for interface in node.get('interfaces', []):
  59                 node_elem.add_instance('interface', interface, ['component_id', 'client_id', 'ipv4'])
  60
  61             #if 'bw_unallocated' in node and node['bw_unallocated']:
  62             #    bw_unallocated = etree.SubElement(node_elem, 'bw_unallocated', units='kbps').text = str(int(node['bw_unallocated'])/1000)
  63
  64             PGv2Services.add_services(node_elem, node.get('services', []))
  65             SFAv1PLTag.add_pl_tags(node_elem, node.get('tags', []))
  66             SFAv1Sliver.add_slivers(node_elem, node.get('slivers', []))
  67
  68     @staticmethod
  69     def add_slivers(xml, slivers):
  70         component_ids = []
  71         for sliver in slivers:
  72             filter = {}
  73             if isinstance(sliver, str):
  74                 filter['component_id'] = '*%s*' % sliver
  75                 sliver = {}
  76             elif 'component_id' in sliver and sliver['component_id']:
  77                 filter['component_id'] = '*%s*' % sliver['component_id']
  78             if not filter:
  79                 continue
  80             nodes = SFAv1Node.get_nodes(xml, filter)
  81             if not nodes:
  82                 continue
  83             node = nodes[0]
  84             SFAv1Sliver.add_slivers(node, sliver)
  85
  86     @staticmethod
  87     def remove_slivers(xml, hostnames):
  88         for hostname in hostnames:
  89             nodes = SFAv1Node.get_nodes(xml, {'component_id': '*%s*' % hostname})
  90             for node in nodes:
  91                 slivers = SFAv1Slivers.get_slivers(node.element)
  92                 for sliver in slivers:
  93                     node.element.remove(sliver.element)
  94
  95     @staticmethod
  96     def get_nodes(xml, filter={}):
  97         xpath = '//node%s | //default:node%s' % (XpathFilter.xpath(filter), XpathFilter.xpath(filter))
  98         node_elems = xml.xpath(xpath)
  99         return SFAv1Node.get_node_objs(node_elems)
 100
 101     @staticmethod
 102     def get_nodes_with_slivers(xml):
 103         xpath = '//node[count(sliver)>0] | //default:node[count(default:sliver)>0]'
 104         node_elems = xml.xpath(xpath)
 105         return SFAv1Node.get_node_objs(node_elems)
 106
 107
 108     @staticmethod
 109     def get_node_objs(node_elems):
 110         nodes = []
 111         for node_elem in node_elems:
 112             node = Node(node_elem.attrib, node_elem)
 113             if 'site_id' in node_elem.attrib:
 114                 node['authority_id'] = node_elem.attrib['site_id']
 115             # get location
 116             location_elems = node_elem.xpath('./default:location | ./location')
 117             locations = [loc_elem.get_instance(Location) for loc_elem in location_elems]
 118             if len(locations) > 0:
 119                 node['location'] = locations[0]
 120             # get bwlimit
 121             bwlimit_elems = node_elem.xpath('./default:bw_limit | ./bw_limit')
 122             bwlimits = [bwlimit_elem.get_instance(BWlimit) for bwlimit_elem in bwlimit_elems]
 123             if len(bwlimits) > 0:
 124                 node['bwlimit'] = bwlimits[0]
 125             # get interfaces
 126             iface_elems = node_elem.xpath('./default:interface | ./interface')
 127             ifaces = [iface_elem.get_instance(Interface) for iface_elem in iface_elems]
 128             node['interfaces'] = ifaces
 129             # get services
 130             node['services'] = PGv2Services.get_services(node_elem)
 131             # get slivers
 132             node['slivers'] = SFAv1Sliver.get_slivers(node_elem)
 133             # get tags
 134             node['tags'] =  SFAv1PLTag.get_pl_tags(node_elem, ignore=Node.fields)
 135             nodes.append(node)
 136         return nodes
 137