from app.core.base_schema import CommonSchema
"""选择规则 —— Pydantic 校验/序列化模型。"""
from datetime import datetime

from pydantic import BaseModel, ConfigDict, Field


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

    rule_name: str = Field(..., description="规则名称")
    target_id: int = Field(..., description="育种目标")
    stage: str | None = Field(default=None, description="适用阶段")
    conditions_json: str = Field(..., description="条件(JSON)")
    logic: str | None = Field(default=None, description="逻辑关系")
    action: str | None = Field(default=None, description="动作")
    priority: int | None = Field(default=None, description="优先级")
    enabled: str | None = Field(default=None, description="是否启用")
    remark: str | None = Field(default=None, description="备注")


class SelectionRuleCreateSchema(SelectionRuleBaseSchema):
    pass


class SelectionRuleUpdateSchema(SelectionRuleBaseSchema):
    rule_name: str | None = Field(default=None, description="规则名称")
    target_id: int | None = Field(default=None, description="育种目标")
    stage: str | None = Field(default=None, description="适用阶段")
    conditions_json: str | None = Field(default=None, description="条件(JSON)")
    logic: str | None = Field(default=None, description="逻辑关系")
    action: str | None = Field(default=None, description="动作")
    priority: int | None = Field(default=None, description="优先级")
    enabled: str | None = Field(default=None, description="是否启用")
    remark: str | None = Field(default=None, description="备注")


class SelectionRuleOutSchema(SelectionRuleBaseSchema):
    id: int
    uuid: str
    rule_name: str | None = None
    target_name: str | None = None  # 由 Service 层联表填充
    stage: str | None = None
    conditions_json: str | None = None
    logic: str | None = None
    action: str | None = None
    priority: int | None = None
    enabled: 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 SelectionRuleQueryParam(BaseModel):
    rule_name: str | None = Field(default=None, description="规则名称", json_schema_extra={"q": "like"})
    target_id: int | None = Field(default=None, description="育种目标", json_schema_extra={"q": "eq"})
    stage: str | None = Field(default=None, description="适用阶段", json_schema_extra={"q": "eq"})
    logic: str | None = Field(default=None, description="逻辑关系", json_schema_extra={"q": "eq"})
    action: str | None = Field(default=None, description="动作", json_schema_extra={"q": "eq"})
    enabled: str | None = Field(default=None, description="是否启用", json_schema_extra={"q": "eq"})