sfa/rspecs/elements/versions/pgv2Link.py

   1 from sfa.util.plxrn import PlXrn
   2 from sfa.util.xrn import Xrn
   3 from sfa.rspecs.elements.element import Element
   4 from sfa.rspecs.elements.link import Link
   5 from sfa.rspecs.elements.interface import Interface
   6 from sfa.rspecs.elements.property import Property
   7
   8 class PGv2Link:
   9     @staticmethod
  10     def add_links(xml, links):
  11         for link in links:
  12             link_elems = Element.add(xml, 'link', link, ['component_name', 'component_id', 'client_id'])
  13             link_elem = link_elems[0]
  14             # set component manager element
  15             if 'component_manager' in link and link['component_manager']:
  16                 cm_element = link_elem.add_element('component_manager', name=link['component_manager'])
  17             # set interface_ref elements
  18             for if_ref in [link['interface1'], link['interface2']]:
  19                 Element.add(link_elem, 'interface_ref', if_ref, Interface.fields)
  20             # set property elements
  21             prop1 = link_elem.add_element('property', source_id = link['interface1']['component_id'],
  22                 dest_id = link['interface2']['component_id'], capacity=link['capacity'],
  23                 latency=link['latency'], packet_loss=link['packet_loss'])
  24             prop2 = link_elem.add_element('property', source_id = link['interface2']['component_id'],
  25                 dest_id = link['interface1']['component_id'], capacity=link['capacity'],
  26                 latency=link['latency'], packet_loss=link['packet_loss'])
  27             if link.get('type'):
  28                 type_elem = link_elem.add_element('link_type', name=link['type'])
  29
  30     @staticmethod
  31     def get_links(xml):
  32         links = []
  33         link_elems = xml.xpath('//default:link | //link')
  34         for link_elem in link_elems:
  35             # set client_id, component_id, component_name
  36             link = Link(link_elem.attrib, link_elem)
  37
  38             # set component manager
  39             component_managers = link_elem.xpath('./default:component_manager | ./component_manager')
  40             if len(component_managers) > 0 and 'name' in component_managers[0].attrib:
  41                 link['component_manager'] = component_managers[0].attrib['name']
  42
  43             # set link type
  44             link_types = link_elem.xpath('./default:link_type | ./link_type')
  45             if len(link_types) > 0 and 'name' in link_types[0].attrib:
  46                 link['type'] = link_types[0].attrib['name']
  47
  48             # get capacity, latency and packet_loss from first property
  49             property_fields = ['capacity', 'latency', 'packet_loss']
  50             property_elems = link_elem.xpath('./default:property | ./property')
  51             if len(propery_elems) > 0:
  52                 prop = property_elems[0]
  53                 for attrib in ['capacity', 'latency', 'packet_loss']:
  54                     if attrib in prop:
  55                         link[attrib] = prop[attrib]
  56
  57             # get interfaces
  58             interfaces = Element.get(Interface, link_elem, './default:interface_ref | ./interface_ref')
  59             if len(interfaces) > 1:
  60                 link['interface1'] = interfaces[0]
  61                 link['interface2'] = interfaces[1]
  62             links.append(link)
  63         return links
  64
  65     @staticmethod
  66     def add_link_requests(xml, link_tuples, append=False):
  67         if not isinstance(link_tuples, set):
  68             link_tuples = set(link_tuples)
  69
  70         available_links = PGv2Link.get_links(xml)
  71         recently_added = []
  72         for link in available_links:
  73             if_name1 =  Xrn(link['interface1']['component_id']).get_leaf()
  74             if_name2 =  Xrn(link['interface2']['component_id']).get_leaf()
  75
  76             requested_link = None
  77             l_tup_1 = (if_name1, if_name2)
  78             l_tup_2 = (if_name2, if_name1)
  79             if link_tuples.issuperset([(if_name1, if_name2)]):
  80                 requested_link = (if_name1, if_name2)
  81             elif link_tuples.issuperset([(if_name2, if_name2)]):
  82                 requested_link = (if_name2, if_name1)
  83             if requested_link:
  84                 # add client id to link ane interface elements
  85                 link.element.set('client_id', link['component_name'])
  86                 link['interface1'].element.set('client_id', Xrn(link['interface1']['component_id']).get_leaf())
  87                 link['interface2'].element.set('client_id', Xrn(link['interface2']['component_id']).get_leaf())
  88                 recently_added.append(link['component_name'])
  89
  90         if not append:
  91             # remove all links that don't have a client id
  92             for link in PGv2Link.get_links(xml):
  93                 if not link['client_id'] or link['component_name'] not in recently_added:
  94                     parent = link.element.getparent()
  95                     parent.remove(link.element)
  96
  97     @staticmethod
  98     def get_link_requests(xml):
  99         link_requests = []
 100         for link in PGv2Link.get_links(xml):
 101             if link['client_id'] != None:
 102                 link_requests.append(link)
 103         return link_requests