Changeset 1066 in lmdz_wrf for trunk/tools/generic_tools.py

Timestamp:

Aug 30, 2016, 6:49:41 PM (8 years ago)

Author:

lfita

Message:

Fixing diagnostics variable retrievals for multiple possibilities (different models)

File:

• trunk/tools/generic_tools.py (modified) (2 diffs)

trunk/tools/generic_tools.py

@@ -666 +666 @@
     >>> CFvar_DIAGvar('hurs')
     {'TSrhs': ['psfc', 't', 'q'], 'LMDZrhs': ['psol', 't2m', 'q2m'], 'WRFrhs': ['PSFC', 'T2', 'Q2']}
+    >>> CFvar_DIAGvar('wss')
+    {'wss': [['U10', 'V10'], ['u10m', 'v10m']]}
     """
     import subprocess as sub
@@ -687 +689 @@
 
     MODvars = {}
+    MODdiags = []
     for line in ncf:
         if line[0:1] != '#':
             values = line.replace('\n','').replace(' ','').split(',')
-            if values[0] == varn: MODvars[values[1]] = values[2].split('@')
+            if values[0] == varn: 
+                MODdiags.append(values[2].split('@'))
+
+    MODvars[varn] = MODdiags
 
     if len(MODvars) == 0: