如何在自定义QMessageBox的alert函数中获取点击按钮的值

刚好碰到过类似的需求，我来帮你搞定这个问题——要在alert()函数内部直接获取用户点击的按钮文本，不用把值传到外部函数，有两种实用的方案可以选：

方案一：利用exec_()执行后获取点击按钮 #

QMessageBox.exec_()运行结束后，你可以调用clickedButton()方法直接拿到被点击的按钮对象，再通过text()提取按钮上的文本，这样就能在函数内部直接处理了：

def alert(**kwargs):
    msg = QMessageBox()
    clicked_button_text = None  # 用来存储点击的按钮文本
    try:
        for key, value in kwargs.items():
            key = key.lower()
            if key == "title":
                msg.setWindowTitle(value)
            elif key == "text":
                msg.setText(value)
            elif key == "buttons":
                buttons = value.split(',')
                # 注意给按钮文本做strip处理，避免逗号前后的空格影响
                for btn_text in buttons:
                    msg.addButton(QPushButton(btn_text.strip()), QMessageBox.ActionRole)
        
        # 弹出对话框，用户操作完成后继续执行
        msg.exec_()
        # 获取被点击的按钮对象
        clicked_btn = msg.clickedButton()
        if clicked_btn:
            clicked_button_text = clicked_btn.text()
            # 这里就可以直接处理按钮值了，比如做逻辑判断
            if clicked_button_text == "Yes":
                print("用户确认了操作")
            elif clicked_button_text == "No":
                print("用户取消了操作")
            # 也可以直接打印查看
            print(f"用户点击了：{clicked_button_text}")
            
    except Exception as error:
        print(error)
    # 按需返回点击的按钮文本给调用者
    return clicked_button_text

方案二：绑定内部槽函数捕获点击信号 #

你也可以在alert()函数内部定义一个槽函数，把buttonClicked信号绑定到这个内部函数上，这样点击按钮时就能直接在内部拿到按钮值：

def alert(**kwargs):
    msg = QMessageBox()
    clicked_button_text = None

    # 内部定义的槽函数，用来处理按钮点击事件
    def on_button_clicked(btn):
        nonlocal clicked_button_text
        clicked_button_text = btn.text()
        # 在这里添加你的业务逻辑
        print(f"内部捕获到点击：{clicked_button_text}")
        if clicked_button_text == "Restore":
            print("执行恢复操作")

    try:
        for key, value in kwargs.items():
            key = key.lower()
            if key == "title":
                msg.setWindowTitle(value)
            elif key == "text":
                msg.setText(value)
            elif key == "buttons":
                buttons = value.split(',')
                for btn_text in buttons:
                    msg.addButton(QPushButton(btn_text.strip()), QMessageBox.ActionRole)
        
        # 绑定按钮点击信号到内部槽函数
        msg.buttonClicked.connect(on_button_clicked)
        msg.exec_()
        
    except Exception as error:
        print(error)
    return clicked_button_text

两种方案的调用方式和你原来的一样：

alert(title="Some Title", text="Some Text", buttons="Yes,No,Restore,Config")

不管选哪种，都能在alert()函数内部直接访问到用户点击的按钮值，不用依赖外部函数来接收这个值啦。

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