from pydantic import BaseModel, Field


class WxLoginRequest(BaseModel):
    code: str = Field(min_length=1)
    subscribe_audit: bool = False
    subscribe_match: bool = False


class WxLoginResponse(BaseModel):
    access_token: str
    refresh_token: str
    token_type: str
    expires_in: int
    user_id: int
    audit_status: int
    is_new_user: bool


class TokenRefreshRequest(BaseModel):
    refresh_token: str = Field(min_length=1)