main.py

from PyQt6.QtCore import QPropertyAnimation
from PyQt6 import QtWidgets as qtw
from UI.UI_base import Ui_MainWindow
from back_end.arduino_serial import arduinoSerial
from back_end.read_database import readDB


class MainWindow(qtw.QMainWindow, Ui_MainWindow):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.setupUi(self)

        self.page_widgets.setCurrentWidget(self.serial_page)
        self.slide_menu.setMaximumWidth(0)
        self.serial = arduinoSerial(self.serial_out)
        self.readDB = readDB('./back_end/database.db')
        self.scrollArea.setVerticalScrollBar(self.verticalScrollBar)
        self.tabelaUsuarios.setVerticalScrollBar(self.scrollUsuarios)
        self.tabelaEventos.setVerticalScrollBar(self.scrollEventos)

        self.btn_menu.clicked.connect(lambda: self.toggleMenu(48))  # slide menu
        self.btn_serial.clicked.connect(self.go_to_serial)
        self.btn_database.clicked.connect(self.go_to_database)
        self.serialInput.returnPressed.connect(self.get_text)

    def go_to_serial(self):  # muda de página
        self.page_widgets.setCurrentWidget(self.serial_page)
        self.highlight_button()

    def go_to_database(self):  # muda de página
        self.page_widgets.setCurrentWidget(self.database_page)
        self.highlight_button()

        self.load_tables()

    def get_text(self):  # envia a linha do EditText para o arduino
        self.serial.enviar_serial(self.serialInput.text())
        self.serialInput.setText('')

    def serial_out(self, entrada):  # mostra a resposta do node no aplicativo
        texto = entrada + '\n' + self.serialOutput.text()
        self.serialOutput.setText(texto)

    def load_tables(self):  # carrega a informação do banco de dados para o aplicativo
        usuarios = self.readDB.get_usuarios()
        eventos = self.readDB.get_eventos()

        self.tabelaUsuarios.setRowCount(len(usuarios))  # define quantas linhas as tabelas terão
        self.tabelaEventos.setRowCount(len(eventos))

        for i, usuario in enumerate(usuarios):
            self.tabelaUsuarios.setItem(i, 0, qtw.QTableWidgetItem(usuario[0]))
            self.tabelaUsuarios.setItem(i, 1, qtw.QTableWidgetItem(usuario[1]))
            self.tabelaUsuarios.setItem(i, 2, qtw.QTableWidgetItem(usuario[2]))

        for i, evento in enumerate(eventos):
            self.tabelaEventos.setItem(i, 0, qtw.QTableWidgetItem(evento[0]))
            self.tabelaEventos.setItem(i, 1, qtw.QTableWidgetItem(evento[1]))
            self.tabelaEventos.setItem(i, 2, qtw.QTableWidgetItem(evento[2]))
            self.tabelaEventos.setItem(i, 3, qtw.QTableWidgetItem(evento[3]))

        self.tabelaUsuarios.resizeColumnsToContents()  # automaticamente redimensiona as colunas
        self.tabelaEventos.resizeColumnsToContents()

    def highlight_button(self):  # muda a cor do botão dependendo da página
        self.btn_serial.setAutoFillBackground(False)
        self.btn_database.setAutoFillBackground(False)

        page = self.page_widgets.currentWidget().objectName()
        if page == 'serial_page':
            self.btn_serial.setAutoFillBackground(True)
        if page == 'database_page':
            self.btn_database.setAutoFillBackground(True)

    def toggleMenu(self, maxWidth):  # animação do menu lateral
        width = self.slide_menu.width()
        standard = 0  # minimum width
        widthExtended = standard

        if width == standard:
            widthExtended = maxWidth

        # ANIMATION
        self.animation = QPropertyAnimation(self.slide_menu, b"maximumWidth")
        self.animation.setDuration(150)
        self.animation.setStartValue(width)
        self.animation.setEndValue(widthExtended)
        self.animation.start()

        self.highlight_button()


if __name__ == '__main__':
    import sys

    app = qtw.QApplication([])

    MainWindow = MainWindow()
    MainWindow.show()

    sys.exit(app.exec())