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

from pydantic import BaseModel, ConfigDict, Field


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

    combination_id: int = Field(..., description="杂交组合")
    plot_id: int = Field(default=None, description="试验地块/位置")
    planting_id: int | None = Field(default=None, description="来源定植批次")
    tree_no: str = Field(..., description="单株编号")
    row_orientation: str = Field(default=None, description="行向")
    row_no: int = Field(default=None, description="行数")
    col_no: int = Field(default=None, description="列数")
    planted_date: str = Field(default=None, description="定植时间")
    bre_personnel_id: int = Field(default=None, description="责任人")
    status: str = Field(default=None, description="状态")
    remark: str = Field(default=None, description="备注")


class TreeCreateSchema(TreeBaseSchema):
    pass


class TreeUpdateSchema(TreeBaseSchema):
    combination_id: int | None = Field(default=None, description="杂交组合")
    plot_id: int | None = Field(default=None, description="试验地块/位置")
    planting_id: int | None = Field(default=None, description="来源定植批次")
    tree_no: str | None = Field(default=None, description="单株编号")
    row_orientation: str | None = Field(default=None, description="行向")
    row_no: int | None = Field(default=None, description="行数")
    col_no: int | None = Field(default=None, description="列数")
    planted_date: str | None = Field(default=None, description="定植时间")
    bre_personnel_id: int | None = Field(default=None, description="责任人")
    status: str | None = Field(default=None, description="状态")
    remark: str | None = Field(default=None, description="备注")


class TreeOutSchema(TreeBaseSchema):
    id: int
    uuid: str
    combination_name: str | None = None  # 由 Service 层联表填充
    plot_name: str | None = None  # 由 Service 层联表填充
    planting_id: int | None = None
    planting_name: str | None = None  # 由 Service 层联表填充（定植批次展示）
    tree_no: str | None = None
    row_orientation: str | None = None
    row_no: int | None = None
    col_no: int | None = None
    planted_date: str | None = None
    bre_personnel_name: str | None = None  # 由 Service 层联表填充
    status: 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 TreeQueryParam(BaseModel):
    combination_id: int | None = Field(default=None, description="杂交组合")
    plot_id: int | None = Field(default=None, description="试验地块/位置")
    planting_id: int | None = Field(default=None, description="来源定植批次")
    tree_no: str | None = Field(default=None, description="单株编号")
    row_orientation: str | None = Field(default=None, description="行向")
    bre_personnel_id: int | None = Field(default=None, description="责任人")
    status: str | None = Field(default=None, description="状态")

    model_config = ConfigDict(
        extra="ignore",
        json_schema_extra={
            "combination_id": ("eq", "{param}"),
            "plot_id": ("eq", "{param}"),
            "planting_id": ("eq", "{param}"),
            "tree_no": ("like", "{param}"),
            "row_orientation": ("eq", "{param}"),
            "bre_personnel_id": ("eq", "{param}"),
            "status": ("eq", "{param}"),
        },
    )