第 12 章　给工具一个窗口

from pathlib import Path

# ============ 逻辑区 · 换壳时一行都不要动 ============
def preview_rename(folder, prefix):
    # 预演：算出每个文件将要改成什么名字，返回一份清单，不改动任何文件
    plan = []
    number = 0
    for old in sorted(p for p in Path(folder).iterdir() if p.is_file()):
        if old.suffix.lower() not in ['.jpg', '.png']:
            continue
        number = number + 1
        new = old.with_name(prefix + '_' + str(number).zfill(3) + old.suffix.lower())
        if new.exists() and new != old:
            new = old.with_name(new.stem + '-2' + new.suffix)
        plan.append((old, new))
    return plan

def do_rename(plan, on_progress=None):
    # 执行：按清单真正改名，返回成功条数；每改完一个调用一次 on_progress
    done = 0
    for old, new in plan:
        old.rename(new)
        done = done + 1
        if on_progress is not None:
            on_progress(done, len(plan))
    return done

# ============ 命令行外壳 · 窗口版会把这一段整个换掉 ============
def print_progress(done, total):
    print('  已处理', done, '/', total)

if __name__ == '__main__':   # 外壳的起点：只在直接运行本文件时执行
    plan = preview_rename(folder, PREFIX)
    # ……打印清单，input 等使用者输入 y
    done = do_rename(plan, print_progress)

# ① 点「选择文件夹」时执行：弹出系统对话框，选完立即预演
def on_choose_folder():
    picked = filedialog.askdirectory(title='选择要处理的文件夹')
    if not picked:                      # 使用者点了取消
        return
    current_plan = preview_rename(Path(picked), prefix_entry.get().strip())
    # ……把清单逐行放进文本框

# ② 点「确认执行」时执行：先二次确认，再调用逻辑区的 do_rename
def on_run():
    agreed = messagebox.askyesno('确认执行',
        '将要改动 ' + str(len(current_plan)) + ' 个文件的名称。\n\n这一步没有撤销，确定继续吗？')
    if not agreed:
        return
    try:
        done = do_rename(current_plan, update_progress)
    except Exception as error:
        messagebox.showerror('执行中断',
            '出错的文件已停下，后面的文件保持原样。\n\n' + str(error))
        return
    messagebox.showinfo('完成', '共改名 ' + str(done) + ' 个文件。')

# ③ 登记：这个按钮被点的时候，去执行 on_run 这一段
tk.Button(top, text='确认执行', command=on_run).pack(side='left')

# ④ 搭完界面之后停在这里等，下面不会再有代码被执行
root.mainloop()