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 LoginAccountRequest(BaseModel):
    email: EmailStr
    password: str


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


class LoginAccountWrapper(TraumaResponseWrapper[LoginAccountResponse]):
    pass

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