Changeset 967 in lmdz_wrf for trunk/tools

Timestamp:

Jun 28, 2016, 7:47:51 PM (8 years ago)

Author:

lfita

Message:

Addding novar' as a value in function get_Variables'

Location:

trunk/tools

Files:

• nc_var.py (modified) (1 diff)
• nc_var_tools.py (modified) (4 diffs)

trunk/tools/nc_var.py

@@ -180 +180 @@
     ncvar.get_namelist_vars(opts.values, opts.ncfile)
 elif oper == 'get_Variables':
-    ncvar.get_Variables(opts.values, opts.ncfile)
+    ncvar.get_Variables(opts.values, opts.ncfile, opts.varname)
 elif oper == 'getvalues_lonlat':
     ncvar.getvalues_lonlat(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -16342 +16342 @@
     return
 
-def get_Variables(values, ncfile):
+def get_Variables(values, ncfile, variables):
     """ Function to get a list of variables from an existing file
-      values = ',' separated list of names of variables to get
+      novar= behaviour when no variable is found
+        'error': an error messages arise and program stops
+        'warn': a warning messages arise and program continues
       ncfile= WRF file to use
+      variables = ',' separated list of names of variables to get
     """
     fname = 'get_Variables'
@@ -16354 +16357 @@
         quit()
 
-    variables = values.split(',')
+    novar = values
+    variables = variables.split(',')
 
     ofile = 'get_Variables.nc'
@@ -16365 +16369 @@
     for var in variables:
         if not gen.searchInlist(filevars, var):
-            print errormsg
-            print '  ' + fname + ": file '" + ncfile + "' does not have variable '" +\
-              var + "' !!"
-            print '    variables in file:', filevars
-            quit(-1)
+            if novar == 'error':
+                print errormsg
+                print '  ' + fname + ": file '" + ncfile + "' does not have " +      \
+                  "variable '" + var + "' !!"
+                print '    variables in file:', filevars
+                quit(-1)
+            elif novar == 'warn':
+                print warnmsg
+                print '  ' + fname + ": file '" + ncfile + "' does not have " +      \
+                  "variable '" + var + "' !!"
+                print '    variables in file:', filevars
+                continue
+            else:
+                print errormsg
+                print '  ' + fname + ": novar '" + novar + "' not ready !!"
+                quit(-1)
 
         print '  ' + fname + ": getting variable '" + var + "'..."
@@ -16391 +16406 @@
         else:
             newvar = onewnc.createVariable(var, nctype(varinf.dtype), tuple(vardims)) 
-
+        newvar[:] = ovar[:]
         for attrn in varinf.attributes:
             attrv = ovar.getncattr(attrn)