Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -58,49 +58,49 @@ def create_admin_user(username, password, additional_info=None):
 
 ############-- Funciones de control del tiempo de sesión ##################
 
+
 def record_login(username):
     """Registra el inicio de sesión de un usuario"""
     try:
-        user_container, _, session_container = get_sql_containers()  # Obtener el contenedor correcto
+        _, _, session_container = get_sql_containers()
         
         session_doc = {
             "id": str(uuid.uuid4()),
             "type": "session",
             "username": username,
             "loginTime": datetime.now(timezone.utc).isoformat(),
-            "partitionKey": username  # Asegurarse que esto coincida con la configuración de la colección
+            "partitionKey": username  # Asegurarse que coincida con la configuración
         }
 
+        # Crear el documento sin enable_cross_partition_query
         result = session_container.create_item(
             body=session_doc,
-            enable_cross_partition_query=True  # Agregado para manejar particiones
+            partition_key=username  # Usar partition_key en lugar de partitionKey
         )
+        logger.info(f"Sesión registrada para {username}")
         return result['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"""
     try:
-        user_container, _, session_container = get_sql_containers()
-        
-        query = "SELECT * FROM c WHERE c.id = @id AND c.username = @username"
-        params = [
-            {"name": "@id", "value": session_id},
-            {"name": "@username", "value": username}
-        ]
+        _, _, session_container = get_sql_containers()
         
-        sessions = list(session_container.query_items(
-            query=query,
-            parameters=params,
-            enable_cross_partition_query=True
+        # Consulta usando la partition key correctamente
+        items = list(session_container.query_items(
+            query="SELECT * FROM c WHERE c.id = @id",
+            parameters=[{"name": "@id", "value": session_id}],
+            partition_key=username  # Usar la partition key correcta
         ))
-
-        if not sessions:
+        
+        if not items:
+            logger.warning(f"Sesión no encontrada: {session_id}")
             return False
 
-        session = sessions[0]
+        session = items[0]
         login_time = datetime.fromisoformat(session['loginTime'].rstrip('Z'))
         logout_time = datetime.now(timezone.utc)
         duration = int((logout_time - login_time).total_seconds())
@@ -110,16 +110,22 @@ def record_logout(username, session_id):
             "sessionDuration": duration
         })
 
-        session_container.upsert_item(body=session)
+        # Actualizar usando la partition key
+        session_container.upsert_item(
+            body=session,
+            partition_key=username
+        )
+        logger.info(f"Sesión cerrada para {username}, duración: {duration}s")
         return True
     except Exception as e:
         logger.error(f"Error registrando logout: {str(e)}")
         return False
 
+
 def get_recent_sessions(limit=10):
     """Obtiene las sesiones más recientes"""
     try:
-        user_container, _, session_container = get_sql_containers()
+        _, _, session_container = get_sql_containers()
         
         query = """
         SELECT c.username, c.loginTime, c.logoutTime, c.sessionDuration
@@ -129,20 +135,17 @@ def get_recent_sessions(limit=10):
         OFFSET 0 LIMIT @limit
         """
         
-        params = [{"name": "@limit", "value": limit}]
-        
-        sessions = session_container.query_items(
+        # Usar query_items sin enable_cross_partition_query
+        sessions = list(session_container.query_items(
             query=query,
-            parameters=params,
-            enable_cross_partition_query=True
-        )
-
-        return list(sessions)
+            parameters=[{"name": "@limit", "value": limit}]
+        ))
+        
+        return sessions
     except Exception as e:
         logger.error(f"Error obteniendo sesiones recientes: {str(e)}")
         return []
 
-
 def get_user_total_time(username):
     """Obtiene el tiempo total que un usuario ha pasado en la plataforma"""
     try: