source: CMIP6dreqbuild/trunk/src/framework/exports.py @ 744

from dreqPy import dreq
import collections, json, time

dq = dreq.loadDreq()
rvgUsage = collections.defaultdict( set )
rvUsage = {}
cmvUsage = collections.defaultdict( set )
vUsage = collections.defaultdict( set )
for i in dq.coll['requestVarGroup'].items:
  for u in dq.inx.iref_by_sect[i.uid].a['requestLink']:
    rl = dq.inx.uid[u]
    rvgUsage[i.uid].add( rl.mip )

for i in dq.coll['requestVar'].items:
  rvUsage[i.uid] = rvgUsage[i.vgid]

for i in dq.coll['CMORvar'].items:
  for u in dq.inx.iref_by_sect[i.uid].a['requestVar']:
    if u in rvUsage:
      for x in rvUsage[u]:
        cmvUsage[i.uid].add(x)

for i in dq.coll['var'].items:
  for u in dq.inx.iref_by_sect[i.uid].a['CMORvar']:
    if u in cmvUsage:
      for x in cmvUsage[u]:
        vUsage[i.uid].add(x)

vars = {i.label: {'long_name':i.title, 'standard_name':i.sn, 'units':i.units, 'usage':sorted(list(vUsage[i.uid])), 'prov':'%s {%s}' % (i.prov,i.provmip)} for i in dq.coll['var'].items if i.uid in vUsage }
info = {'version':dq.version, 'source':'Data Request', 'author':'Martin Juckes (martin.juckes@stfc.ac.uk)', 'creation_date':time.ctime()}
oo = open( 'CV_var.json', 'w' )
json.dump( {'info':info, 'variable':vars}, oo, indent=4, sort_keys=True )
oo.close()