from pydantic import BaseModel


class RoleSnapshotItem(BaseModel):
    site_key: str
    site_display_name: str
    company_key: str
    company_display_name: str
    system_key: str
    system_name: str
    role_key: str
    role_code: str
    role_name: str


class RoleSnapshotResponse(BaseModel):
    user_sub: str
    roles: list[RoleSnapshotItem]