# -*- coding: utf-8 -*- 
#~ #------------------------------------------------------------------
#~ module:wlab 
#~ Filename:wgetfilelist.py 
#~ Function : 
#~ def IsSubString(SubStrList,Str) 
#~ def GetFileList(FindPath,FlagStr=[]): 
#~ 功能:讀取指定目錄下特定類型的文件名列表 
#~ Data: 2013-08-08,星期四 
#~ Author:吳徐平 
#~ Email:wxp07@qq.com 
#~ #------------------------------------------------------------------
#~ #------------------------------------------------------------------
def IsSubString(SubStrList,Str): 
 ''''' 
 #判斷字符串Str是否包含序列SubStrList中的每一個子字符串 
 #>>>SubStrList=['F','EMS','txt'] 
 #>>>Str='F06925EMS91.txt' 
 #>>>IsSubString(SubStrList,Str)#return True (or False) 
 ''' 
 flag=True 
 for substr in SubStrList: 
  if not(substr in Str): 
   flag=False 
 return flag 
#~ #---------------------------------------------------------------------- 
def GetFileList(FindPath,FlagStr=[]): 
 ''''' 
 #獲取目錄中指定的文件名 
 #>>>FlagStr=['F','EMS','txt'] #要求文件名稱中包含這些字符 
 #>>>FileList=GetFileList(FindPath,FlagStr) # 
 ''' 
 import os 
 FileList=[] 
 FileNames=os.listdir(FindPath) 
 if (len(FileNames)>0): 
  for fn in FileNames: 
   if (len(FlagStr)>0): 
    #返回指定類型的文件名 
    if (IsSubString(FlagStr,fn)): 
     fullfilename=os.path.join(FindPath,fn) 
     FileList.append(fullfilename) 
   else: 
    #默認(rèn)直接返回所有文件名 
    fullfilename=os.path.join(FindPath,fn) 
    FileList.append(fullfilename) 
 #對文件名排序 
 if (len(FileList)>0): 
  FileList.sort() 
 return FileList

pip install wlab