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):
        """設(shè)置UI界面"""
        # 設(shè)置字體
        font_style = ("Microsoft YaHei", 11)

        # 創(chuàng)建主框架，增加內(nèi)邊距
        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 = [("下劃線(_)", "_"), ("空格( )", " "), ("短橫線(-)", "-"), ("點(diǎn)(.)", ".")]
        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")

        # 執(zhí)行按鈕 - 放在底部并居中
        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)

        # 配置網(wǎng)格權(quán)重，使內(nèi)容居中
        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):
        """執(zhí)行重命名操作"""
        prefix = self.prefix_entry.get().strip()
        if not prefix:
            messagebox.showerror("錯(cuò)誤", "請輸入文件名前綴!")
            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("錯(cuò)誤", 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

        # 創(chuàng)建輸出文件夾
        try:
            os.makedirs(output_dir, exist_ok=True)
        except Exception as e:
            messagebox.showerror("錯(cuò)誤", f"無法創(chuàng)建輸出文件夾:\n{output_dir}\n錯(cuò)誤: {str(e)}")
            return

        # 創(chuàng)建記錄文件
        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):
                    # 獲取文件名和擴(kuò)展名
                    old_base, ext = os.path.splitext(old_name)
                    # 構(gòu)建新文件名
                    new_name = f"{prefix}{selected_separator}{index}{ext}"

                    # 復(fù)制文件到輸出文件夾并重命名
                    old_path = os.path.join(input_dir, old_name)
                    new_path = os.path.join(output_dir, new_name)
                    shutil.copy2(old_path, new_path)

                    # 寫入記錄（只保留原始文件名，不含擴(kuò)展名）
                    f.write(f"{new_name}{selected_connector}{old_base}\n")

            messagebox.showinfo("完成", f"已成功重命名 {len(image_files)} 個(gè)圖片文件。\n輸出目錄: {output_dir}")
        except Exception as e:
            messagebox.showerror("錯(cuò)誤", f"處理過程中發(fā)生錯(cuò)誤:\n{str(e)}")


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