tianfeng_task_modules/api/task_instance_api.py

"""
任务实例API模块
提供任务实例的增删改查接口
"""
from typing import Dict, Any, List, Optional
from fastapi import APIRouter, HTTPException, Query
from pydantic import BaseModel, Field
from services.task_instance_service import TaskInstanceService, TaskInstanceStatus
from api.models import ApiResponse

# 创建路由器
router = APIRouter(prefix="/task-instances", tags=["任务实例"])

# 创建服务实例
task_instance_service = TaskInstanceService()

# 请求模型
class TaskInstanceCreateInput(BaseModel):
    """任务实例创建输入模型"""
    task_id: str = Field(..., description="任务ID")
    name: Optional[str] = Field(None, description="任务名称")
    variables: Optional[Dict[str, Any]] = Field(None, description="任务变量")
    priority: int = Field(1, description="任务优先级")
    input_params: Optional[Dict[str, Any]] = Field(None, description="任务输入参数")
    block_outputs: Optional[Dict[str, Any]] = Field(None, description="块输出参数")
    context_params: Optional[Dict[str, Any]] = Field(None, description="上下文参数")

class TaskInstanceUpdateInput(BaseModel):
    """任务实例更新输入模型"""
    variables: Optional[Dict[str, Any]] = Field(None, description="任务变量")
    priority: Optional[int] = Field(None, description="任务优先级")
    input_params: Optional[Dict[str, Any]] = Field(None, description="任务输入参数")
    block_outputs: Optional[Dict[str, Any]] = Field(None, description="块输出参数")
    context_params: Optional[Dict[str, Any]] = Field(None, description="上下文参数")

# API接口
@router.post("/create", response_model=ApiResponse)
async def create_task_instance(instance_input: TaskInstanceCreateInput):
    """创建任务实例"""
    try:
        # 创建任务实例
        instance = task_instance_service.create_instance(
            task_id=instance_input.task_id,
            name=instance_input.name,
            variables=instance_input.variables,
            priority=instance_input.priority,
            input_params=instance_input.input_params,
            block_outputs=instance_input.block_outputs,
            context_params=instance_input.context_params
        )
        
        return {
            "code": 200,
            "message": "创建任务实例成功",
            "data": instance
        }
    except ValueError as e:
        return {
            "code": 400,
            "message": str(e),
            "data": None
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"创建任务实例失败: {str(e)}")

@router.get("/{instance_id}", response_model=ApiResponse)
async def get_task_instance(instance_id: str):
    """获取任务实例详情"""
    try:
        # 获取任务实例
        instance = task_instance_service.get_instance_by_id(instance_id)
        if not instance:
            return {
                "code": 404,
                "message": f"任务实例不存在: {instance_id}",
                "data": None
            }
        
        return {
            "code": 200,
            "message": "获取任务实例成功",
            "data": instance
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取任务实例失败: {str(e)}")

@router.get("/task/{task_id}", response_model=ApiResponse)
async def get_task_instances(task_id: str):
    """获取任务的所有实例"""
    try:
        # 获取任务实例
        instances = task_instance_service.get_instances_by_task_id(task_id)
        
        return {
            "code": 200,
            "message": "获取任务实例列表成功",
            "data": {
                "instances": instances,
                "total": len(instances)
            }
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取任务实例列表失败: {str(e)}")

@router.get("/task/{task_id}/latest", response_model=ApiResponse)
async def get_latest_task_instance(task_id: str):
    """获取任务的最新实例"""
    try:
        # 获取最新任务实例
        instance = task_instance_service.get_latest_instance_by_task_id(task_id)
        if not instance:
            return {
                "code": 404,
                "message": f"任务 {task_id} 没有实例",
                "data": None
            }
        
        return {
            "code": 200,
            "message": "获取最新任务实例成功",
            "data": instance
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取最新任务实例失败: {str(e)}")

@router.put("/{instance_id}", response_model=ApiResponse)
async def update_task_instance(instance_id: str, instance_input: TaskInstanceUpdateInput):
    """更新任务实例"""
    try:
        # 更新任务实例
        instance = task_instance_service.update_instance(
            instance_id=instance_id,
            variables=instance_input.variables,
            priority=instance_input.priority,
            input_params=instance_input.input_params,
            block_outputs=instance_input.block_outputs,
            context_params=instance_input.context_params
        )
        
        if not instance:
            return {
                "code": 404,
                "message": f"任务实例不存在: {instance_id}",
                "data": None
            }
        
        return {
            "code": 200,
            "message": "更新任务实例成功",
            "data": instance
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"更新任务实例失败: {str(e)}")

@router.delete("/{instance_id}", response_model=ApiResponse)
async def delete_task_instance(instance_id: str):
    """删除任务实例"""
    try:
        # 删除任务实例
        success = task_instance_service.delete_instance(instance_id)
        
        if not success:
            return {
                "code": 404,
                "message": f"任务实例不存在: {instance_id}",
                "data": None
            }
        
        return {
            "code": 200,
            "message": "删除任务实例成功",
            "data": None
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"删除任务实例失败: {str(e)}")

@router.post("/{instance_id}/publish", response_model=ApiResponse)
async def publish_task_instance(instance_id: str):
    """发布任务实例"""
    try:
        # 发布任务实例
        instance = task_instance_service.publish_instance(instance_id)
        
        if not instance:
            return {
                "code": 404,
                "message": f"任务实例不存在: {instance_id}",
                "data": None
            }
        
        return {
            "code": 200,
            "message": "发布任务实例成功",
            "data": instance
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"发布任务实例失败: {str(e)}")

@router.get("/task/{task_id}/editing", response_model=ApiResponse)
async def get_or_create_editing_instance(task_id: str):
    """获取或创建编辑中的任务实例"""
    try:
        # 获取或创建编辑中的任务实例
        instance = task_instance_service.get_or_create_editing_instance(task_id)
        
        return {
            "code": 200,
            "message": "获取编辑中的任务实例成功",
            "data": instance
        }
    except ValueError as e:
        return {
            "code": 404,
            "message": str(e),
            "data": None
        }
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"获取编辑中的任务实例失败: {str(e)}")