【PyQt5】{14} —— 與對話框進行交互

# -*- coding: utf-8 -*-
"""
Created on Sat May  9 14:33:28 2020

@author: Giyn
"""

import sys
from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QVBoxLayout, QMessageBox


class Simple_Window(QWidget):
    def __init__(self):
        super(Simple_Window, self).__init__() # 使用super函數可以實現子類使用父類的方法
        self.setWindowTitle("QMessageBox")
        self.resize(400, 200)
        self.button1 = QPushButton("Question", self) # self是指定的父類Simple_Window，表示QLabel屬於Simple_Window窗口
        self.button2 = QPushButton("Information", self)
        
        # 連接信號和槽
        self.button1.clicked.connect(lambda: self.show_msg_box(self.button1)) # 槽函數帶有參數，使用lambda表達式封裝函數
        self.button2.clicked.connect(lambda: self.show_msg_box(self.button2))
        
        self.v_layout = QVBoxLayout() # 實例化一個QVBoxLayout對象
        self.v_layout.addWidget(self.button1)
        self.v_layout.addWidget(self.button2)
        
        self.setLayout(self.v_layout) # 調用窗口的setLayout方法將總佈局設置爲窗口的整體佈局
        
    def show_msg_box(self, button):
        if button == self.button1:
            choice = QMessageBox.question(self, "Question", "Do you want to save it?", QMessageBox.Yes | QMessageBox.No)
            if choice == QMessageBox.Yes:
                print("Saved!")
                self.close()
            elif choice == QMessageBox.No:
                self.close()
        
        elif button == self.button2:
            QMessageBox.information(self, "Information", "I am an information.", QMessageBox.Ok)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = Simple_Window()
    window.show()
    sys.exit(app.exec())

Output：

After clicking “Yes”：

After clicking “No”：

The window close and output nothing.