Python 通配符删除文件

来源:互联网 发布:泰和安主机编程程序 编辑:程序博客网 时间:2024/05/20 00:53
# -*- coding: utf-8 -*-"""使用通配符,获取所有文件,或进行操作。"""import globimport osdef files(curr_dir = '.', ext = '*.exe'):    """当前目录下的文件"""    for i in glob.glob(os.path.join(curr_dir, ext)):        yield idef 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 idef 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
原创粉丝点击