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

    model_config = ConfigDict(
        extra="ignore",
        json_schema_extra={
            "rule_name": ("like", "{param}"),
            "target_id": ("eq", "{param}"),
            "stage": ("eq", "{param}"),
            "logic": ("eq", "{param}"),
            "action": ("eq", "{param}"),
            "enabled": ("eq", "{param}"),
        },
    )