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()

    def belongs_to(self, firm):
        for f in self.firms:
            if f.instance == firm.instance and f.firm == firm.firm :
                return True
        return False

class UserUpdateSchema(BaseUserUpdate):
    pass