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

''' Generate vocabulary definition document and sample vocabulary document.
USAGE
.....
ptxt.py [-f srcFile] [samp|defn]
'''
import string, re, collections, sys, uuid

if len(sys.argv) == 1:
  print __doc__
  exit()

args = sys.argv[1:]
if args[0] == '-f':
  srcFn = args[1]
  args = args[2:]
else:
  srcFn = 'parVocabDefn.txt'
  srcFn = 'dreqDefn.txt'
  
mode = args[0]

assert mode in ["defn", "samp"]

re_it = re.compile( '{(.*)}' )
re_is = re.compile( '\[(.*)\]' )
re_atdef = re.compile( '^(?P<n>\S+)\s*(((?P<b>\[.*?])|(?P<a>\{.*?\})|(?P<c>\<.*?\>))\s*){0,3}$' )
re_atdef = re.compile( '^(?P<n>\S+)\s*((\[(?P<b>.*?)\]|\{(?P<a>.*?)\}|\<(?P<c>.*?)\>)\s*){0,3}$' )

### tuble to define attributes of an item
nt__itematt = collections.namedtuple( 'itematt', ['name','type','title','clss','techn'] )

vocab_elTmpl = '''<table label="%(label)s" title="%(title)s" id="%(id)s" itemLabelMode="%(ilm)s">
%(itemAttrList)s
</table>
'''
ial_elTmpl = '  <rowAttribute label="%(label)s"%(wrappedType)s%(wrappedTitle)s%(wrappedClass)s%(wrappedTechn)s/>'

expl_Tmpl = '''<%(label)s class="vocab" title="%(title)s" id="%(id)s">
<!-- <info srcType="dummy" srcRef="ptxt.py">Dummy entries</info> -->
%(exampleItem)s
</%(label)s>
'''
item_Tmpl = '<item id="%(id)s" label="%(label)s" title="%(title)s"%(extras)s/>'

class vocab(object):

  def __init__( self, line, kk=0 ):
     bits = map( string.strip, string.split(line, ';' ) )
     assert len (bits) == 4, '[1] Cannot parse %s' % line
     assert bits[0][:5] == 'vocab', '[2] Cannot parse %s' % line
     self.label = string.split(bits[0], ' ')[1]
     self.title = bits[1]
     self.id = 'cmip.drv.%3.3i' % kk
     self.ilt = bits[3]
     self.kk = kk
     self.msg( '[%s] %s {%s:%s}' % (self.label, self.title, self.id, self.ilt) )
     self.itematts = [  nt__itematt( 'label','xs:string',None,None,None ),
                        nt__itematt( 'title','xs:string',None,None,None ) ]

  def tmpl(self,oo=None,mode="defn"):
     if mode == "defn":
       ss = []
       for i in self.itematts:
         if i.clss != None:
           wrappedClass = ' class="%s"' % i.clss
         else:
           wrappedClass = ''

         if i.techn != None:
           wrappedTechn = ' techNote="%s"' % i.techn
         else:
           wrappedTechn = ''

         if i.title != None:
           wrappedTitle = ' title="%s"' % i.title
         else:
           wrappedTitle = ''

         if i.type != "xs:string":
           wrappedType = ' type="%s"' % i.type
         else:
           wrappedType = ''

         label = i.name
         ss.append(ial_elTmpl % locals())
       sss = string.join( ss, '\n' )

       label = self.label
       title = self.title
       id = self.id
       ilm =  self.ilt
       itemAttrList = sss
       self.vocab = vocab_elTmpl % locals()
     else:
## construct sample item ##
       if self.ilt == 'an':
         label = 'example01'
       elif self.ilt == 'def':
         label = 'example-01'
       elif self.ilt == 'int':
         label = '1'
       title = 'dummy title string'

       extras = ''
       for i in self.itematts:
         if i.name not in ['label','title']:
           value = "noType"
           if i.type == "xs:string":
             if i.name == 'uuid':
               value = str( uuid.uuid1() )
             else:
               value = 'dummyAt'
           elif i.type == "xs:integer":
             value = '25'
           elif i.type == "xs:duration":
             value = 'P1Y'

           extras += ' %s="%s"' % (i.name,value)
       id = '001.%3.3i.%3.3i' % (self.kk,1)
       exampleItem = item_Tmpl % locals()
       label = self.label
       title = self.title
       id = self.id
       ilm =  self.ilt
       self.vocab = expl_Tmpl % locals()
##
## print and write if needed
##
     if oo != None:
         oo.write( self.vocab )
     self.msg( vocab )

  def msg(self,txt):
     swallow=True
     if swallow:
       return
     print text

  def attr(self,line):
     bits = map( string.strip, string.split(line[3:], ';' ) )
     for b in bits:
       if string.strip(b) != '':
         bn = string.split(b, ' ')[0]
         x = self.pb(b)
         self.itematts.append( nt__itematt( bn, x[0], x[1], x[2], x[3] ) )
         self.msg( '%s, %s, %s, %s, %s' % (bn, x[0], x[1], x[2], x[3]) )

  def pb(self, b):
    x = re_it.findall(b)
    m = re_atdef.match( b )
    if m == None:
         print 'COULD NOT SCAN: ',b
         assert False
    else:
      ee = m.groupdict()
      for k in ['a','b','c']:
        if type( ee[k] ) == type( 'x' ):
          assert string.find( ee[k], '"' ) == -1, 'unable to deal with quote in string: %s' % ee[k]
      itemTitle = ee['a']
      itemType = ee['b']
      itemClass = ee['c']
      itemTechNote = None
      if itemType == None:
         itemType = 'xs:string'
      if type(itemClass) == type('x'):
        if string.find(itemClass, '|' ) != -1:
          bits = string.split( itemClass, '|' )
          assert len(bits) == 2, 'ERROR.001.0001: failed to parse attribute definition: %s' % b
          itemClass = bits[0]
          itemTechNote = bits[1]
        
    #if len(x) > 0:
      #itemTitle = x[0]
    #else:
      #itemTitle = None
    #x = re_is.findall(b)
    #if len(x) > 0:
      #itemType = x[0]
    #else:
      #itemType = 'xs:string'
    return [itemType, itemTitle,itemClass,itemTechNote] 

class main(object):

  def __init__(self,fn):
    ii = []
    for l in open('../../docs/%s' % fn).readlines():
      if l[0] != '#':
        ii.append(l)
    kk= 0
    this = None
    fstem = string.split( fn, '.' )[0]
    fns = {'defn':'%sDefn' % fstem, 'samp':'%sSample' % fstem }
    fn = fns[mode]
    oo = open( 'out/%s.xml' % fn, 'w' )
    if mode == 'samp':
      mainEl = "main"
    else:
      mainEl = "defDoc"

    oo.write( '''<?xml version="1.0" ?>
<%s>\n''' % mainEl )

    for l in ii:
      if l[:5] == 'vocab':
        if this != None:
          this.tmpl(oo=oo,mode=mode)
        kk+=1
        this = vocab(l,kk=kk)
      else:
        this.attr( l )

    this.tmpl(oo=oo,mode=mode)
    oo.write( '</%s>\n' % mainEl )
    oo.close()
    

m = main(srcFn)