# 请求模型定义
from typing import List

from pydantic import BaseModel, model_validator


class AutoDiagInput(BaseModel):
    # 数据索引id
    ids: List[int]
    # 风场编号
    windCode: str
    # 风机编号
    engine_code: str
    # 诊断类型
    autodiagType: str

    @model_validator(mode='before')
    def convert_ids(cls, values):
        """将单个id转换为列表形式"""
        if isinstance(values.get('ids'), int):
            values['ids'] = [values['ids']]
        return values

class DiagnosisResult(BaseModel):
    # 每个id对应的状态码列表
    status_codes: List[int]
    # 所有状态码中的最大值
    max_status: int
    # 状态码0的个数
    count_0: int
    # 状态码1的个数
    count_1: int
    # 状态码2的个数
    count_2: int