1 files changed, 16 insertions, 6 deletions
diff --git a/src/gui/pageRegister.py b/src/gui/pageRegister.py
index aef8ce1..f0fc392 100644
--- a/src/gui/pageRegister.py
+++ b/src/gui/pageRegister.py
@@ -4,23 +4,33 @@ from PyQt5.QtWidgets import (
     QLineEdit, QWidget, QVBoxLayout, QComboBox, QGridLayout
 )
 from src.gui import pageHome, pageLogin
-from src.gui.include import connexionInfo
-from src.core.coreUser import core_register
+from src.core.coreUser import core_register, core_logout
 
 
 def main(mainWindow) -> None:
-    connexionState = connexionInfo.get_text(mainWindow)
+    if not mainWindow.userConnected:
+        connexionState = QLabel("Non connecté.")
+    else:
+        connexionState = QLabel(f"Connecté: {mainWindow.userEmail} - {mainWindow.userStatus}")
 
     btnAccueil = QPushButton("Accueil")
     btnAccueil.clicked.connect(lambda: pageHome.main(mainWindow))
 
-    btnConnexion = QPushButton("Connexion")
-    btnConnexion.clicked.connect(lambda: pageLogin.main(mainWindow))
+    if not mainWindow.userConnected:
+        btnConnexion = QPushButton("Connexion")
+        btnConnexion.clicked.connect(lambda: pageLogin.main(mainWindow))
+    else:
+        btnDeconnexion = QPushButton("Déconnexion")
+        btnDeconnexion.clicked.connect(lambda: core_logout(mainWindow))
 
     layoutUser = QHBoxLayout()
     layoutUser.addWidget(btnAccueil)
     layoutUser.addStretch()
-    layoutUser.addWidget(btnConnexion)
+    if not mainWindow.userConnected:
+        layoutUser.addWidget(btnConnexion)
+    else:
+        layoutUser.addWidget(btnDeconnexion)
+
     grpUtilisateur = QGroupBox("Gestion de l'utilisateur")
     grpUtilisateur.setLayout(layoutUser)