from app.core.base_schema import CommonSchema
"""试验处理 —— Pydantic 校验/序列化模型。"""
from datetime import datetime

from pydantic import BaseModel, ConfigDict, Field


class TreatmentBaseSchema(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    trial_study_id: int = Field(..., description="所属研究点")
    factor: str = Field(..., description="因子")
    level: str = Field(..., description="水平")
    description: str = Field(default=None, description="描述")
    remark: str = Field(default=None, description="备注")


class TreatmentCreateSchema(TreatmentBaseSchema):
    pass


class TreatmentUpdateSchema(TreatmentBaseSchema):
    trial_study_id: int | None = Field(default=None, description="所属研究点")
    factor: str | None = Field(default=None, description="因子")
    level: str | None = Field(default=None, description="水平")
    description: str | None = Field(default=None, description="描述")
    remark: str | None = Field(default=None, description="备注")


class TreatmentOutSchema(TreatmentBaseSchema):
    id: int
    uuid: str
    trial_study_name: str | None = None  # 由 Service 层联表填充
    factor: str | None = None
    level: str | None = None
    description: str | None = None
    remark: str | None = None
    created_time: datetime | None = None
    updated_time: datetime | None = None
    created_by: CommonSchema | None = None
    updated_by: CommonSchema | None = None


class TreatmentQueryParam(BaseModel):
    trial_study_id: int | None = Field(default=None, description="所属研究点")
    factor: str | None = Field(default=None, description="因子")
    level: str | None = Field(default=None, description="水平")

    model_config = ConfigDict(
        extra="ignore",
        json_schema_extra={
            "trial_study_id": ("eq", "{param}"),
            "factor": ("like", "{param}"),
            "level": ("like", "{param}"),
        },
    )