Changeset 1061 in lmdz_wrf for trunk

Timestamp:

Aug 30, 2016, 12:52:36 PM (9 years ago)

Author:

lfita

Message:

Adding optional values in `files_folder_HMT': folder='.',head=,middle=,tail=

File:

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

trunk/tools/generic_tools.py

@@ -5362 +5362 @@
     return runmean
 
-def files_folder_HMT(folder,head,middle,tail):
+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']
+    >>> files_folder_HMT(folder='/home/lluis/etudes/WRF_LMDZ/WaquaL_highres/tests/model_graphics/WRF/current', \
+      head='hfls', tail='.nc')
+    ['hfls_wrfout.nc', 'hfls_wrfout_tmean.nc']
     """
     import subprocess as sub
@@ -5389 +5392 @@
             indhead = ifile.index(head)
             indmiddle = ifile.index(middle)
+            if indmiddle == 0: indmiddle = indhead
             indtail = ifile.index(tail)
-            if indhead < indmiddle < indtail: listfiles.append(ifile)
+            if indtail == 0: indtail = inhead + inmiddle
+            if indhead <= indmiddle <= indtail: listfiles.append(ifile)
 
     return listfiles