Update modules/auth/auth.py

modules/auth/auth.py

@@ -14,7 +14,9 @@ from ..database.sql_db import (
     record_login,
     record_logout
 )
+
 import logging
+
 from datetime import datetime, timezone
 
 logger = logging.getLogger(__name__)
@@ -39,12 +41,10 @@ if not endpoint or not key:
 
 key = clean_and_validate_key(key)
 
+
 def authenticate_user(username, password):
-    """
-    Autentica un usuario y registra el inicio de sesión
-    """
+    """Autentica un usuario y registra el inicio de sesión"""
     try:
-        # Primero intentar obtener usuario general
         user_item = get_user(username)
         
         if not user_item:
@@ -54,16 +54,18 @@ def authenticate_user(username, password):
         if verify_password(user_item['password'], password):
             logger.info(f"Usuario autenticado: {username}, Rol: {user_item['role']}")
             
-            # Registrar la sesión
             try:
                 session_id = record_login(username)
                 if session_id:
                     st.session_state.session_id = session_id
+                    st.session_state.username = username
                     st.session_state.login_time = datetime.now(timezone.utc).isoformat()
+                    logger.info(f"Sesión iniciada: {session_id}")
+                else:
+                    logger.warning("No se pudo registrar la sesión")
             except Exception as e:
                 logger.error(f"Error al registrar inicio de sesión: {str(e)}")
-                # Continuar aunque falle el registro de sesión
-                
+            
             return True, user_item['role']
             
         logger.warning(f"Contraseña incorrecta para usuario: {username}")
@@ -150,16 +152,18 @@ def logout():
     """Cierra la sesión del usuario"""
     try:
         if 'session_id' in st.session_state and 'username' in st.session_state:
-            record_logout(
+            success = record_logout(
                 st.session_state.username, 
                 st.session_state.session_id
             )
-            logger.info(f"Sesión cerrada: {st.session_state.username}")
-            
-        st.session_state.clear()
+            if success:
+                logger.info(f"Sesión cerrada: {st.session_state.username}")
+            else:
+                logger.warning(f"Error al registrar cierre de sesión: {st.session_state.username}")
         
     except Exception as e:
         logger.error(f"Error en logout: {str(e)}")
+    finally:
         st.session_state.clear()
 
 def hash_password(password):