python獲取指定路徑下所有指定后綴文件的方法

# 獲取指定路徑下所有指定后綴的文件
# dir 指定路徑
# ext 指定后綴，鏈表&不需要帶點 或者不指定。例子：['xml', 'java']
def GetFileFromThisRootDir(dir,ext = None):
 allfiles = []
 needExtFilter = (ext != None)
 for root,dirs,files in os.walk(dir):
 for filespath in files:
 filepath = os.path.join(root, filespath)
 extension = os.path.splitext(filepath)[1][1:]
 if needExtFilter and extension in ext:
 allfiles.append(filepath)
 elif not needExtFilter:
 allfiles.append(filepath)
 return allfiles