source: CMIP6dreqbuild/trunk/src/framework/appendViacs.py @ 903

import xml, string
import xml.dom, xml.dom.minidom
import utils_wb
import re
import appendEsmVal

empty=re.compile('^$')

import appendViacs_p1 as p1

print p1.res.keys()

class addViacs(object):

  def __init__(self):

    src = 'out/annotated_20150731_i1.xml'
    dest = 'out/annotated_20150731.xml'
    self.doc = xml.dom.minidom.parse( src  )
    section = 'requestVar'
    this = self.doc.getElementsByTagName(section)[0]
    for tt in p1.res[section]:
      item = self.doc.createElement( 'item'  )
      item.setAttribute( 'uid', tt[0] )
      item.setAttribute( 'label', tt[1] )
      item.setAttribute( 'priority', '1' )
      item.setAttribute( 'title', '%s: %s' % ( tt[2], tt[1]) )
      item.setAttribute( 'vid', tt[3] )
      item.setAttribute( 'vgid', tt[4] )
      this.appendChild( item )

    section = 'requestLink'
    this = self.doc.getElementsByTagName(section)[0]
    for tt in p1.res[section]:
      item = self.doc.createElement( 'item'  )
      item.setAttribute( 'uid', tt[0] )
      item.setAttribute( 'objective', tt[1] )
      item.setAttribute( 'mip', 'VIACSAB' )
      item.setAttribute( 'label', tt[3] )
      item.setAttribute( 'title', tt[4] )
      item.setAttribute( 'refid', tt[5] )
      this.appendChild( item )

    section = 'requestItem'
    this = self.doc.getElementsByTagName(section)[0]
    for tt in p1.res[section]:
      item = self.doc.createElement( 'item'  )
      item.setAttribute( 'uid', tt[0] )
      item.setAttribute( 'rlid', tt[1] )
      item.setAttribute( 'esid', tt[2] )
      item.setAttribute( 'mip', 'VIACSAB' )
      item.setAttribute( 'label', tt[3] )
      item.setAttribute( 'title', tt[4] )
      item.setAttribute( 'ny', '160' )
      item.setAttribute( 'preset', '-1' )
      t0 = string.split( tt[4], ':' )[0]
      if t0[-2:] == '3b':
        item.setAttribute( 'tslice', '_slice_cfOff' )
        item.setAttribute( 'nenmax', '-1' )
        item.setAttribute( 'nymax', '1' )
      elif t0[-1] == 'c':
        item.setAttribute( 'nenmax', '1' )
        item.setAttribute( 'nymax', '160' )
      elif t0[-1] == 'd':
        item.setAttribute( 'nenmax', '1' )
        item.setAttribute( 'nymax', '40' )
      else:
        item.setAttribute( 'nenmax', '-1' )
        item.setAttribute( 'nymax', '-1' )
      this.appendChild( item )

    section = 'objective'
    this = self.doc.getElementsByTagName(section)[0]
    for tt in p1.res[section]:
      item = self.doc.createElement( 'item'  )
      item.setAttribute( 'uid', tt[0] )
      item.setAttribute( 'mip', tt[1] )
      item.setAttribute( 'label', tt[2] )
      item.setAttribute( 'title', tt[3] )
      item.setAttribute( 'description', tt[4] )
      this.appendChild( item )

    section = 'objectiveLink'
    this = self.doc.getElementsByTagName(section)[0]
    for tt in p1.res[section]:
      item = self.doc.createElement( 'item'  )
      item.setAttribute( 'uid', tt[0] )
      item.setAttribute( 'label', tt[1] )
      item.setAttribute( 'title', tt[2] )
      item.setAttribute( 'oid', tt[3] )
      item.setAttribute( 'rid', tt[4] )
      this.appendChild( item )


  def write(self):
    txt = self.doc.toprettyxml(indent='\t', newl='\n', encoding=None)
    oo = open( 'out/annotated_20150731.xml', 'w' )
    lines = string.split( txt, '\n' )
    for line in lines:
      l = utils_wb.uniCleanFunc( string.strip(line) )
      if empty.match(l):
        continue
      else:
        oo.write(l + '\n')
    oo.close()

adv = addViacs()
appendEsmVal.addEsmVal( adv.doc )
adv.write()