python删除指定目录下的指定文件和文件夹

具体代码:

#coding:utf-8
import os,sys,platform
class RemoveTagFile(object):
    path=None
    def removeFile(self,path,remove_list,retain_list): #path后面要跟/
        self.path=path
        system_test=platform.system()
        if(system_test=='Windows'):
            path_last=self.path[-1]
            if(path_last!='\\' ):
                self.path=self.path+'\\'
        elif(system_test=='Linux'):
            path_last = self.path[-1]
            if (path_last != '/'):
                self.path = self.path + '/'
        if(len(remove_list)==0 and len(retain_list)==0):  #如果remove_list,retain_list都是空则删除path目录下所有文件及文件夹
            self.remove_file(self.eachFile(self.path))
        elif(len(remove_list)>0 and len(retain_list)==0):
            self.remove_file(remove_list)
        elif(len(remove_list)==0 and len(retain_list)>0):
            list=self.eachFile(self.path)
            for f in retain_list:
                if(f in list):
                    list.remove(f)
                else:
                    print('There is no file in the directory!')
            self.remove_file(list)
        elif (len(remove_list) > 0 and len(retain_list) > 0):
            for f in retain_list:
                if(f in remove_list):
                    remove_list.remove(f)
            self.remove_file(remove_list)

    def remove_file(self,file_list):
        for filename in file_list:
            if(os.path.exists(self.path+filename)):   #判断文件是否存在
                if(os.path.isdir(self.path+filename)):
                    self.del_file(self.path+filename)
                else:
                    if(os.path.exists(self.path+filename)):
                        os.remove(self.path+filename)
            else:
                print(self.path+filename+' is not exist!')
        for filename in file_list:
            if(os.path.exists(self.path+filename)):
                self.del_dir(self.path+filename)
    def del_file(self,path):     #递归删除目录及其子目录下的文件
        for i in os.listdir(path):
            path_file = os.path.join(path, i) #取文件绝对路径
            if os.path.isfile(path_file):     #判断是否是文件
                os.remove(path_file)
            else:
                self.del_file(path_file)
    def del_dir(self,path):  #删除文件夹
        for j in os.listdir(path):
            path_file = os.path.join(path, j)  # 取文件绝对路径
            if not os.listdir(path_file):    #判断文件如果为空
                os.removedirs(path_file)   #则删除该空文件夹,如果不为空删除会报异常
            else:
                self.del_dir(path_file)
    def eachFile(self,filepath):   #获取目录下所有文件的名称
        pathDir = os.listdir(filepath)
        list=[]
        for allDir in pathDir:
            child = os.path.join('%s%s' % (filepath, allDir))
            fileName=child.replace(filepath,'')
            list.append(fileName)
            return list
if __name__ == '__main__':
    rtf=RemoveTagFile()
    #以下表示只删除D:\Test\目录下的a文件夹、a.txt文件、b.txt文件
    """
    规则:
    1、如果remove_list、retain_list都为空则删除path目录下所有文件及文件夹
    2、如果remove_list为空、retain_list不为空,则删除不在retain_list中的所有文件及文件夹
    3、如果remove_list不为空、retain_list为空,则删除在remove_list中的所有文件及文件夹
    4、如果remove_list、retain_list都不为空,则删除不在retain_list中且在remove_list中的所有文件及文件夹
    """
    path = 'D:\Test'
    remove_list = ['a', 'a.txt', 'b.txt']  # 要删除的文件名称
    retain_list = ['c.txt']  # 要保留的文件名称
    rtf.removeFile(path,remove_list,retain_list)

你可能感兴趣的:(python删除指定目录下的指定文件和文件夹)