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)
