Python GUI应用中长时任务的优雅中断机制

import tkinter as tk
import threading
import time

def static_counter():
    # 模拟耗时操作，此处为简单的循环和延迟
    for i in range(10):
        time.sleep(0.2)
    return 10

class MyGUI():
    # ... (初始化代码省略) ...

    def check_stop(self):
        # 检查停止标志并更新GUI状态
        if self.asked_stop:
            self.label_status_var.set("stopped")
            self.root.update()
            self.running = False
            self.asked_stop = False
            return True
        else:
            return False

    def process(self):
        if self.running:
            return
        self.label_status_var.set("0")
        self.running = True

        counter = 0
        while True:
            # 问题所在：如果static_counter内部也耗时，这里检查不够及时
            if self.check_stop(): # 每次循环都需要检查
                return
            counter += static_counter() # 如果static_counter内部没有检查，会一直运行到结束
            self.label_status_var.set(str(counter))
            self.root.update()

def static_counter(check_stop_func):
    """
    一个模拟耗时操作的函数，接受一个中断检查回调函数。

    Args:
        check_stop_func: 一个无参数函数，返回True表示应停止，False表示继续。

    Returns:
        tuple: (计算结果, 是否已停止)。如果停止，计算结果为0。
    """
    for i in range(10):
        if check_stop_func(): # 在每次内部循环迭代时检查是否需要停止
            return 0, True  # 提前返回，并指示已停止
        time.sleep(0.2)
    return 10, False # 正常完成，指示未停止

class MyGUI():
    def __init__(self):
        self.root = tk.Tk()
        self.root.title("Counter")
        self.root.geometry('300x50+200+200')
        self.running = False
        self.asked_stop = False

        # 按钮和标签的初始化 (与原代码相同)
        self.button_start = tk.Button(text="Start", command=lambda: threading.Thread(target=self.process).start())
        self.button_start.grid(row=0, column=0, sticky='NWSE', padx=5, pady=5)
        self.button_stop = tk.Button(text="Stop", command=self.stop)
        self.button_stop.grid(row=0, column=1, sticky='NWSE', padx=5, pady=5)
        self.label_status_var = tk.StringVar()
        self.label_status_var.set("0")
        self.label_status = tk.Label(textvariable=self.label_status_var)
        self.label_status.grid(row=0, column=2, sticky='NWSE', padx=5, pady=5)

        # 配置 (与原代码相同)
        for i in range(3):
            self.root.grid_columnconfigure(i, weight=1)
        self.root.grid_rowconfigure(0, weight=1)

        # 主循环
        self.root.mainloop()

    def stop(self):
        """设置停止标志"""
        self.asked_stop = True

    def check_stop(self):
        """
        检查停止标志的回调函数。
        如果需要停止，则更新GUI状态并重置标志。
        """
        if self.asked_stop:
            self.label_status_var.set("stopped")
            # 注意：在非主线程中直接调用root.update()可能导致问题，
            # 更好的做法是通过queue或tk.after_idle将GUI更新调度到主线程。
            # 为简化示例，此处保留。
            self.root.update_idletasks() # 使用update_idletasks更安全
            self.running = False
            self.asked_stop = False
            return True
        else:
            return False

    def process(self):
        """
        在独立线程中执行的耗时任务。
        """
        if self.running:
            return
        self.label_status_var.set("0")
        self.running = True

        counter = 0
        while True:
            # 调用static_counter，并传入self.check_stop作为回调函数
            count, stop_flag = static_counter(self.check_stop)
            if stop_flag: # 根据static_counter的返回值判断是否停止
                return
            counter += count
            self.label_status_var.set(str(counter))
            self.root.update_idletasks() # 更新GUI

if __name__ == '__main__':
    new = MyGUI()