【5分钟实战】一键搞定照片管理难题！CodeBuddy智能批量重命名让你的图片库井井有条

import os
import re
from datetime import datetime
from PIL import Image
from PIL.ExifTags import TAGS


def get_image_date(image_path):
    """尝试从图片的EXIF信息中获取拍摄日期"""
    try:
        img = Image.open(image_path)
        exif = img._getexif()
        if exif:
            for tag, value in exif.items():
                decoded = TAGS.get(tag, tag)
                if decoded == 'DateTimeOriginal':
                    return value.split()[0].replace(':', '-')
    except (AttributeError, KeyError, IndexError, IOError):
        pass
    return datetime.now().strftime('%Y-%m-%d')  # 如果没有日期信息，使用当前日期


def clean_description(text):
    """清理描述文本"""
    # 移除特殊字符，只保留字母、数字、中文和空格
    cleaned = re.sub(r'[^\w\u4e00-\u9fff\s]', '', text)
    # 移除多余空格
    cleaned = ' '.join(cleaned.split())
    return cleaned.strip()


def batch_rename_images(folder_path, description_source='filename'):
    """批量重命名图片文件"""
    # 获取所有图片文件
    image_exts = ('.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp')
    files = [f for f in os.listdir(folder_path) if f.lower().endswith(image_exts)]
    files.sort()  # 按文件名排序

    print(f"找到 {len(files)} 个图片文件")

    for i, filename in enumerate(files, start=1):
        old_path = os.path.join(folder_path, filename)

        # 获取日期
        date = get_image_date(old_path)

        # 获取描述
        if description_source == 'filename':
            # 从原文件名提取描述（去掉扩展名）
            basename = os.path.splitext(filename)[0]
            description = clean_description(basename)
        else:
            # 手动输入描述
            description = input(f"请输入文件 {filename} 的描述: ")
            description = clean_description(description)

        # 构建新文件名
        new_name = f"{i:03d}_{date}_{description}{os.path.splitext(filename)[1]}"
        new_path = os.path.join(folder_path, new_name)

        # 处理文件名冲突
        counter = 1
        while os.path.exists(new_path):
            new_name = f"{i:03d}_{date}_{description}_{counter}{os.path.splitext(filename)[1]}"
            new_path = os.path.join(folder_path, new_name)
            counter += 1

        # 重命名文件
        os.rename(old_path, new_path)
        print(f"重命名: {filename} -> {new_name}")


if __name__ == "__main__":
    folder = input("请输入图片文件夹路径: ")
    source = input("描述来源 (1: 原文件名, 2: 手动输入): ")
    description_source = 'filename' if source == '1' else 'input'

    batch_rename_images(folder, description_source)
    print("批量重命名完成!")