import os
import shutil
import zipfile
# 首先引入需要的工具包
# shutil為后期移動(dòng)文件所需，可以忽略此項(xiàng)
# 路徑改這里！
parent_path = r'輸入路徑，會(huì)解壓該路徑下的所有zip壓縮文件'
# 文件類型選擇
# 可以自行更改壓縮文件類型，需要引入其它工具包，如tarfile等
# 這里是因?yàn)樵谧约旱膚indows上，zip比較常見，其他類型請(qǐng)自行更改
file_flag = '.zip'
# 刪除已解壓的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
# 因?yàn)槲以谑褂眠^程中發(fā)現(xiàn)有些zip解包后會(huì)混在一起
# 在平時(shí)大家手動(dòng)解壓時(shí)可能也會(huì)遇到提示是否覆蓋的問題
# 下面的兩個(gè)函數(shù)解決這一問題
# 開始要先創(chuàng)建一個(gè)大文件夾 與壓縮包名字相同
# 避免后期混亂和麻煩
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):
  # 遞歸要注意信息的正常處理 搞不好上一個(gè)調(diào)用已經(jīng)改變了東西 而下面的調(diào)用還是使用之前的數(shù)據(jù)
  try:
    # 判斷文件夾重名 開始
    for item in os.listdir(os.path.join(root, father_dir_name)):
      # 第一步判斷是不是一個(gè)文件夾，如果不是則跳過本次循環(huán)
      if not os.path.isdir(os.path.join(root, father_dir_name, item)):
        continue
      # 判斷是否要脫掉一層目錄結(jié)構(gòu)
      # 文件夾名字要相同，且子目錄中只有單獨(dú)的一個(gè)文件夾
      if item == father_dir_name and len(
          os.listdir(os.path.join(root, father_dir_name))) == 1:
        # 改變工作目錄
        os.chdir(root)
        # 將無用文件夾重命名，因?yàn)橹苯右苿?dòng)會(huì)有重名錯(cuò)誤
        os.rename(father_dir_name, father_dir_name + '-old')
        # 移動(dòng)文件后刪除空文件夾
        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ù)的改動(dòng)，所以下面遞歸傳參一定要正確！
        rem_dir_extra(root, item)
      else:
        # 處理那些不滿足上面條件的文件夾
        rem_dir_extra(os.path.join(root, father_dir_name), item)
  except Exception as e:
    # 打印錯(cuò)誤信息
    print("清除文件夾出錯(cuò)" + 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)
          # 刪除解壓后的文件
          # 有點(diǎn)危險(xiǎn)
          # 但不刪除又可能會(huì)重復(fù)運(yùn)行
           # 一定要備份或先測(cè)試，不然可能會(huì)涼，自己選擇修改
          del_old_zip(zip_path)
          # 去掉多余的文件結(jié)構(gòu)
          rem_dir_extra(root, name.replace(file_flag, ''))
          print(f'{root}\\{name}'.join(['文件：', '\n解壓完成\n']))
  # 由于解壓可能解了好幾次 所以可能會(huì)有已經(jīng)解壓好的父級(jí)目錄重名無法處理 這里要再處理一次
  rem_dir_extra(os.path.split(parent_path)[0], os.path.split(parent_path)[1])
  print("解壓完成啦，記得檢查有沒有zip格式之外的呀!\n\n其他格式需要自己改一下了")