python使用zipfile递归压缩和解压缩文件

 

import shutil,zipfile,os

class ToolModel(object):

    def dfs_get_zip_file(self,input_path, result, ignore=[]):
        \'\'\'
        递归目录
        :param input_path: 输入路径
        :param result: 列表
        :param ignore: 忽略文件或目录名
        :return:
        \'\'\'
        files = os.listdir(input_path)
        for file in files:
            filePath = input_path + \'/\' + file
            if file in ignore:
                continue
            if os.path.isdir(filePath):
                self.dfs_get_zip_file(filePath, result, ignore)
            else:
                result.append(filePath)

    def zip_path(self,input_path, output_path, ignore=[]):
        \'\'\'
        :param input_path:  输入路径   /app/adminkit
        :param output_path: 输出路径  /app/adminkit.zip
        :param ignore: 忽略文件或目录名
        :return:
        \'\'\'
        outdir = os.path.dirname(output_path)
        if not os.path.isdir(outdir):
            os.makedirs(outdir)
        f = zipfile.ZipFile(output_path, \'w\', zipfile.ZIP_DEFLATED)
        filelists = []
        self.dfs_get_zip_file(input_path, filelists, ignore)
        for file in filelists:
            file = file.replace(\'\\\', \'/\')
            input_path = input_path.replace(\'\\\', \'/\')
            f.write(file, file.replace(input_path, \'\'))
        f.close()
        return output_path

    def unzip(self,filename: str,dirname):
        \'\'\'
        解压缩
        :param filename: 压缩文件名
        :param dirname: 解压缩输出目录
        :return:
        \'\'\'
        try:
            file = zipfile.ZipFile(filename)
            file.extractall(dirname)
            file.close()
            # 递归修复编码
            self.rename(dirname)
        except:
            print(f\'{filename} unzip fail\')

    def rename(self,pwd: str, filename=\'\'):
        """压缩包内部文件有中文名, 解压后出现乱码，进行恢复"""

        path = f\'{pwd}/{filename}\'
        if os.path.isdir(path):
            for i in os.scandir(path):
                self.rename(path, i.name)
        newname = filename.encode(\'cp437\').decode(\'gbk\')
        os.rename(path, f\'{pwd}/{newname}\')

    def del_file(self,filepath):
        """
        删除指定路径下的所有文件和文件夹
        :param filepath: 路径
        :return:
        """
        del_list = os.listdir(filepath)
        for f in del_list:
            file_path = os.path.join(filepath, f)
            if os.path.isfile(file_path):
                os.remove(file_path)
            elif os.path.isdir(file_path):
                shutil.rmtree(file_path)