Python 通配符删除文件

最新推荐文章于 2024-07-15 21:22:53 发布

转载最新推荐文章于 2024-07-15 21:22:53 发布 · 8.7k 阅读

文章标签：

#python

Python 专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一个Python脚本，该脚本能够实现批量查找并删除指定目录及其子目录下的特定类型文件的功能。脚本提供了两种模式：一种是仅在当前目录下操作，另一种则包括了所有子目录。通过通配符定义文件类型，方便灵活地应用于多种场景。

# -*- coding: utf-8 -*-
"""
使用通配符，获取所有文件，或进行操作。
"""
import glob
import os

def files(curr_dir = '.', ext = '*.exe'):
    """当前目录下的文件"""
    for i in glob.glob(os.path.join(curr_dir, ext)):
        yield i

def all_files(rootdir, ext):
    """当前目录下以及子目录的文件"""
    for name in os.listdir(rootdir):
        if os.path.isdir(os.path.join(rootdir, name)):
            try:
                for i in all_files(os.path.join(rootdir, name), ext):
                    yield i
            except:
                pass
    for i in files(rootdir, ext):
        yield i

def remove_files(rootdir, ext, show = False):
    """删除rootdir目录下的符合的文件"""
    for i in files(rootdir, ext):
        if show:
            print i
        os.remove(i)

def remove_all_files(rootdir, ext, show = False):
    """删除rootdir目录下以及子目录下符合的文件"""
    for i in all_files(rootdir, ext):
        if show:
            print i
        os.remove(i)

if __name__ == '__main__':

    remove_all_files('.', '*.o', show = True)
    # remove_all_files('.', '*.exe', show = True)
    remove_files('.', '*.exe', show = True)
    # for i in files('.','*.c'):
        # print i