Python 通配符删除文件

  • 2020 年 1 月 10 日
  • 笔记

# -*- 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