Changeset 1928 in lmdz_wrf for trunk

Timestamp:

Jul 13, 2018, 3:20:51 PM (6 years ago)

Author:

lfita

Message:

Enhancing `varrm' by adding possibility of ',' separated list of variables

File:

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

trunk/tools/nc_var_tools.py

@@ -1690 +1690 @@
   """ Removing a variable from a file
   ncfile = netCDF object file
-  var = variable name to remove
+  var = ',' separated list of variables' name to remove
   """
   import shutil as shu
@@ -1704 +1704 @@
   ncf.close()
 
-  if not gen.searchInlist(ncvars, var):
-      print '  ' + fname + ": File '" + ncfile + "' does not have variable: '" +     \
-        var + "' !!"
-      print '  variables:', ncvars
-      quit(-1)
+  varns = gen.str_list(var, ',')
+
+  for var in varns:
+      if not gen.searchInlist(ncvars, var):
+          print '  ' + fname + ": File '" + ncfile + "' does not have variable: '" + \
+            var + "' !!"
+          print '  variables:', ncvars
+          quit(-1)
 
   tmpncf = NetCDFFile('tmp_py.nc' , 'w')