使用通配符操作文件夹下所有符合文件

[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