#首先导入模块
import os
import os.path
#定义两个参数,一个是文件夹名称,另外一个是后缀
def findFile(path,suffix):
'''目的就是找到文件夹中以suffix结尾的文件'''
ls = os.listdir(path)
for temp in ls:
#路径已经路径下的文件拼接
tempPath = os.path.join(path,temp)
#判断路径是否是文件
if os.path.isfile(tempPath):
#后缀以什么结尾
if temp.endswith(suffix):
#添加到这个展示列表
fileList.append(tempPath)
else:
findFile(tempPath,suffix)
#定义主函数
def main():
#定义全局变量fileList
global fileList
fileList = []
path = input('请您输入文件夹的路径:')
suffix = input('请您输入想查询以什么后缀结尾的文件:')
#判断路径是否是目录
if os.path.isdir(path):
findFile(path,suffix)
print(fileList)
else:
print('文件夹不存在')
main()