Python如何实现图片批量重命名功能

import os
import shutil
import tkinter as tk
from tkinter import ttk, messagebox, filedialog
from pathlib import Path


class ImageRenamerApp:
    def __init__(self, root):
        self.root = root
        self.root.title("图片批量重命名工具")
        self.setup_ui()
        self.center_window()

    def center_window(self):
        """居中显示窗口"""
        self.root.update_idletasks()
        width = 550  # 增大窗口宽度
        height = 600  # 增大窗口高度
        x = (self.root.winfo_screenwidth() // 2) - (width // 2)
        y = (self.root.winfo_screenheight() // 2) - (height // 2)
        self.root.geometry(f'{width}x{height}+{x}+{y}')

    def setup_ui(self):
        """设置UI界面"""
        # 设置字体
        font_style = ("Microsoft YaHei", 11)

        # 创建主框架，增加内边距
        main_frame = tk.Frame(self.root, padx=20, pady=20)
        main_frame.pack(expand=True, fill="both")

        # 文件名前缀
        tk.Label(main_frame, text="文件名前缀:", font=font_style).grid(row=0, column=0, sticky="w", pady=(0, 10))
        self.prefix_entry = tk.Entry(main_frame, font=font_style)
        self.prefix_entry.grid(row=0, column=1, sticky="ew", pady=(0, 10), padx=(10, 0))

        # 输入文件夹选择
        tk.Label(main_frame, text="输入文件夹:", font=font_style).grid(row=1, column=0, sticky="w", pady=(10, 0))
        self.input_dir_var = tk.StringVar(value=os.getcwd())
        tk.Entry(main_frame, textvariable=self.input_dir_var, font=font_style, state='readonly').grid(row=1, column=1,
                                                                                                      sticky="ew",
                                                                                                      pady=(10, 0),
                                                                                                      padx=(10, 0))
        tk.Button(main_frame, text="浏览...", command=self.select_input_dir, font=font_style).grid(row=1, column=2,
                                                                                                   pady=(10, 0),
                                                                                                   padx=(10, 0))

        # 输出文件夹选择
        tk.Label(main_frame, text="输出文件夹:", font=font_style).grid(row=2, column=0, sticky="w", pady=(10, 0))
        default_output = os.path.join(os.getcwd(), "output")
        self.output_dir_var = tk.StringVar(value=default_output)
        tk.Entry(main_frame, textvariable=self.output_dir_var, font=font_style).grid(row=2, column=1, sticky="ew",
                                                                                     pady=(10, 0), padx=(10, 0))
        tk.Button(main_frame, text="浏览...", command=self.select_output_dir, font=font_style).grid(row=2, column=2,
                                                                                                    pady=(10, 0),
                                                                                                    padx=(10, 0))

        # 分隔符选择
        tk.Label(main_frame, text="选择分隔符:", font=font_style).grid(row=3, column=0, sticky="w", pady=(10, 0))

        separators_frame = tk.Frame(main_frame)
        separators_frame.grid(row=3, column=1, sticky="w", padx=(10, 0), pady=(10, 0))

        self.separator_var = tk.StringVar(value="_")
        separators = [("下划线(_)", "_"), ("空格( )", " "), ("短横线(-)", "-"), ("点(.)", ".")]
        for i, (text, value) in enumerate(separators):
            rb = tk.Radiobutton(separators_frame, text=text, variable=self.separator_var,
                                value=value, font=font_style)
            rb.grid(row=i, column=0, sticky="w")

        # 连接符号选择
        tk.Label(main_frame, text="选择连接符号:", font=font_style).grid(row=4, column=0, sticky="w", pady=(10, 0))

        connectors_frame = tk.Frame(main_frame)
        connectors_frame.grid(row=4, column=1, sticky="w", padx=(10, 0), pady=(10, 0))

        self.connector_var = tk.StringVar(value=":")
        connectors = [("冒号(:)", ":"), ("空格( )", " "), ("短横线(-)", "-"), ("等号(=)", "=")]
        for i, (text, value) in enumerate(connectors):
            rb = tk.Radiobutton(connectors_frame, text=text, variable=self.connector_var,
                                value=value, font=font_style)
            rb.grid(row=i, column=0, sticky="w")

        # 执行按钮 - 放在底部并居中
        button_frame = tk.Frame(main_frame)
        button_frame.grid(row=5, column=0, columnspan=3, pady=(20, 0), sticky="ew")

        tk.Button(button_frame, text="开始重命名", command=self.rename_images,
                  font=font_style, bg="#4CAF50", fg="white", width=15).pack(pady=10)

        # 配置网格权重，使内容居中
        main_frame.columnconfigure(1, weight=1)

    def select_input_dir(self):
        """选择输入文件夹"""
        dir_path = filedialog.askdirectory(title="选择输入文件夹", initialdir=self.input_dir_var.get())
        if dir_path:
            self.input_dir_var.set(dir_path)

    def select_output_dir(self):
        """选择输出文件夹"""
        dir_path = filedialog.askdirectory(title="选择输出文件夹", initialdir=self.output_dir_var.get())
        if dir_path:
            self.output_dir_var.set(dir_path)

    def rename_images(self):
        """执行重命名操作"""
        prefix = self.prefix_entry.get().strip()
        if not prefix:
            messagebox.showerror("错误", "请输入文件名前缀!")
            return

        input_dir = self.input_dir_var.get()
        output_dir = self.output_dir_var.get()
        selected_separator = self.separator_var.get()
        selected_connector = self.connector_var.get()

        # 检查输入文件夹是否存在
        if not os.path.isdir(input_dir):
            messagebox.showerror("错误", f"输入文件夹不存在:\n{input_dir}")
            return

        # 获取输入目录下所有图片文件
        image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp']
        image_files = [f for f in os.listdir(input_dir)
                       if os.path.isfile(os.path.join(input_dir, f)) and
                       os.path.splitext(f)[1].lower() in image_extensions]

        if not image_files:
            messagebox.showinfo("提示", f"输入文件夹中没有找到图片文件:\n{input_dir}")
            return

        # 创建输出文件夹
        try:
            os.makedirs(output_dir, exist_ok=True)
        except Exception as e:
            messagebox.showerror("错误", f"无法创建输出文件夹:\n{output_dir}\n错误: {str(e)}")
            return

        # 创建记录文件
        record_file = os.path.join(output_dir, 'rename_records.txt')

        try:
            with open(record_file, 'w', encoding='utf-8') as f:
                for index, old_name in enumerate(sorted(image_files), start=1):
                    # 获取文件名和扩展名
                    old_base, ext = os.path.splitext(old_name)
                    # 构建新文件名
                    new_name = f"{prefix}{selected_separator}{index}{ext}"

                    # 复制文件到输出文件夹并重命名
                    old_path = os.path.join(input_dir, old_name)
                    new_path = os.path.join(output_dir, new_name)
                    shutil.copy2(old_path, new_path)

                    # 写入记录（只保留原始文件名，不含扩展名）
                    f.write(f"{new_name}{selected_connector}{old_base}\n")

            messagebox.showinfo("完成", f"已成功重命名 {len(image_files)} 个图片文件。\n输出目录: {output_dir}")
        except Exception as e:
            messagebox.showerror("错误", f"处理过程中发生错误:\n{str(e)}")


if __name__ == "__main__":
    root = tk.Tk()
    app = ImageRenamerApp(root)
    root.mainloop()