import os
import zipfile

def compress_folder_to_zip(source_folder_path, output_zip_path, include_file_types=None):
    """
    批量壓縮指定文件夾及其所有內容到一個zip文件。
    這是高效的Python文件壓縮方法，實現(xiàn)批量打包文件。
    :param source_folder_path: 要壓縮的源文件夾路徑
    :param output_zip_path: 輸出zip文件的完整路徑 (例如 "project_archive.zip")
    :param include_file_types: 可選列表，只壓縮指定擴展名的文件 (如 ['.jpg', '.pdf'])
    """
    if not os.path.exists(source_folder_path):
        print(f"? 源文件夾不存在：{source_folder_path}")
        return

    # 確保輸出目錄存在
    os.makedirs(os.path.dirname(output_zip_path), exist_ok=True)

    print(f"?? 正在壓縮文件夾：'{source_folder_path}' 到 '{output_zip_path}'...")
    try:
        with zipfile.ZipFile(output_zip_path, 'w', zipfile.ZIP_DEFLATED) as zf:
            # os.walk() 遍歷文件夾及其子文件夾和文件
            for root, dirs, files in os.walk(source_folder_path):
                # 計算當前文件在zip文件中的相對路徑
                relative_path = os.path.relpath(root, source_folder_path)
                
                for file in files:
                    file_path = os.path.join(root, file)
                    # 檢查文件類型是否在需要包含的列表中
                    if include_file_types and os.path.splitext(file.lower())[1] not in include_file_types:
                        print(f"   ?? 跳過文件（類型不符）：{file}")
                        continue

                    # 將文件添加到zip文件，使用相對路徑作為zip內部路徑
                    zf.write(file_path, os.path.join(relative_path, file))
        
        # 獲取壓縮前后文件大小，進行直觀對比
        original_size_bytes = sum(os.path.getsize(os.path.join(r, f)) for r, d, fs in os.walk(source_folder_path) for f in fs)
        compressed_size_bytes = os.path.getsize(output_zip_path)

        original_size_mb = original_size_bytes / (1024 * 1024) 
        compressed_size_mb = compressed_size_bytes / (1024 * 1024) 
        savings_percentage = ((original_size_bytes - compressed_size_bytes) / original_size_bytes) * 100 if original_size_bytes > 0 else 0

        print(f"? 壓縮成功！")
        print(f"   原始總大小: {original_size_mb:.2f} MB")
        print(f"   壓縮后大小: {compressed_size_mb:.2f} MB")
        print(f"   節(jié)省空間: {savings_percentage:.2f}%")

    except Exception as e:
        print(f"? 壓縮失?。簕e}")

if __name__ == "__main__":
    # --- 請修改為你的實際文件夾路徑 ---
    source_folder_path = os.path.expanduser("~/Desktop/我的項目資料") 
    output_zip_path = os.path.expanduser("~/Desktop/項目資料_歸檔.zip")

    # 確保源文件夾存在并有測試文件，可以創(chuàng)建一些子文件夾和文件
    os.makedirs(source_folder_path, exist_ok=True)
    # 示例：創(chuàng)建一些測試文件用于壓縮
    with open(os.path.join(source_folder_path, "report.txt"), "w") as f: f.write("Hello World\n" * 5000)
    with open(os.path.join(source_folder_path, "image.jpg"), "w") as f: f.write("binary data" * 100000)
    os.makedirs(os.path.join(source_folder_path, "subfolder"), exist_ok=True)
    with open(os.path.join(source_folder_path, "subfolder", "data.csv"), "w") as f: f.write("1,2,3\n" * 2000)

    # 示例1：壓縮整個文件夾
    print("\n--- 示例1：批量壓縮整個文件夾 ---")
    compress_folder_to_zip(source_folder_path, output_zip_path)

    # 示例2：只壓縮特定類型的文件（如只壓縮圖片和PDF）
    # print("\n--- 示例2：只壓縮圖片和PDF ---")
    # compress_folder_to_zip(source_folder_path, os.path.expanduser("~/Desktop/圖片文檔.zip"), include_file_types=['.jpg', '.png', '.pdf'])

import os
import zipfile

def extract_zip_to_folder(zip_file_path, target_folder_path):
    """
    解壓縮zip文件到指定文件夾。
    這是高效的Python文件解壓工具。
    :param zip_file_path: 要解壓的zip文件完整路徑
    :param target_folder_path: 解壓目標文件夾路徑
    """
    if not os.path.exists(zip_file_path):
        print(f"? Zip文件不存在：{zip_file_path}")
        return

    os.makedirs(target_folder_path, exist_ok=True) # 確保目標文件夾存在

    print(f"?? 正在解壓 '{zip_file_path}' 到 '{target_folder_path}'...")
    try:
        with zipfile.ZipFile(zip_file_path, 'r') as zf:
            zf.extractall(target_folder_path) # 解壓所有內容
        print(f"? 解壓成功！所有文件已解壓到：{target_folder_path}")
    except Exception as e:
        print(f"? 解壓失敗：{e}")

if __name__ == "__main__":
    # --- 請修改為你的實際zip文件路徑和目標解壓路徑 ---
    # 假設你已經運行了2.1節(jié)的壓縮代碼，并生成了 '項目資料_歸檔.zip'
    source_zip_file = os.path.expanduser("~/Desktop/項目資料_歸檔.zip") 
    output_extracted_folder = os.path.expanduser("~/Desktop/解壓后資料")

    os.makedirs(output_extracted_folder, exist_ok=True) # 確保目標文件夾存在

    extract_zip_to_folder(source_zip_file, output_extracted_folder)