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

from pydantic import BaseModel, ConfigDict, Field


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

    trial_name: str = Field(..., description="试验名称")
    plan_id: int | None = Field(default=None, description="所属育种计划 id")
    target_id: int | None = Field(default=None, description="服务育种目标 id")
    trial_type: str | None = Field(default=None, description="试验类型")
    design_type: str | None = Field(default=None, description="设计类型")
    start_year: int | None = Field(default=None, description="起始年份")
    end_year: int | None = Field(default=None, description="结束年份")
    objective: str | None = Field(default=None, description="试验目的")
    remark: str | None = Field(default=None, description="备注")


class TrialCreateSchema(TrialBaseSchema):
    pass


class TrialUpdateSchema(TrialBaseSchema):
    trial_name: str | None = Field(default=None, description="试验名称")
    trial_type: str | None = Field(default=None, description="试验类型")
    design_type: str | None = Field(default=None, description="设计类型")
    start_year: int | None = Field(default=None, description="起始年份")
    end_year: int | None = Field(default=None, description="结束年份")
    objective: str | None = Field(default=None, description="试验目的")
    remark: str | None = Field(default=None, description="备注")


class TrialOutSchema(TrialBaseSchema):
    id: int
    uuid: str
    trial_name: str | None = None
    plan_id: int | None = None
    plan_name: str | None = None
    target_id: int | None = None
    target_name: str | None = None
    trial_type: str | None = None
    design_type: str | None = None
    start_year: int | None = None
    end_year: int | None = None
    objective: 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 TrialQueryParam(BaseModel):
    trial_name: str | None = Field(default=None, description="试验名称", json_schema_extra={"q": "like"})
    trial_type: str | None = Field(default=None, description="试验类型", json_schema_extra={"q": "eq"})
    design_type: str | None = Field(default=None, description="设计类型", json_schema_extra={"q": "eq"})
    start_year: int | None = Field(default=None, description="起始年份", json_schema_extra={"q": "eq"})
    end_year: int | None = Field(default=None, description="结束年份", json_schema_extra={"q": "eq"})