1
import os
2
import sys
3
from PIL import Image
4

5
def compress_png_images_recursive(root_directory, quality=85, optimize=True):
6
    """
7
    递归压缩指定根目录下所有子文件夹中的PNG图片（如img/001/P1.png）
8
    保持原文件名、原路径不变，无需修改Markdown中的图片链接
9

10
    Args:
11
        root_directory (str): 根目录（比如存放所有img子文件夹的目录）
12
        quality (int): PNG压缩质量（1-95）
13
        optimize (bool): 是否启用PNG优化
14
    """
15
    # 检查根目录是否存在
16
    if not os.path.isdir(root_directory):
17
        print(f"❌ 错误：根目录 {root_directory} 不存在！")
18
        return
19

20
    # 统计变量
21
    processed_count = 0
22
    total_original_size = 0
23
    total_compressed_size = 0
24

25
    # 递归遍历所有子目录和文件
26
    for root, dirs, files in os.walk(root_directory):
27
        for filename in files:
28
            # 只处理PNG文件
29
            if not filename.lower().endswith('.png'):
30
                continue
31

32
            # 拼接完整文件路径（如 img/001/P1.png）
33
            file_path = os.path.join(root, filename)
34

35
            try:
36
                # 获取原始文件大小
37
                original_size = os.path.getsize(file_path)
38
                total_original_size += original_size
39

40
                # 打开并压缩PNG图片（直接覆盖原文件，路径/名称不变）
41
                with Image.open(file_path) as img:
42
                    img.save(
43
                        file_path,
44
                        format='PNG',
45
                        quality=quality,
46
                        optimize=optimize,
47
                        compress_level=9  # PNG最高压缩级别
48
                    )
49

50
                # 获取压缩后的文件大小
51
                compressed_size = os.path.getsize(file_path)
52
                total_compressed_size += compressed_size
53

54
                # 计算压缩率
55
                compression_ratio = (1 - compressed_size / original_size) * 100
56

57
                # 输出压缩信息（显示相对路径，更清晰）
58
                relative_path = os.path.relpath(file_path, root_directory)
59
                print(f"✅ 已压缩：{relative_path}")
60
                print(f"   原大小：{original_size/1024:.2f} KB → 压缩后：{compressed_size/1024:.2f} KB")
61
                print(f"   压缩率：{compression_ratio:.2f}%\n")
62

63
                processed_count += 1
64

65
            except Exception as e:
66
                relative_path = os.path.relpath(file_path, root_directory)
67
                print(f"❌ 处理 {relative_path} 失败：{str(e)}\n")
68

69
    # 输出汇总信息
70
    print("="*60)
71
    print(f"📊 递归压缩完成！共处理 {processed_count} 个PNG文件")
72
    if processed_count > 0:
73
        total_ratio = (1 - total_compressed_size / total_original_size) * 100
74
        print(f"📈 总原始体积：{total_original_size/1024:.2f} KB")
75
        print(f"📉 总压缩体积：{total_compressed_size/1024:.2f} KB")
76
        print(f"🎯 整体压缩率：{total_ratio:.2f}%")
77

78
if __name__ == "__main__":
79
    # 方式1：命令行传参指定根目录（比如存放img文件夹的目录）
80
    if len(sys.argv) > 1:
81
        root_dir = sys.argv[1]
82
    # 方式2：默认使用当前目录（脚本所在目录）
83
    else:
84
        root_dir = os.getcwd()
85
        print(f"ℹ️  未指定目录，处理当前目录：{root_dir}\n")
86

87
    # 执行递归压缩（quality=85适配博客场景）
88
    compress_png_images_recursive(root_dir, quality=85)
89

90
    # ========== 关键新增：让窗口停留 ==========
91
    print("\n" + "="*60)
92
    print("✅ 所有操作完成！按回车键关闭窗口...")
93
    # 等待用户输入（按回车才会关闭）
94
    input()