Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -9,34 +9,16 @@ import uuid
 logger = logging.getLogger(__name__)
 
 def get_user(username, role=None):
-    """Obtiene un usuario por su username y opcionalmente por rol"""
     container = get_container("users")
-    if not container:
-        logger.error("No se pudo obtener el contenedor de usuarios")
-        return None
-        
     try:
-        # Construir la consulta
-        query = "SELECT * FROM c WHERE c.id = @username"
-        parameters = [{"name": "@username", "value": username}]
-        
+        query = f"SELECT * FROM c WHERE c.id = '{username}'"
         if role:
-            query += " AND c.role = @role"
-            parameters.append({"name": "@role", "value": role})
-        
+            query += f" AND c.role = '{role}'"
         items = list(container.query_items(
             query=query,
-            parameters=parameters,
-            enable_cross_partition_query=True  # Necesario para búsquedas por rol
+            enable_cross_partition_query=True
         ))
-        
-        if items:
-            logger.info(f"Usuario encontrado: {username}")
-            return items[0]
-        else:
-            logger.warning(f"Usuario no encontrado: {username}")
-            return None
-            
+        return items[0] if items else None
     except Exception as e:
         logger.error(f"Error al obtener usuario {username}: {str(e)}")
         return None
@@ -90,11 +72,10 @@ def create_admin_user(username, password, additional_info=None):
 
 
 def record_login(username):
-    """Registra el inicio de sesión de un usuario"""
+    """Registra el inicio de sesión"""
     try:
         container = get_container("users_sessions")
         if not container:
-            logger.error("No se pudo obtener el contenedor users_sessions")
             return None
 
         session_id = str(uuid.uuid4())
@@ -102,36 +83,39 @@ def record_login(username):
             "id": session_id,
             "type": "session",
             "username": username,
-            "loginTime": datetime.now(timezone.utc).isoformat()
+            "loginTime": datetime.now(timezone.utc).isoformat(),
+            "additional_info": {}  # Mantener consistencia con la estructura existente
         }
 
-        result = container.create_item(
+        container.create_item(
             body=session_doc,
             partition_key=username
         )
-        logger.info(f"Sesión {session_id} registrada para {username}")
+        logger.info(f"Login registrado: {username}")
         return session_id
     except Exception as e:
         logger.error(f"Error registrando login: {str(e)}")
         return None
 
 
+
 def record_logout(username, session_id):
-    """Registra el cierre de sesión y calcula la duración"""
+    """Registra el cierre de sesión"""
     try:
         container = get_container("users_sessions")
         if not container:
-            logger.error("No se pudo obtener el contenedor users_sessions")
             return False
 
+        query = "SELECT * FROM c WHERE c.id = @id"
+        params = [{"name": "@id", "value": session_id}]
+        
         items = list(container.query_items(
-            query="SELECT * FROM c WHERE c.id = @id",
-            parameters=[{"name": "@id", "value": session_id}],
+            query=query,
+            parameters=params,
             partition_key=username
         ))
 
         if not items:
-            logger.warning(f"Sesión no encontrada: {session_id}")
             return False
 
         session = items[0]
@@ -148,7 +132,6 @@ def record_logout(username, session_id):
             body=session,
             partition_key=username
         )
-        logger.info(f"Sesión {session_id} cerrada para {username}, duración: {duration}s")
         return True
     except Exception as e:
         logger.error(f"Error registrando logout: {str(e)}")
@@ -156,10 +139,12 @@ def record_logout(username, session_id):
 
 
 def get_recent_sessions(limit=10):
-    """Obtiene las sesiones más recientes"""
+    """Obtiene sesiones recientes"""
     try:
-        _, _, session_container = get_sql_containers()
-        
+        container = get_container("users_sessions")
+        if not container:
+            return []
+
         query = """
         SELECT c.username, c.loginTime, c.logoutTime, c.sessionDuration
         FROM c
@@ -168,15 +153,13 @@ def get_recent_sessions(limit=10):
         OFFSET 0 LIMIT @limit
         """
         
-        # Usar query_items sin enable_cross_partition_query
-        sessions = list(session_container.query_items(
+        return list(container.query_items(
             query=query,
-            parameters=[{"name": "@limit", "value": limit}]
+            parameters=[{"name": "@limit", "value": limit}],
+            enable_cross_partition_query=True
         ))
-        
-        return sessions
     except Exception as e:
-        logger.error(f"Error obteniendo sesiones recientes: {str(e)}")
+        logger.error(f"Error obteniendo sesiones: {str(e)}")
         return []
 
 def get_user_total_time(username):