1 #!/usr/local/bin/python3 2 # -*- coding:utf-8 -*- 3 import sys, os 4 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) 5 sys.path.append(BASE_DIR) 6 7 from tools.project_path import * 8 9 import platform 10 class RemoveTagFile(object): 11 path=None 12 def removeFile(self,path,remove_list,retain_list): #path后面要跟/ 13 self.path=path 14 system_test=platform.system() 15 if(system_test=='Windows'): 16 path_last=self.path[-1] 17 if(path_last!='\' ): 18 self.path=self.path+'\' 19 elif(system_test=='Linux'): 20 path_last = self.path[-1] 21 if (path_last != '/'): 22 self.path = self.path + '/' 23 if(len(remove_list)==0 and len(retain_list)==0): #如果remove_list,retain_list都是空则删除path目录下所有文件及文件夹 24 self.remove_file(self.eachFile(self.path)) 25 elif(len(remove_list)>0 and len(retain_list)==0): 26 self.remove_file(remove_list) 27 elif(len(remove_list)==0 and len(retain_list)>0): 28 list=self.eachFile(self.path) 29 for f in retain_list: 30 if(f in list): 31 list.remove(f) 32 else: 33 print('There is no file in the directory!') 34 self.remove_file(list) 35 elif (len(remove_list) > 0 and len(retain_list) > 0): 36 for f in retain_list: 37 if(f in remove_list): 38 remove_list.remove(f) 39 self.remove_file(remove_list) 40 41 def remove_file(self,file_list): 42 for filename in file_list: 43 if(os.path.exists(self.path+filename)): #判断文件是否存在 44 if(os.path.isdir(self.path+filename)): 45 self.del_file(self.path+filename) 46 else: 47 if(os.path.exists(self.path+filename)): 48 os.remove(self.path+filename) 49 else: 50 print(self.path+filename+' is not exist!') 51 for filename in file_list: 52 if(os.path.exists(self.path+filename)): 53 self.del_dir(self.path+filename) 54 def del_file(self,path): #递归删除目录及其子目录下的文件 55 for i in os.listdir(path): 56 path_file = os.path.join(path, i) #取文件绝对路径 57 if os.path.isfile(path_file): #判断是否是文件 58 os.remove(path_file) 59 else: 60 self.del_file(path_file) 61 def del_dir(self,path): #删除文件夹 62 for j in os.listdir(path): 63 path_file = os.path.join(path, j) # 取文件绝对路径 64 if not os.listdir(path_file): #判断文件如果为空 65 os.removedirs(path_file) #则删除该空文件夹，如果不为空删除会报异常 66 else: 67 self.del_dir(path_file) 68 def eachFile(self,filepath): #获取目录下所有文件的名称 69 pathDir = os.listdir(filepath) 70 list=[] 71 for allDir in pathDir: 72 child = os.path.join('%s%s' % (filepath, allDir)) 73 fileName=child.replace(filepath,'') 74 list.append(fileName) 75 return list 76 if __name__ == '__main__': 77 rtf=RemoveTagFile() 78 #以下表示只删除D:Test目录下的a文件夹、a.txt文件、b.txt文件 79 """ 80 规则： 81 1、如果remove_list、retain_list都为空则删除path目录下所有文件及文件夹 82 2、如果remove_list为空、retain_list不为空，则删除不在retain_list中的所有文件及文件夹 83 3、如果remove_list不为空、retain_list为空，则删除在remove_list中的所有文件及文件夹 84 4、如果remove_list、retain_list都不为空，则删除不在retain_list中且在remove_list中的所有文件及文件夹 85 """ 86 # path = 'D:Test' 87 # remove_list = ['a', 'a.txt', 'b.txt'] # 要删除的文件名称 88 # retain_list = ['c.txt'] # 要保留的文件名称 89 # rtf.removeFile(path,remove_list,retain_list) 90 rtf.del_file(report_path)