from beanie import PydanticObjectId
from fastapi_users.schemas import BaseUser, BaseUserUpdate
from pydantic import Field

from hub.firm import FirmRead


class UserSchema(BaseUser[PydanticObjectId]):
    firms: list[FirmRead] = Field()

class UserUpdateSchema(BaseUserUpdate):
    pass