import zipfile
import os
 
def create_zip(output_path, source_dir):
    with zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(source_dir):
            for file in files:
                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, source_dir)
                zipf.write(file_path, arcname)

def extract_zip(zip_path, extract_dir):
    with zipfile.ZipFile(zip_path, 'r') as zipf:
        zipf.extractall(extract_dir)  # 完整解壓
        # 示例：解壓特定文件
        # zipf.extract('docs/report.pdf', extract_dir)

def inspect_zip(zip_path):
    with zipfile.ZipFile(zip_path, 'r') as zipf:
        for info in zipf.infolist():
            print(f"Name: {info.filename}")
            print(f"Size: {info.file_size} bytes")
            print(f"Compressed: {info.compress_size} bytes")
            print(f"Modified: {info.date_time}")
            print("-" * 30)

def create_encrypted_zip(output_path, source_dir, password):
    with zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        zipf.setpassword(password.encode('utf-8'))
        # 添加文件...
        # 讀取時需使用：
        # zipf.open(name, pwd=password.encode())

def update_zip(zip_path, new_file):
    with zipfile.ZipFile(zip_path, 'a') as zipf:
        zipf.write(new_file, arcname=os.path.basename(new_file))

def split_zip(source_path, output_prefix, chunk_size=100*1024*1024):
    # 創(chuàng)建主壓縮包
    main_zip = f"{output_prefix}.zip"
    with zipfile.ZipFile(main_zip, 'w') as zipf:
        zipf.write(source_path, arcname=os.path.basename(source_path))
    
    # 拆分文件（偽代碼，需實現(xiàn)實際拆分邏輯）
    # split_file(main_zip, chunk_size, output_prefix)

from pathlib import Path
 
def normalize_path(path):
    return str(Path(path).resolve())

try:
    with zipfile.ZipFile('data.zip', 'r') as z:
        z.extractall('/protected/path')
except zipfile.BadZipFile:
    print("錯誤：文件已損壞")
except RuntimeError as e:
    if "Password required" in str(e):
        print("錯誤：需要密碼")
except PermissionError:
    print("錯誤：無寫入權(quán)限")