python获取指定路径下所有指定后缀的文件

"""
作者：皮蛋编程(https://www.pidancode.com)
创建日期：2022/3/25
功能描述：python获取指定路径下所有指定后缀的文件
"""
import os
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

print(GetFileFromThisRootDir('./pidancode/',ext='.py'))