import os
import sys
import tinify

# 設(shè)置 Tinify API Key（請自行到 https://tinypng.com/developers 獲取）
TINIFY_API_KEY = "your_tinify_api_key"
tinify.key = TINIFY_API_KEY

# 支持的圖片格式
SUPPORTED_FORMATS = (".jpg", ".jpeg", ".png", ".webp")

# 遞歸壓縮圖片
def compress_images_in_directory(directory):
    total_original_size = 0
    total_compressed_size = 0
    compressed_count = 0

    # 遍歷目錄及子目錄
    for root, _, files in os.walk(directory):
        for file in files:
            if file.lower().endswith(SUPPORTED_FORMATS):  # 僅處理指定格式的圖片
                file_path = os.path.join(root, file)
                original_size = os.path.getsize(file_path)  # 獲取原始文件大小

                try:
                    # 壓縮圖片
                    source = tinify.from_file(file_path)
                    source.to_file(file_path)
                    
                    compressed_size = os.path.getsize(file_path)  # 獲取壓縮后大小
                    compressed_count += 1
                    total_original_size += original_size
                    total_compressed_size += compressed_size

                    # 輸出壓縮日志
                    saved_size = original_size - compressed_size
                    print(f"? 壓縮成功: {file_path} | 節(jié)省 {saved_size / 1024:.2f} KB")
                except tinify.errors.AccountError:
                    print("? 賬戶驗(yàn)證失敗，請檢查 API Key 是否正確")
                    sys.exit(1)
                except tinify.errors.ClientError:
                    print(f"? 無法處理圖片: {file_path}")
                except tinify.errors.ServerError:
                    print("? TinyPNG 服務(wù)器錯(cuò)誤，稍后再試")
                except Exception as e:
                    print(f"? 發(fā)生錯(cuò)誤: {e}")

    # 總結(jié)壓縮結(jié)果
    if compressed_count > 0:
        saved_space = total_original_size - total_compressed_size
        print("\n?? 壓縮統(tǒng)計(jì):")
        print(f"- 處理圖片數(shù)量: {compressed_count}")
        print(f"- 壓縮前總大小: {total_original_size / 1024:.2f} KB")
        print(f"- 壓縮后總大小: {total_compressed_size / 1024:.2f} KB")
        print(f"- 總共節(jié)省空間: {saved_space / 1024:.2f} KB")
    else:
        print("?? 未找到需要壓縮的圖片")

# 獲取命令行參數(shù)（目標(biāo)目錄）
if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("? 請輸入要壓縮的目錄路徑")
        sys.exit(1)

    target_directory = sys.argv[1]
    
    if not os.path.isdir(target_directory):
        print("? 指定的路徑不是一個(gè)有效目錄")
        sys.exit(1)

    print(f"?? 開始壓縮目錄: {target_directory}\n")
    compress_images_in_directory(target_directory)

pip install tinify

python compress_images.py "/your/image/directory"

python compress_images.py "./images"