import os
import sys
from PIL import Image
import subprocess
def compress_png_aggressively(input_path, quality):
    """直接覆蓋源文件的PNG壓縮方案"""
    try:
        # 1. 使用pngquant直接覆蓋源文件（添加--speed 1加快速度）
        subprocess.run([
            "pngquant",
            "--force",
            "--speed", "1",  # 加速處理
            "--quality", f"{max(10, 100 - quality)}-100",
            "--output", input_path,  # 直接輸出到原文件
            "--",
            input_path
        ], check=True)
        # 2. 直接運(yùn)行optipng覆蓋原文件（添加-clobber覆蓋寫(xiě)入）
        subprocess.run([
            "optipng",
            "-o5",
            "-quiet",
            "-clobber",  # 允許覆蓋原文件
            input_path
        ], check=True)
    except subprocess.CalledProcessError as e:
        print(f"壓縮失敗（已保留原文件）: {e}")
    except Exception as e:
        print(f"未知錯(cuò)誤: {e}")
def compress_image_by_quality(input_path, quality):
    """
    按質(zhì)量壓縮圖片并覆蓋原文件
    :param input_path: 圖片路徑
    :param quality: 壓縮質(zhì)量 (0-100)
    """
    if not os.path.isfile(input_path):
        print(f"文件不存在: {input_path}")
        return
    file_ext = os.path.splitext(input_path)[1].lower()
    supported_formats = ['.jpg', '.jpeg', '.png', '.webp']
    if file_ext not in supported_formats:
        print(f"不支持的圖片格式: {input_path}")
        return
    original_size = os.path.getsize(input_path) / 1024  # KB
    try:
        if file_ext in ['.jpg', '.jpeg']:
            img = Image.open(input_path)
            img.save(input_path, quality=quality, optimize=True, progressive=True)
        elif file_ext == '.png':
            # 對(duì)PNG使用強(qiáng)力壓縮
            compress_png_aggressively(input_path, quality)
        elif file_ext == '.webp':
            img = Image.open(input_path)
            img.save(input_path, quality=quality, method=6)
        compressed_size = os.path.getsize(input_path) / 1024
        print(
            f"壓縮完成: {os.path.basename(input_path)} 質(zhì)量: {quality} 大小: {original_size:.2f}KB -> {compressed_size:.2f}KB")
    except Exception as e:
        print(f"壓縮過(guò)程中出錯(cuò): {e}")
def compress_images_in_directory(directory, quality):
    """
    壓縮目錄下的所有圖片
    :param directory: 目錄路徑
    :param quality: 壓縮質(zhì)量 (0-100)
    """
    if not os.path.isdir(directory):
        print(f"目錄不存在: {directory}")
        return
    print(f"開(kāi)始?jí)嚎s目錄: {directory}，質(zhì)量設(shè)置: {quality}")
    for filename in os.listdir(directory):
        filepath = os.path.join(directory, filename)
        if os.path.isfile(filepath):
            compress_image_by_quality(filepath, quality)
    print("所有圖片壓縮完成")
if __name__ == "__main__":
    # 檢查必要工具是否安裝
    try:
        subprocess.run(["pngquant", "--version"], check=True, stdout=subprocess.PIPE)
        subprocess.run(["optipng", "--version"], check=True, stdout=subprocess.PIPE)
    except:
        print("警告: 請(qǐng)先安裝pngquant和optipng以獲得最佳PNG壓縮效果")
        print("Ubuntu/Debian: sudo apt-get install pngquant optipng")
        print("CentOS/RHEL: sudo yum install pngquant optipng")
        print("MacOS: brew install pngquant optipng")
        sys.exit(1)  # 工具未安裝時(shí)直接退出
    directory = r"/www/wwwroot/www.youluoyuan.com/wp-content/uploads/2025/01" #含有圖片的目錄
    try:
        quality = 60 #壓縮質(zhì)量 范圍1~100，建議50~75，數(shù)字越小則壓縮越強(qiáng)，但圖片質(zhì)量會(huì)更低
        if not 0 <= quality <= 100:
            raise ValueError
    except ValueError:
        print("錯(cuò)誤: 質(zhì)量參數(shù)必須是0到100之間的整數(shù)")
        sys.exit(1)
    try:
        compress_images_in_directory(directory, quality)
        print("\n? 已全部壓縮完成")  # 新增完成提示
        sys.exit(0)  # 正常退出
    except Exception as e:
        print(f"\n? 壓縮過(guò)程中發(fā)生嚴(yán)重錯(cuò)誤: {e}")
        sys.exit(1)  # 異常退出

# Ubuntu/Debian
sudo apt-get install pngquant optipng
# CentOS/RHEL
sudo yum install pngquant optipng
# MacOS
brew install pngquant optipng

pngquant --version
optipng -version