import sys from lib.db import Session, Users from PyQt5.QtCore import QTimer, pyqtSignal from PyQt5.QtGui import QKeySequence from PyQt5.QtWidgets import QMessageBox, QShortcut from ui.widget import Widget class Login(Widget): wrong_credentials = pyqtSignal() successful_login = pyqtSignal() def __init__(self): super().__init__() self.welcome_l.setText("BENVENUTO, PER INIZIARE IL COLLAUDO, EFFETTUA L'ACCESSO:") self.user_cb.addItems(Users.get_usernames()) QShortcut(QKeySequence("Return"), self).activated.connect(self.login_b.click) self.login_b.clicked.connect(self.try_login) # TESTING if "--auto-login-admin" in sys.argv: self.user_cb.setCurrentText("ADMIN") self.password_le.setText("123123") if "--auto-login-user" in sys.argv or "--test" in sys.argv: self.user_cb.setCurrentText("USER") self.password_le.setText("user") if "--auto-login-admin" in sys.argv or "--auto-login-user" in sys.argv or "--test" in sys.argv: self.test_timer = QTimer() self.test_timer.setSingleShot(True) self.test_timer.timeout.connect(self.login_b.clicked.emit) self.test_timer.start(500) # /TESTING def try_login(self): user = self.user_cb.currentText().upper() password = self.password_le.text() session = Users.login(user, password) if type(session) is not Session: self.wrong_credentials.emit() QMessageBox.critical(None, "Errore login", "Credenziali errate") else: self.successful_login.emit()