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

from pydantic import BaseModel, ConfigDict, Field


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

    combination_id: int = Field(..., description="来源杂交组合")
    child_code: str = Field(..., description="子代种质编号")
    dam_id: int | None = Field(default=None, description="母本种质(规格 §3.18)")
    sire_id: int | None = Field(default=None, description="父本种质(规格 §3.18)")
    generation: str | None = Field(default=None, description="世代")
    remark: str | None = Field(default=None, description="备注")


class PedigreeCreateSchema(PedigreeBaseSchema):
    pass


class PedigreeUpdateSchema(PedigreeBaseSchema):
    combination_id: int | None = Field(default=None, description="来源杂交组合")
    child_code: str | None = Field(default=None, description="子代种质编号")
    dam_id: int | None = Field(default=None, description="母本种质(规格 §3.18)")
    sire_id: int | None = Field(default=None, description="父本种质(规格 §3.18)")
    generation: str | None = Field(default=None, description="世代")
    remark: str | None = Field(default=None, description="备注")


class PedigreeOutSchema(PedigreeBaseSchema):
    id: int
    uuid: str
    combination_name: str | None = None  # 由 Service 层联表填充
    dam_name: str | None = None          # 母本名称(联表填充)
    sire_name: str | None = None         # 父本名称(联表填充)
    child_code: str | None = None
    generation: 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 PedigreeQueryParam(BaseModel):
    combination_id: int | None = Field(default=None, description="来源杂交组合", json_schema_extra={"q": "eq"})
    child_code: str | None = Field(default=None, description="子代种质编号", json_schema_extra={"q": "like"})
    dam_id: int | None = Field(default=None, description="母本种质", json_schema_extra={"q": "eq"})
    sire_id: int | None = Field(default=None, description="父本种质", json_schema_extra={"q": "eq"})
    generation: str | None = Field(default=None, description="世代", json_schema_extra={"q": "eq"})