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