查看目录下所有某种类型的文件

直接看源码：

import os
import fnmatch
    
    
def iterfindfile(root, fname_exp="*.*"):
    for path, dirs, files in os.walk(root):
        # fnmatch.filter()  判断文件类型
        for filename in fnmatch.filter(files, fname_exp):
            yield os.path.join(path, filename)
    
    
def tester(directory_path, filetype):
    file_type = '*.' + filetype
    for filename in iterfindfile(directory_path, file_type):
        print(filename)
    
    
if __name__ == "__main__":     
    directory_path = input('输入文件夹的路径：')
    filetype = input('输入将要查找的文件类型：')
    # 默认查找当前目录中的py文件
    if not directory_path:
        directory_path = '.'
    if not filetype:
        filetype = 'py'
    tester(directory_path,filetype)