Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -15,13 +15,16 @@ def get_user(username, role=None):
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
         if role:
             query += f" AND c.role = '{role}'"
-        items = list(container.query_items(query=query))
+        # Agregar enable_cross_partition_query=True
+        items = list(container.query_items(
+            query=query,
+            enable_cross_partition_query=True
+        ))
         return items[0] if items else None
     except Exception as e:
         logger.error(f"Error al obtener usuario {username}: {str(e)}")
         return None
 
-
 #########################################
 def get_admin_user(username):
     return get_user(username, role='Administrador')
@@ -108,6 +111,7 @@ def record_login(username):
         return None
 
 #########################################
+
 def record_logout(username, session_id):
     """Registra el cierre de sesión y calcula la duración"""
     try:
@@ -122,9 +126,11 @@ def record_logout(username, session_id):
             {"name": "@username", "value": username}
         ]
         
+        # Agregar enable_cross_partition_query=True
         items = list(container.query_items(
             query=query,
-            parameters=params
+            parameters=params,
+            enable_cross_partition_query=True
         ))
 
         if not items:
@@ -138,8 +144,7 @@ def record_logout(username, session_id):
 
         session.update({
             "logoutTime": logout_time.isoformat(),
-            "sessionDuration": duration,
-            "partitionKey": username
+            "sessionDuration": duration
         })
 
         container.upsert_item(body=session)
@@ -166,11 +171,14 @@ def get_recent_sessions(limit=10):
         OFFSET 0 LIMIT @limit
         """
         
+        # Agregar enable_cross_partition_query=True
         sessions = list(container.query_items(
             query=query,
-            parameters=[{"name": "@limit", "value": limit}]
+            parameters=[{"name": "@limit", "value": limit}],
+            enable_cross_partition_query=True
         ))
 
+        # Validar y limpiar los datos
         clean_sessions = []
         for session in sessions:
             try:
@@ -190,6 +198,7 @@ def get_recent_sessions(limit=10):
         return []
 
 #########################################
+
 def get_user_total_time(username):
     """Obtiene el tiempo total que un usuario ha pasado en la plataforma"""
     try:
@@ -205,9 +214,11 @@ def get_user_total_time(username):
         AND IS_DEFINED(c.sessionDuration)
         """
         
+        # Agregar enable_cross_partition_query=True
         result = list(container.query_items(
             query=query,
-            parameters=[{"name": "@username", "value": username}]
+            parameters=[{"name": "@username", "value": username}],
+            enable_cross_partition_query=True
         ))
 
         return result[0] if result and result[0] is not None else 0