source: CMIP6dreqbuild/trunk/src/framework/dreq.py @ 361

import xml, string, collections
import xml.dom
import xml.dom.minidom
class dreqItemBase(object):
       def __init__(self,dict=None,xmlMiniDom=None,id='defaultId'):
         dictMode = dict != None
         mdMode = xmlMiniDom != None
         assert not( dictMode and mdMode), 'Mode must be either dictionary of minidom: both assigned'
         assert dictMode or mdMode, 'Mode must be either dictionary of minidom: neither assigned'
         self.defaults = {}
         self.globalDefault = '__unset__'
         if dictMode:
           self.dictInit( dict )
         elif mdMode:
           self.mdInit( xmlMiniDom )

       def dictInit( self, dict ):
         for a in self.attributes:
           if dict.has_key(a):
             self.__dict__[a] = dict[a]
           else:
             self.__dict__[a] = self.defaults.get( a, self.globalDefault )

       def mdInit( self, el ):
         __doc__ = """Initialisation from a mindom XML element. The list of attributes must be set by the class factory before the class is initialised"""
         for a in self.attributes:
           if el.hasAttribute( a ):
             v = el.getAttribute( a )
             self.__dict__[a] = v
           else:
             self.__dict__[a] = self.defaults.get( a, self.globalDefault )

class config(object):

  def __init__(self):
    self.vdef = 'out/dreqDefn.xml' 
    self.vsamp = 'out/vocabSample.xml'
    self.vsamp = '../workbook/trial.xml'
    self.vsamp = '../workbook/trial_20150724.xml'
    self.nts = collections.namedtuple( 'sectdef', ['tag','label','title','id','itemLabelMode'] )
    self.nti = collections.namedtuple( 'itemdef', ['tag','label','title','type','rClass','techNote'] )
    self.ntt = collections.namedtuple( 'sect', ['header','attributes'] )

    doc = xml.dom.minidom.parse( self.vdef  )
    self.contentDoc = xml.dom.minidom.parse( self.vsamp )
##<vocab label="institute" title="Institute" id="cmip.drv.001" itemLabelMode="def">
##  <itemAttribute label="label"/>
    vl = doc.getElementsByTagName( 'table' )
    print len(vl)
    self.slist = []
    self.tables = {}
    tables = {}
    self.tableClasses = {}
    self.tableItems = collections.defaultdict( list )
    for v in vl:
      t = self.parsevcfg(v)
      tables[t[0].label] = t
      self.tableClasses[t[0].label] = self.itemClassFact( t.header.itemLabelMode, t.attributes.keys() )
      self.slist.append( t )

    for k in tables.keys():
      vl = self.contentDoc.getElementsByTagName( k )
      if len(vl) == 1:
        v = vl[0]
        t = v.getAttribute( 'title' )
        i = v.getAttribute( 'id' )
        il = v.getElementsByTagName( 'item' )
        print k, t, i, len(il)
        self.tables[k] = (i,t,len(il))
        
        for i in il:
          ii = self.tableClasses[k](xmlMiniDom=i)
          self.tableItems[k].append( ii )
      elif len(vl) > 1:
        l1 = []
        l2 = []
        print '#### %s: %s' % (k,len(vl) )
        for v in vl:
          t = v.getAttribute( 'title' )
          i = v.getAttribute( 'id' )
          il = v.getElementsByTagName( 'item' )
          print k, t, i, len(il)
          l1.append( (i,t,len(il)) )
          
          l2i = []
          for i in il:
            ii = self.tableClasses[k](xmlMiniDom=i)
            l2i.append( ii )
          l2.append( l2i )
        self.tables[k] = l1
        self.tableItems[k] = l2

  def itemClassFact(self,itemLabelMode,attributes):
     class dreqItem(dreqItemBase):
       """Inherits all methods from dreqItemBase"""
       
     dreqItem.itemLabelMode = itemLabelMode
     dreqItem.attributes = attributes
     return dreqItem
         
  def parsevcfg(self,v):
      l = v.getAttribute( 'label' )
      t = v.getAttribute( 'title' )
      i = v.getAttribute( 'id' )
      ilm = v.getAttribute( 'itemLabelMode' )
      il = v.getElementsByTagName( 'rowAttribute' )
      vtt = self.nts( v.nodeName, l,t,i,ilm )
      ll = []
      idict = {}
      for i in il:
        tt = self.parseicfg(i)
        idict[tt.label] = tt
        ll.append(tt)
      return self.ntt( vtt, idict )

  def parseicfg(self,i):
      l = i.getAttribute( 'label' )
      if i.hasAttribute( 'title' ):
        t = i.getAttribute( 'title' )
        self.lastTitle = t
      else:
        t = None
      if i.hasAttribute( 'type' ):
        ty = i.getAttribute( 'type' )
      else:
        ty = "xs:string"
      if i.hasAttribute( 'class' ):
        cls = i.getAttribute( 'class' )
      else:
        cls = None
      if i.hasAttribute( 'techNote' ):
        tn = i.getAttribute( 'techNote' )
      else:
        tn = None
      return self.nti( i.nodeName, l,t,ty,cls,tn )

c = config()