Changeset 656 in lmdz_wrf for trunk

Timestamp:

Oct 15, 2015, 11:02:49 AM (9 years ago)

Author:

lfita

Message:

Adding `getting_atribute', function to get the value of a given attribute

Location:

trunk/tools

Files:

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

trunk/tools/nc_var.py

@@ -28 +28 @@
   'fdimadd', 'fgaddattr', 'field_stats', 'file_creation', 'file_oper_alongdims',     \
   'filter_2dim',                                                                     \
-  'flipdim', 'fvaradd', 'gaddattrk', 'gaddattr', 'get_namelist_vars',                \
+  'flipdim', 'fvaradd', 'gaddattrk', 'gaddattr', 'get_attribute',                    \
+  'get_namelist_vars',                                                               \
   'getvalues_lonlat', 'grattr',                                                      \
   'grmattr', 'igattrs', 'increaseDimvar', 'isgattrs', 'isvattrs', 'ivars', 'ivattrs',\
@@ -162 +163 @@
 elif oper == 'gaddattr':
     ncvar.gaddattr(opts.values, opts.ncfile)
+elif oper == 'get_attribute':
+    ncvar.get_attribute(opts.values, opts.ncfile, opts.varname)
 elif oper == 'get_namelist_vars':
     ncvar.get_namelist_vars(opts.values, opts.ncfile)

trunk/tools/nc_var_tools.py

@@ -17015 +17015 @@
 #DataSetSection_multivars('WRFt,20011111060000,20011111180000,-1','/home/lluis/PY/wrfout_d01_2001-11-11_00:00:00','all')
 
+def get_attribute(values, filen, varn):
+    """ Function to get an attribute from a netCDF file
+      values= [attrname]
+        attrname: attribute name
+      filen= name of the netCDF file
+      varn= name of the variable ('global' for a global attribute)
+    """
+    fname = 'get_attribute'
+    attrn = values
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print get_attribute.__doc__
+        quit()
+
+    onc = NetCDFFile(filen, 'r')
+
+    if varn == 'global':
+        if not searchInlist(onc.ncattrs(), attrn):
+            print errormsg
+            print '  ' + fname + ": file '" + filen + "' does not have atribute '" + \
+              attrn + "' !!"
+            quit(-1)
+        else:
+            attrv = onc.getncattr(attrn)
+    else:
+        if not searchInlist(onc.variables, varn): 
+            print errormsg
+            print '  ' + fname + ": file '" + filen + "' does not have variable '" + \
+              varn + "' !!"
+            quit(-1)
+        else:
+            ovar = onc.variables[varn]
+            if not searchInlist(ovar.ncattrs(), attrn):
+                print errormsg
+                print '  ' + fname + ": variable '" + varn + "' does not have " +    \
+                  "atribute '" + attrn + "' !!"
+                print '    it has:',ovar.ncattrs()
+                quit(-1)
+            else:
+                attrv = ovar.getncattr(attrn)
+
+    onc.close()
+
+    print attrv
+
+    return
+
+
 #quit()