# -*- 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