slice-attributes/SlicesParse.py

   1 #!/usr/bin/env python
   2
   3 import xml.dom.minidom
   4 import time
   5 import sys
   6
   7 def dom_debug (node,level):
   8     print level,'<<< dom_debug 1>>>',node.nodeType
   9 #    print level,'<<< dom_debug 2>>>',dir(node)
  10 #    print level,'<<< dom_debug 4>>>',node.nodeValue
  11     for child in node.childNodes:
  12         dom_debug (child,level+1)
  13     print level,'<<end>>',len(node.childNodes)
  14
  15 # returns node's value - tries to be flexible
  16 def dom_get_text (node):
  17 #    dom_debug(node,0)
  18     if node.nodeType == node.TEXT_NODE:
  19         return node.data
  20     elif node.nodeType == node.ELEMENT_NODE and len(node.childNodes) == 1:
  21         return dom_get_text(node.childNodes[0])
  22     elif node.nodeType == node.ELEMENT_NODE:
  23         return 'dom_get_text: element does not have 1 child'
  24     return ""
  25
  26 class SlicesParse:
  27
  28     def __init__ (self, filename):
  29         self.filename=filename
  30
  31         impl = xml.dom.minidom.getDOMImplementation()
  32         self.dom = impl.createDocument (None, 'slices', None)
  33
  34
  35     def load (self):
  36
  37         self.doc = xml.dom.minidom.parse(self.filename)
  38
  39     def scan_rspecs (self):
  40
  41         slices = {}
  42         for slice in self.doc.getElementsByTagName('slice'):
  43             rspecs = []
  44             resources = slice.getElementsByTagName('rspec')[0]
  45             for resource in resources.getElementsByTagName('resource'):
  46                 value=resource.getElementsByTagName('value')[0]
  47                 rspecs.append({ 'name': value.getAttribute('name'),
  48                                 'type':value.getAttribute('type'),
  49                                 'value':dom_get_text(value) } )
  50             if rspecs:
  51                 slices [slice.getAttribute('name')] = rspecs
  52
  53         return slices
  54
  55 if __name__ == '__main__':
  56
  57     for filename in sys.argv[1:]:
  58         export = SlicesParse(filename)
  59
  60         start = time.time()
  61         export.load()
  62         now = time.time()
  63         print 'parsing : %d seconds'%(now-start)
  64         start=now
  65
  66         slices=export.scan_rspecs()
  67         now = time.time()
  68         print 'scanning : %d seconds'%(now-start)
  69         start=now
  70
  71         storage=open(filename+".bin","wb")
  72         import pickle
  73         pickle.dump(slices,storage)
  74         storage.close()
  75         print 'storage : %d seconds'%(now-start)
  76         start=now