sfa/rspecs/pg_rspec.py

   1 #!/usr/bin/python
   2 from lxml import etree
   3 from StringIO import StringIO
   4 from sfa.rspecs.rspec import RSpec
   5 from sfa.util.xrn import *
   6 from sfa.util.plxrn import hostname_to_urn
   7 from sfa.util.config import Config
   8
   9
  10 class PGRSpec(RSpec):
  11     header = '<?xml version="1.0"?>\n'
  12     template = """
  13 <rspec xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://www.protogeni.net/resources/rspec/0.2" xsi:schemaLocation="http://www.protogeni.net/resources/rspec/0.2 http://www.protogeni.net/resources/rspec/0.2/ad.xsd">\n
  14 </rspec>
  15 """
  16     namespaces = {'rspecv2':'http://www.protogeni.net/resources/rspec/0.2',
  17                   'xsi': 'http://www.w3.org/2001/XMLSchema-instance'
  18                  }
  19     schemas =  {'xsi': 'http://www.protogeni.net/resources/rspec/0.2 http://www.protogeni.net/resources/rspec/0.2/ad.xsd'
  20             }
  21     format = 'pg'
  22     xml = None
  23
  24     def get_network(self):
  25         network = None
  26         nodes = self.xml.xpath('//rspecv2:node[@component_manager_uuid][1]', self.namespaces)
  27         if nodes:
  28             network  = nodes[0].get('component_manager_uuid')
  29         return network
  30
  31     def get_networks(self):
  32         networks = self.xml.xpath('//rspecv2:node[@component_manager_uuid]/@component_manager_uuid')
  33         return set(networks)
  34
  35     def get_node_elements(self):
  36         nodes = self.xml.xpath('//rspecv2:node', self.namespaces)
  37         return nodes
  38
  39     def get_nodes(self, network=None):
  40         return self.xml.xpath('//rspecv2:node[@component_uuid]/@component_uuid', self.namespaces)
  41
  42     def get_nodes_with_slivers(self, network=None):
  43         if network:
  44             return self.xml.xpath('//node[@component_manager_uuid="%s"][sliver_type]/@component_uuid' % network, self.namespaces)
  45         else:
  46             return self.xml.xpath('//node[sliver_type]/@component_uuid' % network, self.namespaces)
  47
  48     def get_nodes_without_slivers(self, network=None):
  49         pass
  50
  51     def add_nodes(self, nodes, check_for_dupes=False):
  52         if not isinstance(nodes, list):
  53             nodes = [nodes]
  54         for node in nodes:
  55             urn = ""
  56             if check_for_dupes and \
  57               self.xml.xpath('//rspecv2:node[@component_uuid="%s"]' % urn, self.namespaces):
  58                 # node already exists
  59                 continue
  60
  61             node_tag = etree.SubElement(self.xml, 'node')
  62             node_type_tag = etree.SubElement(node_tag, 'node_type', type_name='pcvm', type_slots='100')
  63             available_tag = etree.SubElement(node_tag, 'available').text = 'true'
  64             exclusive_tag = etree.SubElement(node_tag, 'exclusive').text = 'false'
  65             location_tag = etree.SubElement(node_tag, 'location')
  66             interface_tag = etree.SubElement(node_tag, 'interface')
  67
  68
  69     def add_slivers(self, slivers, check_for_dupes=False):
  70         pass
  71
  72     def add_interfaces(self, interfaces, check_for_dupes=False):
  73         pass
  74
  75     def add_links(self, links, check_for_dupes=False):
  76         pass
  77
  78
  79 if __name__ == '__main__':
  80     rspec = PGRSpec()
  81     rspec.add_nodes([1])
  82     print rspec