src/schemas/schemas.py

from pydantic import BaseModel
from enum import Enum
from typing import Optional

class Roles(Enum):
    Engineering = "engineering"
    Employee = "employee"
    Finance = "finance"
    Executives = "executives"
    Marketing = "marketing"
    HR = "hr"

class User(BaseModel):
    id:int
    username:str
    role: Roles

class UserModel(BaseModel):
    status: bool
    user: User
    token: Optional[str] = None