批量重命名

# -*- coding: utf-8 -*-

import os
import shutil
from tkinter import filedialog, Button, Label, Entry, StringVar, Tk, IntVar, BooleanVar, Checkbutton, messagebox, font

def get_file_extension(filename):
    """获取文件的扩展名"""
    _, ext = os.path.splitext(filename)
    return ext

def get_default_suffix(files_dir):
    """根据文件夹中的文件类型获取默认后缀"""
    extensions = set()
    for filename in os.listdir(files_dir):
        if os.path.isfile(os.path.join(files_dir, filename)):
            _, ext = os.path.splitext(filename)
            extensions.add(ext.lower())
    return " ".join(extensions)

def batch_rename(files_dir, prefix, suffix, starting_number):
    i = starting_number  # 初始编号
    num_files = len(os.listdir(files_dir))
    num_digits = len(str(num_files))  # 计算文件数量的位数

    for filename in os.listdir(files_dir):
        file_path = os.path.join(files_dir, filename)
        if os.path.isfile(file_path):
            # 根据文件数量的位数确定编号的格式
            num_str = str(i).zfill(num_digits)
            new_filename = prefix + num_str + suffix
            destination = os.path.join(files_dir, new_filename)
            shutil.move(file_path, destination)
            i += 1

def backup(directory):
    # 创建bak文件夹
    backup_dir = os.path.join(directory, "bak")
    os.makedirs(backup_dir, exist_ok=True)

    # 复制所有文件
    for filename in os.listdir(directory):
        # 忽略备份目录本身
        if filename == "bak":
            continue
        source = os.path.join(directory, filename)
        # 只有文件会被复制，目录会被忽略
        if os.path.isfile(source):
            destination = os.path.join(backup_dir, filename)
            shutil.copy2(source, destination)  # copy2保留文件的元数据(包括修改时间等信息)

    # 返回备份目录
    return backup_dir

def main():
    root = Tk()
    root.title("批量重命名")

    directory = StringVar()
    prefix = StringVar()
    suffix = StringVar()
    starting_number = IntVar()
    is_backup = BooleanVar()

    Label(root, text="文件夹:", padx=5, pady=10).grid(row=0, column=0, sticky="E")
    Entry(root, textvariable=directory).grid(row=0, column=1, padx=5, pady=10, sticky="EW")

    def browse_directory():
        dir_name = filedialog.askdirectory()
        directory.set(dir_name)
        # 根据文件夹中的文件类型获取默认后缀
        default_suffix = get_default_suffix(dir_name)
        suffix.set(default_suffix)

    Button(root, text="浏览", command=browse_directory).grid(row=0, column=2, padx=5, pady=10)

    Label(root, text="前缀:", padx=5, pady=10).grid(row=1, column=0, sticky="E")
    Entry(root, textvariable=prefix).grid(row=1, column=1, padx=5, pady=10, sticky="EW")

    Label(root, text="后缀:", padx=5, pady=10).grid(row=2, column=0, sticky="E")
    Entry(root, textvariable=suffix).grid(row=2, column=1, padx=5, pady=10, sticky="EW")

    Label(root, text="起始编号:", padx=5, pady=10).grid(row=3, column=0, sticky="E")
    Entry(root, textvariable=starting_number).grid(row=3, column=1, padx=5, pady=10, sticky="EW")

    Checkbutton(root, text='备份', variable=is_backup).grid(row=4,column=1, padx=5, pady=10, sticky="W")

    def rename_files():
        if is_backup.get():
            files_dir = backup(directory.get())
        else:
            files_dir = directory.get()

        batch_rename(files_dir, prefix.get(), suffix.get(), starting_number.get())
        messagebox.showinfo("成功", "文件重命名成功!")
        root.destroy()

    # 改变重命名按钮的大小和样式
    rename_button = Button(root, text="重命名", command=rename_files)
    rename_button.grid(row=5, column=1, padx=5, pady=20, sticky="E")
    rename_button.config(font=font.Font(size=14, weight='bold'))

    root.grid_columnconfigure(1, weight=1)
    root.resizable(True, False)
    root.mainloop()

if __name__ == "__main__":
    main()