Python实现图片批量无损压缩脚本

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. 直接运行optipng覆盖原文件（添加-clobber覆盖写入）
        subprocess.run([
            "optipng",
            "-o5",
            "-quiet",
            "-clobber",  # 允许覆盖原文件
            input_path
        ], check=True)
    except subprocess.CalledProcessError as e:
        print(f"压缩失败（已保留原文件）: {e}")
    except Exception as e:
        print(f"未知错误: {e}")
def compress_image_by_quality(input_path, quality):
    """
    按质量压缩图片并覆盖原文件
    :param input_path: 图片路径
    :param quality: 压缩质量 (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':
            # 对PNG使用强力压缩
            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)} 质量: {quality} 大小: {original_size:.2f}KB -> {compressed_size:.2f}KB")
    except Exception as e:
        print(f"压缩过程中出错: {e}")
def compress_images_in_directory(directory, quality):
    """
    压缩目录下的所有图片
    :param directory: 目录路径
    :param quality: 压缩质量 (0-100)
    """
    if not os.path.isdir(directory):
        print(f"目录不存在: {directory}")
        return
    print(f"开始压缩目录: {directory}，质量设置: {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("警告: 请先安装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)  # 工具未安装时直接退出
    directory = r"/www/wwwroot/www.youluoyuan.com/wp-content/uploads/2025/01" #含有图片的目录
    try:
        quality = 60 #压缩质量 范围1~100，建议50~75，数字越小则压缩越强，但图片质量会更低
        if not 0 <= quality <= 100:
            raise ValueError
    except ValueError:
        print("错误: 质量参数必须是0到100之间的整数")
        sys.exit(1)
    try:
        compress_images_in_directory(directory, quality)
        print("\n✅ 已全部压缩完成")  # 新增完成提示
        sys.exit(0)  # 正常退出
    except Exception as e:
        print(f"\n❌ 压缩过程中发生严重错误: {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