如何在tkinter.messagebox中添加自定义按钮

在Tkinter中，可以使用messagebox模块创建对话框，其中包含了常用的几种类型，如信息提示、错误提示、警告提示等。然而，messagebox默认情况下只有确定按钮，无法添加自定义按钮。

如果想要添加自定义按钮，可以通过创建一个自定义的对话框类来实现。下面是一个使用Toplevel来创建自定义对话框的例子，以添加“确定”和“取消”按钮为例。

首先，我们需要导入tkinter和messagebox模块：

import tkinter as tk
from tkinter import messagebox

然后，我们创建一个继承自tk.Toplevel的自定义对话框类CustomDialog：

class CustomDialog(tk.Toplevel):
    def __init__(self, parent, title=None, message=None):
        super().__init__(parent)
        self.title(title)
        self.geometry("300x100")
        
        # 创建标签和消息
        label = tk.Label(self, text=message)
        label.pack(padx=20, pady=20)
        
        # 创建确认和取消按钮
        confirm_button = tk.Button(self, text="确定", command=self.confirm)
        confirm_button.pack(side=tk.LEFT, padx=20, pady=10)
        cancel_button = tk.Button(self, text="取消", command=self.cancel)
        cancel_button.pack(side=tk.RIGHT, padx=20, pady=10)
        
        # 将对话框设置为模态（即只能在对话框关闭后才能操作其他窗口）
        self.grab_set()
        self.wait_window(self)

在CustomDialog类的__init__方法中，我们首先调用父类的__init__方法来创建一个Toplevel窗口。然后，我们设置对话框的标题和尺寸。

接下来，我们创建一个标签控件来显示消息内容。通过调用pack方法将其放置在对话框中，并设置一定的内边距。

然后，我们创建了“确定”和“取消”按钮。这两个按钮都绑定了相应的命令，self.confirm和self.cancel。当用户点击“确定”按钮时，self.confirm方法将被调用，对话框将关闭并返回一个确定的结果给调用者；同样，点击“取消”按钮时，对话框将关闭并返回一个取消的结果给调用者。

最后，我们通过调用self.grab_set()方法将对话框设置为模态（即只能在对话框关闭后才能操作其他窗口），再调用self.wait_window(self)方法使主窗口等待对话框关闭后再进行后续操作。

接下来，我们创建一个测试用的主窗口，并添加一个按钮来打开自定义对话框：

class App(tk.Tk):
    def __init__(self):
        super().__init__()
        
        self.title("自定义对话框示例")
        self.geometry("400x300")
        
        self.button = tk.Button(self, text="打开对话框", command=self.open_dialog)
        self.button.pack(pady=20)
        
    def open_dialog(self):
        dialog = CustomDialog(self, title="自定义对话框", message="这是一个自定义对话框")
        result = dialog.result
        if result == "confirm":
            messagebox.showinfo("结果", "确定按钮被点击")
        elif result == "cancel":
            messagebox.showinfo("结果", "取消按钮被点击")

在App类的__init__方法中，我们创建了一个主窗口。在主窗口中，我们添加了一个按钮，点击按钮时调用open_dialog方法打开自定义对话框。

open_dialog方法中，我们创建了一个CustomDialog实例，并传递了相应的参数。根据对话框的返回结果，我们使用messagebox.showinfo方法来显示相应的提示信息。

最后，我们创建一个App实例，并调用Tk类的mainloop方法来运行主窗口。

完整的代码如下：

import tkinter as tk
from tkinter import messagebox

class CustomDialog(tk.Toplevel):
    def __init__(self, parent, title=None, message=None):
        super().__init__(parent)
        self.title(title)
        self.geometry("300x100")
        
        # 创建标签和消息
        label = tk.Label(self, text=message)
        label.pack(padx=20, pady=20)
        
        # 创建确认和取消按钮
        confirm_button = tk.Button(self, text="确定", command=self.confirm)
        confirm_button.pack(side=tk.LEFT, padx=20, pady=10)
        cancel_button = tk.Button(self, text="取消", command=self.cancel)
        cancel_button.pack(side=tk.RIGHT, padx=20, pady=10)
        
        # 将对话框设置为模态（即只能在对话框关闭后才能操作其他窗口）
        self.grab_set()
        self.wait_window(self)
        
    def confirm(self):
        self.result = "confirm"
        self.destroy()
        
    def cancel(self):
        self.result = "cancel"
        self.destroy()
        

class App(tk.Tk):
    def __init__(self):
        super().__init__()
        
        self.title("自定义对话框示例")
        self.geometry("400x300")
        
        self.button = tk.Button(self, text="打开对话框", command=self.open_dialog)
        self.button.pack(pady=20)
        
    def open_dialog(self):
        dialog = CustomDialog(self, title="自定义对话框", message="这是一个自定义对话框")
        result = dialog.result
        if result == "confirm":
            messagebox.showinfo("结果", "确定按钮被点击")
        elif result == "cancel":
            messagebox.showinfo("结果", "取消按钮被点击")

if __name__ == "__main__":
    app = App()
    app.mainloop()

运行以上代码，可以看到一个主窗口和一个按钮。点击按钮时，会弹出一个自定义对话框，其中包含一个消息和“确定”和“取消”按钮。点击对应的按钮时，对话框会关闭，并弹出一个提示框显示相应的结果。

这就是如何在Tkinter的对话框中添加自定义按钮的方法，并通过一个完整的示例演示了如何实现。根据这个例子的思路，你可以根据具体的需求进行修改和扩展，实现更复杂的自定义对话框。