Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -13,11 +13,13 @@ logger = logging.getLogger(__name__)
 def get_user(username, role=None):
     container = get_container("users")
     try:
+        # Para una búsqueda por id (que es el partition key), no necesitamos cross partition query
         query = f"SELECT * FROM c WHERE c.id = '{username}'"
         if role:
             query += f" AND c.role = '{role}'"
         items = list(container.query_items(
-            query=query  # Removido enable_cross_partition_query
+            query=query,
+            partition_key=username  # Especificamos el partition key
         ))
         return items[0] if items else None
     except Exception as e:
@@ -36,6 +38,8 @@ def get_student_user(username):
 def get_teacher_user(username):
     return get_user(username, role='Profesor')
 
+
+###########################################
 def create_user(username, password, role, additional_info=None):
     """Crea un nuevo usuario"""
     container = get_container("users")
@@ -45,7 +49,7 @@ def create_user(username, password, role, additional_info=None):
         
     try:
         user_data = {
-            'id': username,
+            'id': username,  # Este será usado como partition key
             'password': password,
             'role': role,
             'timestamp': datetime.now(timezone.utc).isoformat(),
@@ -54,7 +58,7 @@ def create_user(username, password, role, additional_info=None):
         
         container.create_item(
             body=user_data,
-            partition_key=username
+            partition_key=username  # Usamos el username (que es el id) como partition key
         )
         logger.info(f"Usuario {role} creado: {username}")
         return True
@@ -63,6 +67,7 @@ def create_user(username, password, role, additional_info=None):
         logger.error(f"Error al crear usuario {role}: {str(e)}")
         return False
 
+###########################################
 def create_student_user(username, password, additional_info=None):
     return create_user(username, password, 'Estudiante', additional_info)