Gitで整理！画像を200KB圧縮＋コピーライト追加【第3回：ファイル名から日付情報取得】

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("✅ 全処理完了")

"""ファイル名から撮影日を抽出するためのユーティリティ。"""

import re
from datetime import datetime

def extract_photo_date_from_filename(filename: str) -> str:
    """
    ファイル名から撮影日を抽出する関数

    【目的】
    ファイル名に「YYYY年MM月DD日」の形式で日付が含まれている場合はそれを撮影日（photo_date）として返す。
    含まれていない場合は、実行日時（今日の日付）をデフォルトとして返す。

    【使用例】
    ファイル名例:
        - "2025年05月18日_風景.jpeg" → "2025年05月18日" を返す
        - "sunset.jpeg" → 実行日の "2025年06月05日"（例）を返す

    【パラメータ】
    filename (str): 処理対象のファイル名（拡張子含むことも可）

    【戻り値】
    str: 撮影日（"YYYY年MM月DD日" 形式）

    【注意点】
    - 正規表現で日本語日付にマッチする形式のみ対応
    - 先頭で見つかった日付部分1箇所だけを対象とする
    """

    # 正規表現で「YYYY年MM月DD日」の形式を検索
    match = re.search(r"\d{4}年\d{2}月\d{2}日", filename)

    if match:
        # 該当する日付文字列が見つかった場合は、それを返す
        return match.group()
    else:
        # 見つからなかった場合は、現在の日付を"YYYY年MM月DD日"形式で返す
        return datetime.now().strftime("%Y年%m月%d日")

if __name__ == "__main__":
    from pathlib import Path

    # テスト用のファイル名例
    sample_filenames = [
        "2025年05月18日_風景.jpeg",
        "holiday_2024年12月25日.png",
        "no_date_image.jpg",
    ]

    print("[INFO] extract_photo_date_from_filename() テスト")
    for fname in sample_filenames:
        date_str = extract_photo_date_from_filename(fname)
        print(f"{fname} → {date_str}")

    # 実フォルダ内の画像ファイルでも試す（相対パス指定）
    project_root = Path(__file__).resolve().parent.parent
    input_dir = project_root / "input_images"

    if input_dir.exists():
        print(f"\n[INFO] Checking files in: {input_dir}")
        for img_file in input_dir.iterdir():
            if img_file.suffix.lower() in [".jpg", ".jpeg", ".png"]:
                date_str = extract_photo_date_from_filename(img_file.name)
                print(f"{img_file.name} → {date_str}")
    else:
        print(f"[WARN] {input_dir} が見つかりませんでした。")

# ※ 実行結果中のパスは、環境依存情報（ユーザー名やPCのフォルダ構成）を含むため一部を「~」で省略しています。
# 2025年8月11日実行した場合の結果を表示しています。

[INFO] extract_photo_date_from_filename() テスト
2025年05月18日_風景.jpeg → 2025年05月18日
holiday_2024年12月25日.png → 2024年12月25日
no_date_image.jpg → 2025年08月11日

[INFO] Checking files in: ~/image-compressor-with-text/input_images
comp_input_sample_comp_2025年04月20日.jpg → 2025年04月20日