trauma/api/security/schemas.py

from pydantic import BaseModel, EmailStr

from trauma.api.account.dto import AccessToken
from trauma.api.account.model import AccountModel
from trauma.core.wrappers import TraumaResponseWrapper


class RegisterAccountRequest(BaseModel):
    email: EmailStr
    password: str


class RegisterAccountWrapper(TraumaResponseWrapper[AccountModel]):
    pass


class LoginAccountRequest(BaseModel):
    email: EmailStr
    password: str


class LoginAccountResponse(BaseModel):
    accessToken: AccessToken
    account: AccountModel


class LoginAccountWrapper(TraumaResponseWrapper[LoginAccountResponse]):
    pass