python筛选特定后缀的文件，并输出其路径

import os

path = input("请输入要筛选的文件夹路径")
def get_files(path, suffix, res):#路径，后缀，保存筛选路径信息
	datanames = os.listdir(path)
	for dataname in datanames:
		rpath = os.path.join(path, dataname)#若文件夹下有文件夹用rpath保存
		if os.path.isfile(rpath):
			if dataname.endswith(suffix):#判断文件后缀
				res.append(os.path.join(path, dataname))
		elif os.path.isdir(rpath):
			get_files(rpath, suffix, res)#递归调用

if __name__ == '__main__':
	reslist = []
	suffixlist = ['.class','.exe']
	for suf in range(len(suffixlist )):#将筛选后的路径信息输出到文件中
		get_files(path, suffixlist[suf], reslist)
		fp = open(path+'/log.txt','w')
		for r in reslist:
			print(r, file=fp)
		fp.close
print('请查看'+path+'\log.txt'+'此文件')