Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -9,16 +9,34 @@ 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:
-        query = f"SELECT * FROM c WHERE c.id = '{username}'"
+        # Construir la consulta
+        query = "SELECT * FROM c WHERE c.id = @username"
+        parameters = [{"name": "@username", "value": username}]
+        
         if role:
-            query += f" AND c.role = '{role}'"
+            query += " AND c.role = @role"
+            parameters.append({"name": "@role", "value": role})
+        
         items = list(container.query_items(
             query=query,
-            partition_key=username
+            parameters=parameters,
+            enable_cross_partition_query=True  # Necesario para búsquedas por rol
         ))
-        return items[0] if items else None
+        
+        if items:
+            logger.info(f"Usuario encontrado: {username}")
+            return items[0]
+        else:
+            logger.warning(f"Usuario no encontrado: {username}")
+            return None
+            
     except Exception as e:
         logger.error(f"Error al obtener usuario {username}: {str(e)}")
         return None
@@ -33,21 +51,28 @@ 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")
+    if not container:
+        logger.error("No se pudo obtener el contenedor de usuarios")
+        return False
+        
     try:
         user_data = {
             'id': username,
-            'password': password,  # Ya viene hasheado
+            'password': password,
             'role': role,
             'timestamp': datetime.now(timezone.utc).isoformat(),
             'additional_info': additional_info or {}
         }
+        
         container.create_item(
             body=user_data,
             partition_key=username
         )
         logger.info(f"Usuario {role} creado: {username}")
         return True
+        
     except Exception as e:
         logger.error(f"Error al crear usuario {role}: {str(e)}")
         return False