"""基础档案待确认分类的批量补充。"""

from typing import Literal

from pydantic import BaseModel, ConfigDict, Field, field_validator


class CategoryBatch(BaseModel):
    model_config = ConfigDict(extra="forbid")
    ids: list[int] = Field(min_length=1, max_length=200)

    @field_validator("ids")
    @classmethod
    def validate_ids(cls, value):
        if any(item < 1 for item in value) or len(value) != len(set(value)):
            raise ValueError("档案 ID 必须为不重复的正整数")
        return value


class CustomerCategoryBatch(CategoryBatch):
    category: Literal["personal", "manager_personal", "enterprise"]


class ProductCategoryBatch(CategoryBatch):
    category: Literal["municipal_water", "rail_transit"]


class CustomerOwnershipBatch(CategoryBatch):
    new_owner_personnel_id: int = Field(ge=1)
    reason: str | None = Field(default=None, max_length=255)