from pydantic import BaseModel, ConfigDict, Field

from app.schemas.common import NaiveDatetime


class CreditRatioCreate(BaseModel):
    model_config_id: str = Field(
        ...,
        max_length=32,
        description=(
            "引擎ID，兼容旧字段名。gen_type=image 时填写 image_engines.id；"
            "gen_type=video 时填写 video_engines.id。当前不再绑定 model_configs.id 外键"
        ),
        examples=["engine_xxx"],
    )
    gen_type: str = Field(
        default="video",
        max_length=16,
        description="生成类型：image=图片积分规则，video=视频积分规则",
        examples=["video"],
    )
    resolution: str = Field(
        ...,
        max_length=16,
        description="计费参数。视频为分辨率，例如 480p/720p/1080p；图片为分辨率档位，例如 2K/4K",
        examples=["720p"],
    )
    ratio: float = Field(
        ...,
        gt=0,
        description="积分倍率。最终扣费会乘以该倍率",
        examples=[1.0],
    )
    base_credits: float = Field(
        default=80.0,
        ge=0,
        description="基础积分。视频按 base_credits + per_second_credits * duration 后再乘 ratio；图片按 base_credits * ratio",
        examples=[80.0],
    )
    per_second_credits: float = Field(
        default=2.0,
        ge=0,
        description="视频每秒积分。图片规则通常为 0",
        examples=[2.0],
    )
    input_video_ratio: float = Field(
        default=1.0,
        ge=0,
        description="传入视频积分倍率。视频生成时，用户上传参考视频的额外积分倍率",
        examples=[1.0],
    )
    input_video_base_credits: float = Field(
        default=0.0,
        ge=0,
        description="传入视频基础积分。视频生成时，用户上传参考视频的基础积分",
        examples=[0.0],
    )
    input_video_per_second_credits: float = Field(
        default=0.5,
        ge=0,
        description="传入视频每秒积分。视频生成时，用户上传参考视频每秒消耗的积分",
        examples=[0.5],
    )
    input_image_ratio: float = Field(
        default=1.0,
        ge=0,
        description="传入图片积分倍率。图片生成时，用户上传参考图片的额外积分倍率",
        examples=[1.0],
    )
    input_image_base_credits: float = Field(
        default=0.0,
        ge=0,
        description="传入图片基础积分。图片生成时，用户上传参考图片的基础积分",
        examples=[0.0],
    )
    input_image_per_image_credits: float = Field(
        default=0.0,
        ge=0,
        description="传入图片每张积分。图片生成时，用户上传参考图片每张消耗的积分",
        examples=[0.0],
    )


class CreditRatioOut(CreditRatioCreate):
    id: str = Field(..., description="积分规则ID")
    created_at: NaiveDatetime = Field(..., description="创建时间")

    model_config = ConfigDict(from_attributes=True)