Python tkinter开发记事本 打开文件时保存提示弹窗逻辑异常

问题根因 #

两个bug的核心是状态标记逻辑存在漏洞：

1. save_flag 仅会在保存操作后置为True，用户后续编辑内容时没有逻辑把它改回False，导致只要执行过一次保存，后续不管怎么修改内容，程序都会判定为已保存状态，不会弹出保存提示。
2. 没有留存文件打开/上次保存时的原始内容做比对，仅靠文本长度判断空内容，既无法精准识别无修改状态，也覆盖不了新建空白文档的场景。

修复方案 #

第一步：调整全局变量 #

新增变量存储文本快照，用于和当前内容做比对：

#variable
 url = ''
 save_flag = False
 original_content = ''  # 存储上次打开/保存时的原始内容

第二步：绑定文本修改事件 #

在初始化text_editor的代码后追加事件绑定，只要用户改动文本内容，自动标记为未保存状态：

# 文本框内容变动时自动标记未保存
def on_content_modified(event=None):
    global save_flag
    if text_editor.edit_modified():
        save_flag = False
        text_editor.edit_modified(False)  # 重置触发标记，方便下次变动识别

text_editor.bind('<<Modified>>', on_content_modified)

第三步：修复所有状态重置逻辑 #

所有完成文件加载、保存的节点，都要同步更新save_flag和original_content，保证状态和实际内容一致：

1. 修复open_file_function，加载文件成功后重置状态：

def open_file_function():
    global url, save_flag, original_content

    url = filedialog.askopenfilename(initialdir="\Desktop", title='Select File',
                                     filetypes=(
                                     ('Text File', '*.txt'), ("Python Files", "*.py"), ("HTML Files", "*.html"),
                                     ('All files', '*.*')))
    try:
        with open(url, 'r') as fr:
            content = fr.read()
            text_editor.delete(1.0, tk.END)
            text_editor.insert(1.0, content)
            # 加载成功后重置状态
            original_content = content
            save_flag = True
    except FileNotFoundError:
        return
    except:
        return

    name = pathlib.Path(url).name
    main_application.title(f'{name} - NotePad')

2. 修复save_file，保存成功后重置状态：

def save_file(e=None):
    global url, save_flag, original_content

    if url:
        current_content = text_editor.get(1.0, END)
        with open(url, 'w') as text_file:
            text_file.write(current_content)
        # 保存成功后重置状态
        original_content = current_content
        save_flag = True
        name = pathlib.Path(url).name
        main_application.title(f'{name} - NotePad')
    else:
        save_as_file()

3. 修复save_as_file，补上原代码遗漏的状态设置：

def save_as_file(e=None):
    global url, save_flag, original_content

    url = filedialog.asksaveasfilename(initialdir="\Desktop", title="Save As", defaultextension=".txt", filetypes=(
        ("Text Files", "*.txt"), ("Python Files", "*.py"), ("HTML Files", "*.html")))
    if url:
        current_content = text_editor.get(1.0, END)
        with open(url, 'w') as take_file:
            take_file.write(current_content)
        # 另存成功后重置状态
        original_content = current_content
        save_flag = True
        name = pathlib.Path(url).name
        main_application.title(f'{name} - NotePad')
        messagebox.showinfo("Successfully Saved", "Your file saved successfully")

第四步：简化打开、退出的判断逻辑 #

去掉原代码靠文本长度判断的逻辑，直接通过save_flag判定是否有未保存改动：

1. 修复后的open_file：

def open_file(e=None):
    # 已保存状态直接打开新文件
    if save_flag:
        open_file_function()
    else:
        choice = askyesnocancel('Notepad', 'Do you want to save your changes to ' + main_application.title()
                                + '?')
        if choice:
            save_file()
            # 保存成功才打开新文件，用户取消保存则终止操作
            if save_flag:
                open_file_function()
        elif choice is None:
            # 点取消直接返回
            return
        else:
            # 选择不保存直接打开
            open_file_function()

2. 修复后的exit_application：

def exit_application():
    if save_flag:
        main_application.destroy()
    else:
        choice = askyesnocancel('Notepad', 'Do you want to save your changes to ' + main_application.title()
                                + '?')
        if choice:
            save_file()
            if save_flag:
                main_application.destroy()
        elif choice is None:
            return
        else:
            main_application.destroy()

补充：你写new_file新建功能时，记得同步重置状态：清空文本框后将url设为空字符串、original_content设为空字符串、save_flag设为True，窗口标题改回untitled - NotePad即可。

内容的提问来源于stack exchange，提问作者coldfire1500