写真フォルダを一括圧縮してzipにまとめるスクリプト

import sys
import json
import os
import zipfile
from datetime import datetime
from concurrent.futures import ProcessPoolExecutor, as_completed
from PIL import Image

# 前提:
# - 対応形式: JPG/JPEG, PNG, WebP, BMP, TIFF
# - 出力ファイル名は元のファイル名・拡張子をそのまま維持
# - PNGは可逆圧縮のまま最適化のみ実施(透過情報を保持するため)
# - 圧縮度ごとのパラメータは画質と容量のバランスを考慮した一般的な値を採用
# - 並列処理はプロセスプールを使用(画像処理はCPUバウンドのためスレッドではなくプロセス並列)
# - ワーカー数は CPU コア数 - 1(最低1)
# - zipファイル名は「{局名}_写真_{タイムスタンプ}.zip」で出力先フォルダに保存
# - zip内には圧縮後の画像をフラット構造で格納(サブフォルダなし)
# - zip自体の圧縮はZIP_DEFLATED(画像は既に圧縮済みなので効果は限定的だが標準的な選択)
# - 圧縮画像を削除はzip作成成功 かつ testzip()整合性OK かつ ファイル数一致の場合のみ実行
# - サマリーログは出力先フォルダに「{局名}_ログ_{タイムスタンプ}.txt」で保存

# 圧縮度ごとの設定
圧縮設定 = {
    "軽め圧縮(画質キープ)": {"jpeg_quality": 85, "max_size": None},
    "軽め圧縮(画質キープ)": {"jpeg_quality": 85, "max_size": None},
    "標準圧縮(バランス型)": {"jpeg_quality": 75, "max_size": 2560},
    "標準圧縮(バランス型)": {"jpeg_quality": 75, "max_size": 2560},
    "強め圧縮(容量最優先)": {"jpeg_quality": 60, "max_size": 1920},
    "強め圧縮(容量最優先)": {"jpeg_quality": 60, "max_size": 1920},
}

対応拡張子 = {".jpg", ".jpeg", ".png", ".webp", ".bmp", ".tif", ".tiff"}


def 圧縮処理(args):
    """1ファイル分の圧縮処理。プロセスプールから呼ばれる。"""
    入力パス, 出力パス, jpeg_quality, max_size = args
    ファイル名 = os.path.basename(入力パス)
    拡張子 = os.path.splitext(ファイル名)[1].lower()

    try:
        元サイズ = os.path.getsize(入力パス)

        with Image.open(入力パス) as img:
            exif = img.info.get("exif")

            if max_size is not None:
                w, h = img.size
                長辺 = max(w, h)
                if 長辺 > max_size:
                    比率 = max_size / 長辺
                    新w = int(w * 比率)
                    新h = int(h * 比率)
                    img = img.resize((新w, 新h), Image.LANCZOS)

            if 拡張子 in (".jpg", ".jpeg"):
                if img.mode not in ("RGB", "L"):
                    img = img.convert("RGB")
                save_kwargs = {
                    "format": "JPEG",
                    "quality": jpeg_quality,
                    "optimize": True,
                    "progressive": True,
                }
                if exif:
                    save_kwargs["exif"] = exif
                img.save(出力パス, **save_kwargs)

            elif 拡張子 == ".png":
                img.save(出力パス, format="PNG", optimize=True)

            elif 拡張子 == ".webp":
                img.save(出力パス, format="WEBP", quality=jpeg_quality, method=6)

            elif 拡張子 == ".bmp":
                img.save(出力パス, format="BMP")

            elif 拡張子 in (".tif", ".tiff"):
                img.save(出力パス, format="TIFF", compression="tiff_lzw")

        圧縮後サイズ = os.path.getsize(出力パス)
        return {
            "ok": True,
            "ファイル名": ファイル名,
            "元サイズ": 元サイズ,
            "圧縮後サイズ": 圧縮後サイズ,
        }

    except Exception as e:
        return {
            "ok": False,
            "ファイル名": ファイル名,
            "エラー": str(e),
        }


