TimerMessageBox.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-

import sys
from soma.qt_gui.qt_backend import QtCore, QtWidgets


class TimerMessageBox(QtWidgets.QMessageBox):
    """ Message box with a countdown timer that closes automatically """
    def __init__(self, timeout=3, parent=None):
        super(TimerMessageBox, self).__init__(parent)
        self.setWindowTitle("NOT FOR MEDICAL USE")
        self.time_to_wait = timeout
        self.changeContent()
        self.setStandardButtons(QtWidgets.QMessageBox.NoButton)
        self.timer = QtCore.QTimer(self)
        self.timer.setInterval(1000)
        self.timer.timeout.connect(self.changeContent)
        self.timer.start()

    def changeContent(self):
        self.setText("NOT FOR MEDICAL USE (closing automatically in {0} secondes.)".format(self.time_to_wait))

        if self.time_to_wait <= 0:
            self.close()
        
        self.time_to_wait -= 1

    def closeEvent(self, event):
        self.timer.stop()
        event.accept()


class ExampleTimerMessageBox(QtWidgets.QWidget):
    """ Class for testing TimerMessageBox """
    def __init__(self):
        super(ExampleTimerMessageBox, self).__init__()
        btn = QtWidgets.QPushButton('Button', self)
        btn.resize(btn.sizeHint())
        btn.move(50, 50)
        self.setWindowTitle('Example')
        btn.clicked.connect(self.warning)

    def warning(self):
        messagebox = TimerMessageBox(5, self)
        messagebox.exec_()


def main():
    app = QtWidgets.QApplication(sys.argv)
    ex = ExampleTimerMessageBox()
    ex.show()
    sys.exit(app.exec_())


if __name__ == '__main__':
    main()