python 获取当前目录\指定目录下的所有指定格式的文件

import os
#获取当前目录下的指定格式文件
def getFiles(suffix):                   #suffix为str，示例:   '.txt'      '.json'       '.h5'
	BASE_DIR = os.path.dirname(os.path.abspath(__file__))       #这里如果有报错，则在__file__上加引号
	res = []
	for root,directory,files in os.walk(BASE_DIR):
    	for filename in files:
        	name,suf = os.path.splitext(filename)
        	if suf == suffix:
           		res.append(os.path.join(root,filename))
	return(res)    
#获取指定目录下的指定格式的文件
dir = '/usr/file'
def getFiles(dir,suffix):
	res = []
	for root,directory,files in os.walk(dir):
    	for filename in files:
        	name,suf = os.path.splitext(filename)
        	if suf == suffix:
           		res.append(os.path.join(root,filename))
	return(res)