def main():
    with open(sys.argv[1], encoding="utf-8") as f:
        inputs = json.load(f)

    写真フォルダ = inputs["写真フォルダ"]
    出力先フォルダ = inputs["出力先フォルダ"]
    圧縮度 = inputs["圧縮度"]
    局名 = inputs["zipファイル名（作成しない場合でも入力要）"]
    zip作成フラグ = inputs["zipファイルを作成"] == "true"
    画像削除フラグ = inputs["圧縮画像を削除"] == "true"

    if 圧縮度 not in 圧縮設定:
        normalized = 圧縮度.replace("(", "(").replace(")", ")")
        if normalized in 圧縮設定:
            圧縮度 = normalized
        else:
            print(f"エラー: 圧縮度の値が不正です: {圧縮度}", file=sys.stderr)
            sys.exit(1)

    設定 = 圧縮設定[圧縮度]
    jpeg_quality = 設定["jpeg_quality"]
    max_size = 設定["max_size"]

    コア数 = os.cpu_count() or 1
    ワーカー数 = max(1, コア数 - 1)
    タイムスタンプ = datetime.now().strftime("%Y%m%d_%H%M%S")

    print(f"局名: {局名}")
    print(f"圧縮度: {圧縮度}")
    print(f"JPEG品質: {jpeg_quality}")
    print(f"長辺最大: {max_size if max_size else '元サイズ維持'}")
    print(f"CPUコア数: {コア数} / 並列ワーカー数: {ワーカー数}")
    print(f"zip作成: {'あり' if zip作成フラグ else 'なし'}")
    print(f"圧縮画像を削除: {'あり（zip整合性・ファイル数確認後）' if 画像削除フラグ else 'なし'}")
    print(f"入力フォルダ: {写真フォルダ}")
    print(f"出力フォルダ: {出力先フォルダ}")
    print("-" * 50)

    os.makedirs(出力先フォルダ, exist_ok=True)

    対象ファイル = []
    for ファイル名 in os.listdir(写真フォルダ):
        フルパス = os.path.join(写真フォルダ, ファイル名)
        if os.path.isfile(フルパス):
            拡張子 = os.path.splitext(ファイル名)[1].lower()
            if 拡張子 in 対応拡張子:
                対象ファイル.append(ファイル名)

    総数 = len(対象ファイル)
    if 総数 == 0:
        print("対応する画像ファイルが見つかりませんでした。")
        return

    print(f"対象ファイル数: {総数}件")
    print("-" * 50)

    タスク一覧 = [
        (
            os.path.join(写真フォルダ, ファイル名),
            os.path.join(出力先フォルダ, ファイル名),
            jpeg_quality,
            max_size,
        )
        for ファイル名 in 対象ファイル
    ]

    成功数 = 0
    失敗数 = 0
    合計元サイズ = 0
    合計圧縮後サイズ = 0
    完了数 = 0
    成功ファイル一覧 = []
    失敗ファイル一覧 = []
    ファイル別結果 = []  # サマリー用

    with ProcessPoolExecutor(max_workers=ワーカー数) as executor:
        future_to_name = {
            executor.submit(圧縮処理, タスク): タスク[0] for タスク in タスク一覧
        }

        for future in as_completed(future_to_name):
            完了数 += 1
            結果 = future.result()

            if 結果["ok"]:
                元 = 結果["元サイズ"]
                後 = 結果["圧縮後サイズ"]
                合計元サイズ += 元
                合計圧縮後サイズ += 後
                削減率 = (1 - 後 / 元) * 100 if 元 > 0 else 0
                print(
                    f"[{完了数}/{総数}] {結果['ファイル名']}  "
                    f"{元/1024:.1f}KB → {後/1024:.1f}KB  "
                    f"({削減率:+.1f}%)"
                )
                成功数 += 1
                成功ファイル一覧.append(結果["ファイル名"])
                ファイル別結果.append({
                    "ファイル名": 結果["ファイル名"],
                    "元サイズ": 元,
                    "圧縮後サイズ": 後,
                    "削減率": 削減率,
                    "状態": "成功",
                })
            else:
                print(
                    f"[{完了数}/{総数}] {結果['ファイル名']}  失敗: {結果['エラー']}",
                    file=sys.stderr,
                )
                失敗数 += 1
                失敗ファイル一覧.append(結果["ファイル名"])
                ファイル別結果.append({
                    "ファイル名": 結果["ファイル名"],
                    "エラー": 結果["エラー"],
                    "状態": "失敗",
                })

    print("-" * 50)
    print(f"完了: 成功 {成功数}件 / 失敗 {失敗数}件")
    全体削減率 = 0.0
    if 合計元サイズ > 0:
        全体削減率 = (1 - 合計圧縮後サイズ / 合計元サイズ) * 100
        print(
            f"合計サイズ: {合計元サイズ/1024/1024:.2f}MB → "
            f"{合計圧縮後サイズ/1024/1024:.2f}MB  "
            f"({全体削減率:+.1f}%)"
        )
    print(f"出力先: {出力先フォルダ}")

    # zip化
    zip成功 = False
    zipパス = None
    zipファイル名 = None
    if zip作成フラグ and 成功ファイル一覧:
        print("-" * 50)
        print("zipファイルを作成中...")
        zipファイル名 = f"{局名}_写真_{タイムスタンプ}.zip"
        zipパス = os.path.join(出力先フォルダ, zipファイル名)

        try:
            with zipfile.ZipFile(zipパス, "w", zipfile.ZIP_DEFLATED) as zf:
                for i, ファイル名 in enumerate(成功ファイル一覧, 1):
                    ファイルパス = os.path.join(出力先フォルダ, ファイル名)
                    zf.write(ファイルパス, arcname=ファイル名)
                    print(f"  [{i}/{len(成功ファイル一覧)}] {ファイル名} を追加")

            zipサイズ = os.path.getsize(zipパス)
            print(f"zip作成完了: {zipファイル名} ({zipサイズ/1024/1024:.2f}MB)")
            print(f"zipパス: {zipパス}")
            zip成功 = True
        except Exception as e:
            print(f"zip作成失敗: {e}", file=sys.stderr)
    elif not zip作成フラグ:
        print("zip作成はスキップしました。")

    # zip整合性チェック + ファイル数照合 → 問題なければ圧縮画像を削除
    zip整合性結果 = "チェックなし"
    画像削除結果 = "実行なし"
    if 画像削除フラグ and zip成功 and zipパス:
        print("-" * 50)
        print("zipの整合性とファイル数を確認中...")
        try:
            with zipfile.ZipFile(zipパス, "r") as zf:
                不正ファイル = zf.testzip()
                if 不正ファイル is not None:
                    zip整合性結果 = f"CRC失敗: {不正ファイル}"
                    print(
                        f"整合性チェック失敗: {不正ファイル} が破損している可能性があります。",
                        file=sys.stderr,
                    )
                    print("安全のため、圧縮画像の削除をスキップしました。", file=sys.stderr)
                    画像削除結果 = "スキップ（CRC失敗）"
                else:
                    zip内ファイル一覧 = set(zf.namelist())
                    期待ファイル一覧 = set(成功ファイル一覧)
                    不足ファイル = 期待ファイル一覧 - zip内ファイル一覧
                    余分ファイル = zip内ファイル一覧 - 期待ファイル一覧

                    print(f"  圧縮対象: {len(期待ファイル一覧)}件 / zip内: {len(zip内ファイル一覧)}件")

                    if 不足ファイル:
                        zip整合性結果 = f"ファイル数不足: {len(不足ファイル)}件欠落"
                        print(
                            f"ファイル数照合失敗: zip内に {len(不足ファイル)}件 不足しています。",
                            file=sys.stderr,
                        )
                        for f in sorted(不足ファイル):
                            print(f"  不足: {f}", file=sys.stderr)
                        print("安全のため、圧縮画像の削除をスキップしました。", file=sys.stderr)
                        画像削除結果 = "スキップ（ファイル数不足）"
                    else:
                        zip整合性結果 = "OK"
                        if 余分ファイル:
                            print(f"警告: zip内に想定外のファイルが {len(余分ファイル)}件 含まれています。")
                            for f in sorted(余分ファイル):
                                print(f"  想定外: {f}")

                        print("整合性・ファイル数チェックOK。圧縮画像を削除します...")
                        削除成功数 = 0
                        削除失敗数 = 0
                        for ファイル名 in 成功ファイル一覧:
                            ファイルパス = os.path.join(出力先フォルダ, ファイル名)
                            try:
                                os.remove(ファイルパス)
                                print(f"  削除: {ファイル名}")
                                削除成功数 += 1
                            except Exception as e:
                                print(f"  削除失敗: {ファイル名} - {e}", file=sys.stderr)
                                削除失敗数 += 1
                        print(f"削除完了: {削除成功数}件 / 失敗: {削除失敗数}件")
                        画像削除結果 = f"完了（削除 {削除成功数}件 / 失敗 {削除失敗数}件）"

        except Exception as e:
            zip整合性結果 = f"チェック中にエラー: {e}"
            print(f"整合性チェック中にエラーが発生しました: {e}", file=sys.stderr)
            print("安全のため、圧縮画像の削除をスキップしました。", file=sys.stderr)
            画像削除結果 = "スキップ（チェックエラー）"
    elif 画像削除フラグ and not zip成功:
        print("zip作成が成功していないため、圧縮画像の削除はスキップしました。")
        画像削除結果 = "スキップ（zip未作成）"

    # サマリーログ出力
    print("-" * 50)
    print("サマリーログを出力中...")
    ログファイル名 = f"{局名}_ログ_{タイムスタンプ}.txt"
    ログパス = os.path.join(出力先フォルダ, ログファイル名)
    try:
        with open(ログパス, "w", encoding="utf-8") as log:
            log.write(f"{'=' * 50}\n")
            log.write(f"  処理サマリー\n")
            log.write(f"{'=' * 50}\n")
            log.write(f"局名          : {局名}\n")
            log.write(f"実行日時      : {datetime.now().strftime('%Y年%m月%d日 %H:%M:%S')}\n")
            log.write(f"圧縮度        : {圧縮度}\n")
            log.write(f"JPEG品質      : {jpeg_quality}\n")
            log.write(f"長辺最大      : {max_size if max_size else '元サイズ維持'}\n")
            log.write(f"入力フォルダ  : {写真フォルダ}\n")
            log.write(f"出力フォルダ  : {出力先フォルダ}\n")
            log.write(f"\n{'─' * 50}\n")
            log.write(f"  処理結果\n")
            log.write(f"{'─' * 50}\n")
            log.write(f"対象ファイル数: {総数}件\n")
            log.write(f"成功          : {成功数}件\n")
            log.write(f"失敗          : {失敗数}件\n")
            if 合計元サイズ > 0:
                log.write(f"合計サイズ    : {合計元サイズ/1024/1024:.2f}MB → {合計圧縮後サイズ/1024/1024:.2f}MB ({全体削減率:+.1f}%)\n")
            log.write(f"\n{'─' * 50}\n")
            log.write(f"  zip / 削除\n")
            log.write(f"{'─' * 50}\n")
            log.write(f"zip作成       : {'あり' if zip作成フラグ else 'なし'}\n")
            if zipファイル名:
                log.write(f"zipファイル名 : {zipファイル名}\n")
                log.write(f"zipパス       : {zipパス}\n")
            log.write(f"zip整合性     : {zip整合性結果}\n")
            log.write(f"画像削除      : {画像削除結果}\n")
            log.write(f"\n{'─' * 50}\n")
            log.write(f"  ファイル別結果\n")
            log.write(f"{'─' * 50}\n")
            for r in sorted(ファイル別結果, key=lambda x: x["ファイル名"]):
                if r["状態"] == "成功":
                    log.write(
                        f"  [成功] {r['ファイル名']}  "
                        f"{r['元サイズ']/1024:.1f}KB → {r['圧縮後サイズ']/1024:.1f}KB  "
                        f"({r['削減率']:+.1f}%)\n"
                    )
                else:
                    log.write(f"  [失敗] {r['ファイル名']}  エラー: {r['エラー']}\n")
            log.write(f"{'=' * 50}\n")

        print(f"ログ出力完了: {ログファイル名}")
        print(f"ログパス: {ログパス}")
    except Exception as e:
        print(f"ログ出力失敗: {e}", file=sys.stderr)


if __name__ == "__main__":
    main()