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

from pydantic import BaseModel, ConfigDict, Field


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

    combination_id: int = Field(..., description="杂交组合")
    plot_id: int | None = Field(default=None, description="试验地块")
    planting_date: str | None = Field(default=None, description="定植日期")
    tree_count: int | None = Field(default=None, description="定植株数")
    row_no: int | None = Field(default=None, description="行号")
    col_no: int | None = Field(default=None, description="列号")
    bre_personnel_id: int | None = Field(default=None, description="定植人")
    seedling_id: int | None = Field(default=None, description="育苗批次")
    seed_lot_id: int | None = Field(default=None, description="种子来源批(选择强度链)")
    rootstock_id: int | None = Field(default=None, description="砧木")
    entry_id: int | None = Field(default=None, description="参试条目")
    trial_study_id: int | None = Field(default=None, description="所属试验研究点")
    block_no: int | None = Field(default=None, description="区组号")
    remark: str | None = Field(default=None, description="备注")


class PlantingCreateSchema(PlantingBaseSchema):
    pass


class PlantingUpdateSchema(PlantingBaseSchema):
    combination_id: int | None = Field(default=None, description="杂交组合")
    plot_id: int | None = Field(default=None, description="试验地块")
    planting_date: str | None = Field(default=None, description="定植日期")
    tree_count: int | None = Field(default=None, description="定植株数")
    row_no: int | None = Field(default=None, description="行号")
    col_no: int | None = Field(default=None, description="列号")
    bre_personnel_id: int | None = Field(default=None, description="定植人")
    seedling_id: int | None = Field(default=None, description="育苗批次")
    seed_lot_id: int | None = Field(default=None, description="种子来源批(选择强度链)")
    rootstock_id: int | None = Field(default=None, description="砧木")
    entry_id: int | None = Field(default=None, description="参试条目")
    trial_study_id: int | None = Field(default=None, description="所属试验研究点")
    block_no: int | None = Field(default=None, description="区组号")
    remark: str | None = Field(default=None, description="备注")


class PlantingOutSchema(PlantingBaseSchema):
    id: int
    uuid: str
    combination_name: str | None = None  # 由 Service 层联表填充
    plot_name: str | None = None  # 由 Service 层联表填充
    planting_date: str | None = None
    tree_count: int | None = None
    row_no: int | None = None
    col_no: int | None = None
    bre_personnel_name: str | None = None  # 由 Service 层联表填充
    seedling_name: str | None = None  # 由 Service 层联表填充
    lot_code: str | None = None  # 由 Service 层联表填充（种子来源批号）
    rootstock_name: str | None = None  # 由 Service 层联表填充
    entry_number: str | None = None  # 由 Service 层联表填充（参试编号）
    study_name: str | None = None  # 由 Service 层联表填充（研究点名称）
    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 PlantingQueryParam(BaseModel):
    combination_id: int | None = Field(default=None, description="杂交组合", json_schema_extra={"q": "eq"})
    plot_id: int | None = Field(default=None, description="试验地块", json_schema_extra={"q": "eq"})
    bre_personnel_id: int | None = Field(default=None, description="定植人", json_schema_extra={"q": "eq"})
    seedling_id: int | None = Field(default=None, description="育苗批次", json_schema_extra={"q": "eq"})
    seed_lot_id: int | None = Field(default=None, description="种子来源批", json_schema_extra={"q": "eq"})
    rootstock_id: int | None = Field(default=None, description="砧木", json_schema_extra={"q": "eq"})
    entry_id: int | None = Field(default=None, description="参试条目", json_schema_extra={"q": "eq"})
    trial_study_id: int | None = Field(default=None, description="所属试验研究点", json_schema_extra={"q": "eq"})
    block_no: int | None = Field(default=None, description="区组号", json_schema_extra={"q": "eq"})