python 遍历文件夹下全部文件并选择不同类型

最新推荐文章于 2023-02-23 15:39:47 发布

Y_z001

最新推荐文章于 2023-02-23 15:39:47 发布

阅读量341

点赞数

文章标签： python

本文链接：https://blog.csdn.net/Y_z001/article/details/126886829

版权

问题场景：文件夹和子文件夹深度不定，并且文件存放深度不定，需要挑选全部视频文件。

考虑使用os.walk函数进行遍历，并根据文件后缀挑选。

VIDEO_EXTENTIONS = ['.avi', '.AVI', '.mp4']  # 可以添加更多后缀类型


def is_video_file(filename):
    return any(filename.endswith(extension) for extension in VIDEO_EXTENSIONS)


def select_videos(dir_root):
    videos = []
    
    for root, _, fnames in os.walk(dir_root):
        for fname in fnames:   
            if is_video_file(fname):
                path = os.path.join(root, fname)
                videos.append(path)
    
    return videos