python zip 解压，压缩相关工具

# !/usr/bin/env python3
# -*- coding: UTF-8 -*-
"""
@author  : [email protected]
@des     : zip 解压，压缩 相关
        仅支持 .zip
"""
import re
import os
import json
import zipfile


def unzip(zip_path,save_path):
    """
    同时解压同一目录下的多个压缩包
    :param zip_path: 压缩包保存文件夹路径
    :param save_path: 解压后保存文件夹路径
    :return:
    """
    map = {
    
    }
    map["success"] = []
    map["error"] = []
    for name in os.listdir(zip_path):
        try:
            _save_path = os.path.join(save_path,re.sub("\.|zip","",name))
            os.makedirs(_save_path,exist_ok=True)
            _zip_file = os.path.join(zip_path,name)
            zip = zipfile.ZipFile(_zip_file,"r")
            zip.extractall(_save_path)
            zip.close()
            map["success"].append({
    
    zip_path:name})
        except Exception as e:
            print(e,e.__traceback__.tb_lineno)
            map["error"].append({
    
    zip_path:name})
    print(json.dumps(map,ensure_ascii=False,indent=4))
    return map

def zip_single(_path,save_file):
    """
    压缩 单个文件夹
    :param _path: 待压缩文件夹位置
    :param save_name: 压缩后文件保存位置 save_name.zip
    :return:
    """
    map = {
    
    }
    map["success"] = []
    map["error"] = []
    z = zipfile.ZipFile(save_file, 'w', zipfile.ZIP_DEFLATED)
    success_num = 0
    name_list = os.listdir(_path)
    for name in name_list:
        try:
            file = os.path.join(_path,name)
            z.write(file,name)
            print(file,name)
            success_num += 1
            map["success"].append({
    
    _path:name})
        except Exception as e:
            print(e, e.__traceback__.tb_lineno)
            map["error"].append({
    
    _path:name})
    z.close()
    if len(map["error"]) == 0:
        print(f"{
      
      _path}下的文件 全部压缩成功")
    return map