import re

from pydantic import BaseModel, Field, field_validator


class LawyerTemplatesCreate(BaseModel):
    lawyer_id: int = Field(gt=0, description="律所ID")
    law_firm_name: str = Field(description="律所名称")
    business_segment: str = Field(description="业务板块")
    contact_name: str = Field(description="联系人姓名")
    contact_phone: str = Field(description="联系电话")
    ord_id: str = Field(description="统一社会信用代码")

    @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