Gitで整理！画像を200KB圧縮＋コピーライト追加【第2回：圧縮モジュール】

import os
from utils.compress import compress_and_resize_jpeg, compress_image_to_target_size
from utils.add_text import add_text_with_background
from utils.date_utils import extract_photo_date_from_filename
from config.config import quality, width, target_size_kb, image_file_dir, save_dir, font_path  # ← 設定をインポート

# 出力ディレクトリがなければ作成
os.makedirs(save_dir, exist_ok=True)

# 入力フォルダ内のJPEG画像を順に処理
for filename in os.listdir(image_file_dir):
    if filename.lower().endswith((".jpg", ".jpeg")):
        input_path = os.path.join(image_file_dir, filename)
        print(f"処理対象: {filename}")

        base_name = os.path.splitext(filename)[0]
        output_filename = f"compress_{base_name}.jpg"
        output_path = os.path.join(save_dir, output_filename)

        # Step 1: リサイズ＋Exif除去処理（品質95で一時保存）
        temp_path = output_path + ".temp.jpg"
        print("▶ Step 1: リサイズ＋Exif除去")
        compress_and_resize_jpeg(input_path, temp_path, quality=95, width=width)

        if os.path.exists(temp_path):
            size_kb = os.path.getsize(temp_path) / 1024
            print(f"初期サイズ: {size_kb:.2f} KB")

        # Step 2: 目標サイズ以下に再圧縮
        print("▶ Step 2: サイズ調整（目標: {} KB）".format(target_size_kb))
        compress_image_to_target_size(temp_path, output_path, target_size_kb=target_size_kb)

        # Step 3: 画像に日付＋著作権テキストを追加
        if os.path.exists(output_path):
            photo_date = extract_photo_date_from_filename(filename)
            text = f"{photo_date} ©ふくゆー"
            print("▶ Step 3: テキスト追加")
            add_text_with_background(output_path, output_path, text, font_path)

            final_size_kb = os.path.getsize(output_path) / 1024
            print(f"最終サイズ: {final_size_kb:.2f} KB")
            print("-" * 40)

        # Step 4: 一時ファイルを削除
        if os.path.exists(temp_path):
            os.remove(temp_path)

print("✅ 全処理完了")

from PIL import Image
import os

def compress_and_resize_jpeg(input_path, output_path, quality, width):
    # 元画像を開き、RGBに変換（JPEG形式に対応）
    img = Image.open(input_path).convert("RGB")

    # アスペクト比を保持しながらリサイズ
    aspect_ratio = img.height / img.width
    
    # 指定した横幅に合わせて、高さを計算
    height = int(width * aspect_ratio)

    # 高品質なリサンプリング（LANCZOS）でリサイズ
    img = img.resize((width, height), Image.Resampling.LANCZOS)

    # Exifなどのメタデータを完全に削除するため、新しい画像に貼り付ける
    clean_img = Image.new(img.mode, img.size)
    clean_img.putdata(list(img.getdata()))

    # JPEGで保存（optimizeオプション付き）
    clean_img.save(output_path, "JPEG", quality=quality, optimize=True)

    print(f"Saved {output_path} (Exif removed, quality={quality}, size=({width}x{height}))")

def compress_image_to_target_size(input_path, output_path, target_size_kb, initial_quality=85, step=5):
    # 画像を開く
    img = Image.open(input_path)
    
    quality = initial_quality
    while True:
        # 一時的に画像を保存してサイズを確認
        img.save(output_path, "JPEG", quality=quality)
        
        # ファイルサイズを取得（KBに変換）
        file_size_kb = os.path.getsize(output_path) / 1024
        
        # ファイルサイズが目標以下になったら終了
        if file_size_kb <= target_size_kb or quality <= 5:
            print(f"Image compressed to {file_size_kb:.2f} KB with quality={quality}")
            break
        
        # 品質を下げて再度圧縮
        quality -= step