import os
import shutil
import zipfile
 
# 首先引入需要的工具包
# shutil為后期移動文件所需，可以忽略此項 
# 路徑改這里！
#parent_path = r'輸入路徑，會解壓該路徑下的所有zip壓縮文件'
parent_path = r'E:\py\python3.7\test\test99\zip' 
# 文件類型選擇
# 可以自行更改壓縮文件類型，需要引入其它工具包，如tarfile等
# 這里是因為在自己的windows上，zip比較常見，其他類型請自行更改
file_flag = '.zip'   #修改需解壓的格式 例如：.rar
# 刪除已解壓的zip文件
# 不建議初次使用，在確定程序無誤后可以添加使用
def del_old_zip(file_path):
    os.remove(file_path)  
# 解壓
def decompress(file_path, root):
    # 開始
    # zipfile打開zip文件
    z = zipfile.ZipFile(f'{file_path}', 'r') 
    # 解壓
    z.extractall(path=f"{root}")    # path為解壓路徑，解包后位于該路徑下 
    # 判斷是否需要重復(fù)解包
    for names in z.namelist():
        if names.endswith(file_flag):
            z.close()
            return 1 
    # 結(jié)束
    z.close() 
    return 0
     
# 因為我在使用過程中發(fā)現(xiàn)有些zip解包后會混在一起
# 在平時大家手動解壓時可能也會遇到提示是否覆蓋的問題
# 下面的兩個函數(shù)解決這一問題 
# 開始要先創(chuàng)建一個大文件夾  與壓縮包名字相同
# 避免后期混亂和麻煩
def start_dir_make(root, dirname):
    os.chdir(root)
    os.mkdir(dirname)
    return os.path.join(root, dirname) 
# 去除多余文件夾
def rem_dir_extra(root, father_dir_name):
    # 遞歸要注意信息的正常處理  搞不好上一個調(diào)用已經(jīng)改變了東西  而下面的調(diào)用還是使用之前的數(shù)據(jù) 
    try: 
       # 判斷文件夾重名  開始
        for item in os.listdir(os.path.join(root, father_dir_name)): 
            # 第一步判斷是不是一個文件夾，如果不是則跳過本次循環(huán)
            if not os.path.isdir(os.path.join(root, father_dir_name, item)):
                continue 
            # 判斷是否要脫掉一層目錄結(jié)構(gòu)
            # 文件夾名字要相同，且子目錄中只有單獨的一個文件夾
            if item == father_dir_name and len(
                    os.listdir(os.path.join(root, father_dir_name))) == 1: 
                # 改變工作目錄
                os.chdir(root)                 
                # 將無用文件夾重命名，因為直接移動會有重名錯誤
                os.rename(father_dir_name, father_dir_name + '-old')                 
                # 移動文件后刪除空文件夾
                shutil.move(os.path.join(root, father_dir_name + '-old', item), os.path.join(root))
                os.rmdir(os.path.join(root, father_dir_name + '-old')) 
                # 將去掉一層目錄結(jié)構(gòu)后的文件夾繼續(xù)作為父本遞歸處理下去
                # 這里要注意，上面已經(jīng)發(fā)生過數(shù)據(jù)的改動，所以下面遞歸傳參一定要正確！
                rem_dir_extra(root, item) 
            else: 
                # 處理那些不滿足上面條件的文件夾
                rem_dir_extra(os.path.join(root, father_dir_name), item) 
    except Exception as e: 
        # 打印錯誤信息
        print("清除文件夾出錯" + str(e)) 
# 入口
if __name__ == '__main__': 
    flag = 1 
    while flag: 
        #  循環(huán)遍歷文件夾
        for root, dirs, files in os.walk(parent_path): 
            # 讀取文件名
            for name in files:
                if name.endswith(file_flag): 
                    # 創(chuàng)建文件夾
                    new_ws = start_dir_make(root, name.replace(file_flag, '')) 
                    # zip文件地址
                    zip_path = os.path.join(root, name) 
                    # 解壓
                    flag = decompress(zip_path, new_ws) 
                    # 刪除解壓后的文件
                    # 有點危險
                    # 但不刪除又可能會重復(fù)運行
                     # 一定要備份或先測試，不然可能會涼，自己選擇修改
                    del_old_zip(zip_path) 
                    # 去掉多余的文件結(jié)構(gòu)
                    rem_dir_extra(root, name.replace(file_flag, '')) 
                    print(f'{root}\\{name}'.join(['文件：', '\n解壓完成\n']))
  
    # 由于解壓可能解了好幾次 所以可能會有已經(jīng)解壓好的父級目錄重名無法處理 這里要再處理一次
    rem_dir_extra(os.path.split(parent_path)[0], os.path.split(parent_path)[1])  
    print("解壓完成啦，記得檢查有沒有zip格式之外的呀!\n\n其他格式需要自己改一下了")

import zipfile
import os

def zip_yasuo(start_dir):
    file_news = start_dir + '.zip'
    if not os.path.isfile(file_news):
        z = zipfile.ZipFile(file_news, 'w', zipfile.ZIP_DEFLATED)
        for dir_path, dir_names, file_names in os.walk(start_dir):
            file_path = dir_path.replace(start_dir, '')
            file_path = file_path and file_path + os.sep or ''
            for filename in file_names:
                z.write(os.path.join(dir_path, filename), file_path+filename)
        z.close()

if __name__ == "__main__":
    base_path = r"E:\py\python3.7\test\test99\zip"
    base_path_list = os.listdir(base_path)
    for base_path_list_one in base_path_list:
        base_path_list_one_dir = os.path.join(base_path,base_path_list_one)
        # 子目錄
        print("準(zhǔn)備壓縮需要壓縮的子目錄", base_path_list_one_dir)
        if os.path.isdir(base_path_list_one_dir):
            zip_yasuo(base_path_list_one_dir)