VWED_server/routes/task_record_api.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
任务运行记录API模块
提供任务运行记录相关的API接口
"""

from typing import Dict, List, Any, Optional
from fastapi import APIRouter, Query, Path, Request

from routes.common_api import format_response, error_response
from utils.logger import get_logger
from services.task_record_service import TaskRecordService
from routes.model.task_model import SetTaskErrorRequest

# 创建路由
router = APIRouter(
    prefix="/api/vwed-task-record",
    tags=["VWED任务运行记录"]
)
# router.add_route
# 设置日志
logger = get_logger("app.task_record_api")

@router.get("/blocks/{task_record_id}")
async def get_task_blocks(
    task_record_id: str = Path(..., description="任务记录ID")
):
    """
    获取指定任务记录下的所有块运行情况
    
    Args:
        task_record_id: 任务记录ID，由run_task接口返回的taskRecordId
        
    Returns:
        包含该任务记录下所有块运行记录的响应
    """
    try:
        result = await TaskRecordService.get_task_blocks(task_record_id)
        if not result["success"]:
            return error_response(message=result["message"], code=500)
        return format_response(data=result["data"], message=result["message"])
    except Exception as e:
        logger.error(f"获取任务块运行情况失败: {str(e)}")
        return error_response(message=f"获取任务块运行情况失败: {str(e)}", code=500)

@router.get("/block/{block_record_id}")
async def get_block_detail(
    block_record_id: str = Path(..., description="块记录ID")
):
    """
    获取指定块记录的详细信息
    
    Args:
        block_record_id: 块记录ID
        
    Returns:
        包含该块记录详细信息的响应
        
    """
    try:
        result = await TaskRecordService.get_block_detail(block_record_id)
        if not result["success"]:
            return error_response(message=result["message"], code=404)
        return format_response(data=result["data"], message=result["message"])
    except Exception as e:
        logger.error(f"获取块记录详情失败: {str(e)}")
        return error_response(message=f"获取块记录详情失败: {str(e)}", code=500)


@router.post("/stop/{task_record_id}")
async def stop_task_definition(task_record_id: str = Path(..., description="任务记录ID")):
    """
    停止指定任务记录下的所有运行任务实例，同时禁用定时任务
    
    Args:
        task_record_id: 任务记录ID
        
    Returns:
        包含停止结果的响应，包括成功停止的任务数量、是否为定时任务等信息
    """
    try:
        # 调用服务层方法
        result = await TaskRecordService.stop_task_record(task_record_id)
        if not result.get("success", False):
            return error_response(
                message=result.get("message", "停止任务失败"), 
                code=400
            )
        return format_response(
            data=result.get("data", {}),
            message=result.get("message", "任务终止成功")
        )
    except Exception as e:
        logger.log_error_with_trace(f"停止任务定义异常", e)
        return error_response(message=f"任务记录终止失败: {str(e)}", code=500)


@router.post("/set-error/{task_record_id}")
async def set_task_error(
    task_record_id: str = Path(..., description="任务记录ID"),
    request_data: SetTaskErrorRequest = None
):
    """
    将指定任务记录及其相关任务块状态设置为错误状态
    
    Args:
        task_record_id: 任务记录ID
        request_data: 请求数据，包含错误原因
        
    Returns:
        包含设置结果的响应，包括任务状态、错误原因、更新的任务块数量等信息
    """
    try:
        # 获取错误原因
        error_reason = request_data.error_reason if request_data else "未知错误"
        
        # 调用服务层方法
        result = await TaskRecordService.set_task_error(task_record_id, error_reason)
        if not result.get("success", False):
            return error_response(
                message=result.get("message", "设置任务错误状态失败"), 
                code=400
            )
        return format_response(
            data=result.get("data", {}),
            message=result.get("message", "任务状态设置为错误成功")
        )
    except Exception as e:
        logger.error(f"设置任务错误状态异常: {str(e)}")
        return error_response(message=f"设置任务错误状态失败: {str(e)}", code=500)


@router.get("/execution/block/results/{task_record_id}")
async def get_block_result(task_record_id: str = Path(..., description="任务记录ID")):
    """
    获取指定任务记录的执行结果
    
    Args:
        task_record_id: 任务记录ID
        
    Returns:
        包含执行结果的响应
    """
    try:
        result = await TaskRecordService.get_block_results(task_record_id)
        if not result["success"]:
            return error_response(message=result["message"], code=404)
        return format_response(data=result["data"], message=result["message"])
    except Exception as e:
        logger.error(f"获取任务记录执行结果失败: {str(e)}")
        return error_response(message=f"获取任务记录执行结果失败: {str(e)}", code=500)

@router.get("/detail/{task_record_id}")
async def get_task_record_detail(
    task_record_id: str = Path(..., description="任务记录ID")
):
    """
    获取指定任务记录的详细信息
    
    Args:
        task_record_id: 任务记录ID
        
    Returns:
        包含该任务记录详细信息的响应
    """
    try:
        result = await TaskRecordService.get_task_record_detail(task_record_id)
        if not result.get("success", False):
            return error_response(message=result.get("message", "任务记录不存在"), code=404)
        return format_response(data=result.get("data", {}), message="成功获取任务记录详情")
    except Exception as e:
        logger.error(f"获取任务记录详情失败: {str(e)}")
        return error_response(message=f"获取任务记录详情失败: {str(e)}", code=500)


@router.get("/test")
async def test_add_route():
    """
    动态添加路由测试
    """
    # 导入主应用实例
    from app import app
    
    # 定义新的路由处理函数
    def test_handler():
        return {"message": "动态添加的路由成功了!"}
    
    # 直接在应用实例上添加路由
    # 注意：路径必须是绝对路径，包含前缀
    full_path = "/api/vwed-task-record/name"
    
    # 检查路由是否已存在
    route_exists = False
    for route in app.routes:
        if getattr(route, "path", "") == full_path:
            route_exists = True
            break
    
    if not route_exists:
        app.add_api_route(full_path, test_handler, methods=["GET"])
        return {"message": f"路由已添加，请访问 {full_path} 测试"}
    else:
        return {"message": f"路由 {full_path} 已存在"}


@router.get("/remove-route")
async def remove_dynamic_route(route_path: str = Query("/name", description="要删除的路由路径")):
    """
    删除动态添加的路由
    
    Args:
        route_path: 要删除的路由路径，不包含前缀，例如 "/name"
        
    Returns:
        删除结果
    """
    # 导入主应用实例
    from app import app
    from fastapi.routing import APIRoute
    
    # 构建完整路径
    if not route_path.startswith("/"):
        route_path = "/" + route_path
    full_path = f"/api/vwed-task-record{route_path}"
    
    # 查找要删除的路由
    route_found = False
    new_routes = []
    
    for route in app.routes:
        if getattr(route, "path", "") == full_path:
            route_found = True
            continue
        new_routes.append(route)
    
    if route_found:
        # 替换路由列表
        app.router.routes = new_routes
        
        # 重新生成路由查找表
        app.router.routes_by_name = {}
        for route in new_routes:
            if isinstance(route, APIRoute) and route.name:
                app.router.routes_by_name[route.name] = route
        
        return {"success": True, "message": f"路由 {full_path} 已删除"}
    else:
        return {"success": False, "message": f"路由 {full_path} 不存在"}