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

from pydantic import BaseModel, ConfigDict, Field


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

    combination_id: int = Field(..., description="杂交组合")
    seed_lot_id: int | None = Field(default=None, description="种子批(关联补链)")
    treatment_method: str | None = Field(default=None, description="处理方式")
    treatment_start: str | None = Field(default=None, description="处理开始日期")
    treatment_end: str | None = Field(default=None, description="处理结束日期")
    germination_rate: float | None = Field(default=None, description="发芽率(%)")
    bre_personnel_id: int | None = Field(default=None, description="种子处理人")
    remark: str | None = Field(default=None, description="备注")


class SeedTreatmentCreateSchema(SeedTreatmentBaseSchema):
    pass


class SeedTreatmentUpdateSchema(SeedTreatmentBaseSchema):
    combination_id: int | None = Field(default=None, description="杂交组合")
    seed_lot_id: int | None = Field(default=None, description="种子批(关联补链)")
    treatment_method: str | None = Field(default=None, description="处理方式")
    treatment_start: str | None = Field(default=None, description="处理开始日期")
    treatment_end: str | None = Field(default=None, description="处理结束日期")
    germination_rate: float | None = Field(default=None, description="发芽率(%)")
    bre_personnel_id: int | None = Field(default=None, description="种子处理人")
    remark: str | None = Field(default=None, description="备注")


class SeedTreatmentOutSchema(SeedTreatmentBaseSchema):
    id: int
    uuid: str
    combination_name: str | None = None  # 由 Service 层联表填充
    seed_lot_code: str | None = None  # 由 Service 层联表填充（种子批号）
    treatment_method: str | None = None
    treatment_start: str | None = None
    treatment_end: str | None = None
    germination_rate: float | None = None
    bre_personnel_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 SeedTreatmentQueryParam(BaseModel):
    combination_id: int | None = Field(default=None, description="杂交组合", json_schema_extra={"q": "eq"})
    treatment_method: str | None = Field(default=None, description="处理方式", json_schema_extra={"q": "eq"})
    bre_personnel_id: int | None = Field(default=None, description="种子处理人", json_schema_extra={"q": "eq"})