Password promt

backend/src/api/routes/migration.py

@@ -55,12 +55,17 @@ async def execute_migration(selection: DashboardSelection, config_manager=Depend
 
     # Create migration task with debug logging
     from ...core.logger import logger
-    logger.info(f"Creating migration task with selection: {selection.dict()}")
+    
+    # Include replace_db_config in the task parameters
+    task_params = selection.dict()
+    task_params['replace_db_config'] = selection.replace_db_config
+    
+    logger.info(f"Creating migration task with params: {task_params}")
     logger.info(f"Available environments: {env_ids}")
     logger.info(f"Source env: {selection.source_env_id}, Target env: {selection.target_env_id}")
     
     try:
-        task = await task_manager.create_task("superset-migration", selection.dict())
+        task = await task_manager.create_task("superset-migration", task_params)
         logger.info(f"Task created successfully: {task.id}")
         return {"task_id": task.id, "message": "Migration initiated"}
     except Exception as e:

backend/src/api/routes/tasks.py

@@ -41,12 +41,15 @@ async def create_task(
 
 @router.get("/", response_model=List[Task])
 async def list_tasks(
+    limit: int = 10,
+    offset: int = 0,
+    status: Optional[TaskStatus] = None,
     task_manager: TaskManager = Depends(get_task_manager)
 ):
     """
-    Retrieve a list of all tasks.
+    Retrieve a list of tasks with pagination and optional status filter.
     """
-    return task_manager.get_all_tasks()
+    return task_manager.get_tasks(limit=limit, offset=offset, status=status)
 
 @router.get("/{task_id}", response_model=Task)
 async def get_task(
@@ -61,6 +64,19 @@ async def get_task(
         raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Task not found")
     return task
 
+@router.get("/{task_id}/logs", response_model=List[LogEntry])
+async def get_task_logs(
+    task_id: str,
+    task_manager: TaskManager = Depends(get_task_manager)
+):
+    """
+    Retrieve logs for a specific task.
+    """
+    task = task_manager.get_task(task_id)
+    if not task:
+        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Task not found")
+    return task_manager.get_task_logs(task_id)
+
 @router.post("/{task_id}/resolve", response_model=Task)
 async def resolve_task(
     task_id: str,
@@ -90,4 +106,15 @@ async def resume_task(
         return task_manager.get_task(task_id)
     except ValueError as e:
         raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e))
+
+@router.delete("/", status_code=status.HTTP_204_NO_CONTENT)
+async def clear_tasks(
+    status: Optional[TaskStatus] = None,
+    task_manager: TaskManager = Depends(get_task_manager)
+):
+    """
+    Clear tasks matching the status filter. If no filter, clears all non-running tasks.
+    """
+    task_manager.clear_tasks(status)
+    return
 # [/DEF]