AgentCoreService/pqai_agent_server/const/status_enum.py

from enum import Enum


class TestTaskStatus(Enum):
    NOT_STARTED = 0
    IN_PROGRESS = 1
    COMPLETED = 2
    CANCELLED = 3

    @property
    def description(self):
        descriptions = {
            self.NOT_STARTED: "未开始",
            self.IN_PROGRESS: "进行中",
            self.COMPLETED: "已完成",
            self.CANCELLED: "已取消"
        }
        return descriptions.get(self)

class TestTaskConversationsStatus(Enum):
    """任务状态枚举类"""
    PENDING = 0  # 待执行
    RUNNING = 1  # 执行中
    SUCCESS = 2  # 执行成功
    FAILED = 3  # 执行失败
    CANCELLED = 4  # 已取消

    @property
    def description(self):
        descriptions = {
            self.PENDING: "待执行",
            self.RUNNING: "执行中",
            self.SUCCESS: "执行成功",
            self.FAILED: "执行失败",
            self.CANCELLED: "已取消"
        }
        return descriptions.get(self)


# 使用示例
def get_test_task_status_desc(status_code):
    try:
        status = TestTaskStatus(status_code)
        return status.description
    except ValueError:
        return f"未知状态: {status_code}"

# 使用示例
def get_test_task_conversations_status_desc(status_code):
    try:
        status = TestTaskConversationsStatus(status_code)
        return status.description
    except ValueError:
        return f"未知状态: {status_code}"