sfa/rspecs/elements/SFAv1Network.py

   1 #!/usr/bin/python
   2 from sfa.rspecs.elements.network import Network
   3
   4 class SFAv1Network(Network):
   5
   6     def get_network_elements(self):
   7         return self.root_node.xpath('//network')
   8
   9     def get_networks(self):
  10         network_elems = self.get_network_elements()
  11         networks = [self.get_attributes(network_elem) \
  12                     for network_elem in network_elems]
  13         return networks
  14
  15     def add_networks(self, networks):
  16         if not isinstance(networks, list):
  17             networks = [networks]
  18         return self.add_element('network', {'id': network}, self.root_node)
  19
  20
  21
  22 if __name__ == '__main__':
  23     import sys
  24     from lxml import etree
  25     args = sys.argv[1:]
  26     filename = args[0]
  27
  28     root_node = etree.parse(filename)
  29     network = SFAv1Network(root_node)
  30     print network.get_networks()
  31
  32