Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -8,6 +8,8 @@ import uuid
 
 logger = logging.getLogger(__name__)
 
+###########################################
+
 def get_user(username, role=None):
     container = get_container("users")
     try:
@@ -15,14 +17,16 @@ def get_user(username, role=None):
         if role:
             query += f" AND c.role = '{role}'"
         items = list(container.query_items(
-            query=query,
-            enable_cross_partition_query=True
+            query=query  # Removido enable_cross_partition_query
         ))
         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')
 
@@ -87,14 +91,11 @@ def record_login(username):
             "username": username,
             "loginTime": datetime.now(timezone.utc).isoformat(),
             "additional_info": {},
-            # El campo para partition key debe estar en el documento
             "partitionKey": username
         }
 
-        # Crear el documento usando options
         result = container.create_item(
-            body=session_doc,
-            enable_cross_partition_query=True
+            body=session_doc  # Solo el body
         )
         logger.info(f"Sesión {session_id} registrada para {username}")
         return session_id
@@ -102,7 +103,6 @@ def record_login(username):
         logger.error(f"Error registrando login: {str(e)}")
         return None
 
-
 ###########################################################
 
 def record_logout(username, session_id):
@@ -113,7 +113,6 @@ def record_logout(username, session_id):
             logger.error("No se pudo obtener el contenedor users_sessions")
             return False
 
-        # Obtener la sesión actual
         query = "SELECT * FROM c WHERE c.id = @id AND c.username = @username"
         params = [
             {"name": "@id", "value": session_id},
@@ -122,8 +121,7 @@ def record_logout(username, session_id):
         
         items = list(container.query_items(
             query=query,
-            parameters=params,
-            enable_cross_partition_query=True
+            parameters=params  # Removido enable_cross_partition_query
         ))
 
         if not items:
@@ -137,12 +135,12 @@ def record_logout(username, session_id):
 
         session.update({
             "logoutTime": logout_time.isoformat(),
-            "sessionDuration": duration
+            "sessionDuration": duration,
+            "partitionKey": username  # Asegurar que partitionKey esté presente
         })
 
-        # Actualizar el documento
         container.upsert_item(
-            body=session
+            body=session  # Solo el body
         )
         logger.info(f"Sesión {session_id} cerrada para {username}, duración: {duration}s")
         return True
@@ -171,11 +169,9 @@ def get_recent_sessions(limit=10):
         
         sessions = list(container.query_items(
             query=query,
-            parameters=[{"name": "@limit", "value": limit}],
-            enable_cross_partition_query=True
+            parameters=[{"name": "@limit", "value": limit}]  # Removido enable_cross_partition_query
         ))
 
-        # Validar y limpiar los datos
         clean_sessions = []
         for session in sessions:
             try:
@@ -215,8 +211,7 @@ def get_user_total_time(username):
         
         result = list(container.query_items(
             query=query,
-            parameters=[{"name": "@username", "value": username}],
-            enable_cross_partition_query=True
+            parameters=[{"name": "@username", "value": username}]  # Removido enable_cross_partition_query
         ))
 
         return result[0] if result and result[0] is not None else 0