# -*- 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:     #默認直接返回所有文件名     fullfilename=os.path.join(FindPath,fn)     FileList.append(fullfilename)  #對文件名排序  if (len(FileList)>0):   FileList.sort()  return FileList

pip install wlab