21 lines
565 B
Python
21 lines
565 B
Python
from beanie import Document
|
|
from fastapi_users_db_beanie import BaseOAuthAccount, BeanieUserDatabase, BeanieBaseUser
|
|
from pydantic import Field
|
|
|
|
from hub.firm import FirmRead
|
|
from hub.user.schemas import UserSchema, UserUpdateSchema
|
|
|
|
|
|
class OAuthAccount(BaseOAuthAccount):
|
|
pass
|
|
|
|
class User(BeanieBaseUser, Document):
|
|
oauth_accounts: list[OAuthAccount] = Field(default_factory=list)
|
|
firms: list[FirmRead] = Field(default_factory=list)
|
|
|
|
class UserDatabase(BeanieUserDatabase):
|
|
pass
|
|
|
|
async def get_user_db():
|
|
yield UserDatabase(User, OAuthAccount)
|