检查QMessageBox中的哪个按钮被单击

# 导入必要的模块
from PyQt5.QtWidgets import QMessageBox, QApplication

# 创建应用程序对象
app = QApplication([])

# 创建消息框对象
msg_box = QMessageBox()

# 设置消息框的标题和文本内容
msg_box.setWindowTitle("提示")
msg_box.setText("请选择一个按钮")

# 添加按钮到消息框
msg_box.addButton("按钮1", QMessageBox.AcceptRole)
msg_box.addButton("按钮2", QMessageBox.RejectRole)
msg_box.addButton("按钮3", QMessageBox.DestructiveRole)

# 显示消息框并获取返回值
button_clicked = msg_box.exec()

# 根据返回值确定哪个按钮被单击
if button_clicked == QMessageBox.AcceptRole:
    print("按钮1被单击")
elif button_clicked == QMessageBox.RejectRole:
    print("按钮2被单击")
elif button_clicked == QMessageBox.DestructiveRole:
    print("按钮3被单击")
else:
    print("没有按钮被单击")

# 运行应用程序
app.exec()