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)
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

哈希哥

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值