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

from pydantic import BaseModel, ConfigDict, Field


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

    plan_code: str = Field(..., description="计划编号")
    plan_name: str = Field(..., description="计划名称")
    objective: str = Field(default=None, description="计划目标")
    start_year: int = Field(default=None, description="起始年份")
    end_year: int = Field(default=None, description="结束年份")
    leader: str = Field(default=None, description="负责人")
    remark: str = Field(default=None, description="备注")


class PlanCreateSchema(PlanBaseSchema):
    pass


class PlanUpdateSchema(PlanBaseSchema):
    plan_code: str | None = Field(default=None, description="计划编号")
    plan_name: str | None = Field(default=None, description="计划名称")
    objective: str | None = Field(default=None, description="计划目标")
    start_year: int | None = Field(default=None, description="起始年份")
    end_year: int | None = Field(default=None, description="结束年份")
    leader: str | None = Field(default=None, description="负责人")
    remark: str | None = Field(default=None, description="备注")


class PlanOutSchema(PlanBaseSchema):
    id: int
    uuid: str
    plan_code: str | None = None
    plan_name: str | None = None
    objective: str | None = None
    start_year: int | None = None
    end_year: int | None = None
    leader: 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 PlanQueryParam(BaseModel):
    plan_code: str | None = Field(default=None, description="计划编号")
    plan_name: str | None = Field(default=None, description="计划名称")
    start_year: int | None = Field(default=None, description="起始年份")
    end_year: int | None = Field(default=None, description="结束年份")

    model_config = ConfigDict(
        extra="ignore",
        json_schema_extra={
            "plan_code": ("like", "{param}"),
            "plan_name": ("like", "{param}"),
            "start_year": ("eq", "{param}"),
            "end_year": ("eq", "{param}"),
        },
    )