alien_py_cloud/alien_contract/schemas/request/contract.py

import re
from typing import Literal

from pydantic import BaseModel, Field, field_validator

BundleType = Literal["STORE_STANDARD", "LAWYER_STANDARD"]


class BundleCreateRequest(BaseModel):
    subject_type: Literal["store", "lawyer"] = Field(description="主体类型")
    subject_id: int = Field(gt=0, description="主体ID")
    subject_name: str = Field(description="主体名称")
    business_segment: str = Field(description="业务板块")
    contact_name: str = Field(description="联系人姓名")
    contact_phone: str = Field(description="联系电话")
    ord_id: str = Field(description="统一社会信用代码")
    bundle_type: BundleType | None = Field(default=None, description="合同包类型，支持 STORE_STANDARD 或 LAWYER_STANDARD")

    @field_validator("contact_phone")
    @classmethod
    def validate_contact_phone(cls, value: str) -> str:
        if not re.fullmatch(r"^1\d{10}$", value):
            raise ValueError("contact_phone 格式错误，应为11位手机号")
        return value

    @field_validator("ord_id")
    @classmethod
    def validate_ord_id(cls, value: str) -> str:
        if not re.fullmatch(r"^[0-9A-Z]{18}$", value):
            raise ValueError("ord_id 格式错误，应为18位大写字母或数字")
        return value