19 lines
467 B
Python
19 lines
467 B
Python
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
|