sfa/rspecs/elements/versions/pgv2Link.py

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