gui.py

#!/usr/bin/env python3
import sys
from time import sleep

from PyQt5 import QtWidgets
from PyQt5.QtCore import QTimer
from gpiozero import DigitalOutputDevice, BadPinFactory

from ui.test import Ui_MainWindow
from virtual_digital_output_device import VirtualDigitalOutputDevice


class MainWindow(Ui_MainWindow):
    def __init__(self):
        self.app = QtWidgets.QApplication(sys.argv)
        self.main_window = QtWidgets.QMainWindow()
        self.setupUi(self.main_window)

        self.pushButton.clicked.connect(self.button1)
        self.pushButton_2.clicked.connect(self.button2)
        self.pushButton_3.clicked.connect(self.button3)
        self.pushButton_4.clicked.connect(self.button4)
        self.pushButton_5.clicked.connect(self.button5)
        self.pushButton_6.clicked.connect(self.button6)
        self.pushButtonTurnOnPrecharge.clicked.connect(self.turn_on_precharge)
        timer = QTimer(self.main_window)
        timer.timeout.connect(self.check_states)
        timer.start(1000)

    def show(self):
        self.main_window.show()
        self.app.exec_()

    def turn_on_precharge(self):
        out2.on()
        self.check_states()
        sleep(1)
        out3.on()
        self.check_states()
        sleep(5)
        out1.on()
        self.check_states()
        sleep(0.5)
        out3.off()
        self.check_states()

    def check_states(self):
        self.label.setText('On' if out1.is_active else 'Off')
        self.label_2.setText('On' if out2.is_active else 'Off')
        self.label_3.setText('On' if out3.is_active else 'Off')
        self.label_4.setText('On' if out4.is_active else 'Off')
        self.label_5.setText('On' if out5.is_active else 'Off')
        self.label_6.setText('On' if out6.is_active else 'Off')
        self.main_window.repaint()

    def button1(self):
        out1.toggle()
        self.check_states()

    def button2(self):
        out2.toggle()
        self.check_states()

    def button3(self):
        out3.toggle()
        self.check_states()

    def button4(self):
        out4.toggle()
        self.check_states()

    def button5(self):
        out5.toggle()
        self.check_states()

    def button6(self):
        out6.toggle()
        self.check_states()


if __name__ == '__main__':
    try:
        out5 = DigitalOutputDevice(12)
        out6 = DigitalOutputDevice(6)
        out1 = DigitalOutputDevice(5)
        out2 = DigitalOutputDevice(13)
        out3 = DigitalOutputDevice(17)
        out4 = DigitalOutputDevice(18)
    except BadPinFactory as e:
        print(f'{e}')
        out5 = VirtualDigitalOutputDevice(12)
        out6 = VirtualDigitalOutputDevice(6)
        out1 = VirtualDigitalOutputDevice(5)
        out2 = VirtualDigitalOutputDevice(13)
        out3 = VirtualDigitalOutputDevice(17)
        out4 = VirtualDigitalOutputDevice(18)

    main_window = MainWindow()
    main_window.show()