from pydantic import BaseModel, validator


class PasswordSchema(BaseModel):
    user_id: int
    code: int
    password: str

    @validator('password')
    def password_must_not_be_empty(cls, v):
        if not v:
            raise ValueError("Password cannot be empty")
        return v