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
11 def add_links(xml, links):
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'])
35 type_elem = link_elem.add_element(
36 'link_type', name=link['type'])
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)
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']
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']
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]
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]
78 def add_link_requests(xml, link_tuples, append=False):
79 if not isinstance(link_tuples, set):
80 link_tuples = set(link_tuples)
82 available_links = PGv2Link.get_links(xml)
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()
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)
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'])
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)
112 def get_link_requests(xml):
114 for link in PGv2Link.get_links(xml):
115 if link['client_id'] != None:
116 link_requests.append(link)