ewandor/roleplay-contract
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

Improving user management and auto-refreshing user firms

Browse Source
This commit is contained in:
ewandor
2025-04-10 01:31:37 +02:00
parent bc059de65b
commit f1fe81a146
13 changed files with 131 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/rpk-api/hub/user/__init__.py
View File

@@ -1,13 +1,17 @@
from beanie import PydanticObjectId
from fastapi_users_db_beanie import BaseOAuthAccount, BeanieUserDatabase, BeanieBaseUserDocument
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(BeanieBaseUserDocument):
class User(BeanieBaseUser, Document):
oauth_accounts: list[OAuthAccount] = Field(default_factory=list)
firms: list[PydanticObjectId] = Field(default_factory=list)
firms: list[FirmRead] = Field(default_factory=list)
class UserDatabase(BeanieUserDatabase):
pass

9
api/rpk-api/hub/user/schemas.py
View File

@@ -1,7 +1,12 @@
from beanie import PydanticObjectId
from fastapi_users.schemas import BaseUser
from fastapi_users.schemas import BaseUser, BaseUserUpdate
from pydantic import Field
from hub.firm import FirmRead
class UserSchema(BaseUser[PydanticObjectId]):
firms: list[PydanticObjectId] = Field()
firms: list[FirmRead] = Field()
class UserUpdateSchema(BaseUserUpdate):
pass