Changeset 904 in lmdz_wrf for trunk/tools

Timestamp:

Jun 19, 2016, 7:59:25 PM (8 years ago)

Author:

lfita

Message:

Adding `files_folder_HMT': Function to retrieve a list of files from a folder [fold] and files with [head]*[middle]*[tail]

File:

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

trunk/tools/generic_tools.py

@@ -30 +30 @@
 # dictionary_key: Function to provide the first key in a dictionay with a given value
 # dictionary_key_list: Function to provide the first key in a dictionay of lists with a given value
+# files_folder_HMT: Function to retrieve a list of files from a folder [fold] and files with [head]*[middle]*[tail]
 # ijlonlat: Function to provide the imin,jmin imax,jmax of a lon,lat box
 # incomming_flow: Function to determine if a fgrid-flow inflows to the central grid point
@@ -5269 +5270 @@
     return runmean
 
+def files_folder_HMT(folder,head,middle,tail):
+    """ Function to retrieve a list of files from a folder [fold] and files with [head]*[middle]*[tail]
+    >>> files_folder_HMT('.','t','wrf','mean.nc')
+    ['clt_wrfout_tmean.nc', 'tas_wrfout_tmean.nc', 'ta_wrfout_xmean.nc']
+    """
+    import subprocess as sub
+    fname = 'files_folder'
+
+    if folder == 'h':
+        print fname + '_____________________________________________________________'
+        print files_folder.__doc__
+        quit()
+
+#    ins = folder + "/" + head + "*" + middle + '*' + tail
+    ins = folder + "/"
+#    print 'ins:',ins
+    files = sub.Popen(["/bin/ls","-1",ins], stdout=sub.PIPE)
+    fileslist = files.communicate()
+    listfiles0 = str(fileslist).replace("'",'').replace('(','').replace(')','').split('\\n')
+
+# Filtering output
+    listfiles = []
+    for ifile in listfiles0:
+        if ifile.find(head) != -1 and  ifile.find(middle) != -1 and                  \
+          ifile.find(tail) != -1:
+            indhead = ifile.index(head)
+            indmiddle = ifile.index(middle)
+            indtail = ifile.index(tail)
+            if indhead < indmiddle < indtail: listfiles.append(ifile)
+
+    return listfiles
+
 def files_folder(folder,headfile):
     """ Function to retrieve a list of files from a folder [fold] and head [headfile]