mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-10-31 01:17:12 +08:00
23 lines
570 B
Python
23 lines
570 B
Python
import motor.motor_asyncio
|
|
from beanie import Document
|
|
from fastapi_users.db import BaseOAuthAccount, BeanieBaseUser, BeanieUserDatabase
|
|
from pydantic import Field
|
|
|
|
DATABASE_URL = "mongodb://localhost:27017"
|
|
client = motor.motor_asyncio.AsyncIOMotorClient(
|
|
DATABASE_URL, uuidRepresentation="standard"
|
|
)
|
|
db = client["database_name"]
|
|
|
|
|
|
class OAuthAccount(BaseOAuthAccount):
|
|
pass
|
|
|
|
|
|
class User(BeanieBaseUser, Document):
|
|
oauth_accounts: list[OAuthAccount] = Field(default_factory=list)
|
|
|
|
|
|
async def get_user_db():
|
|
yield BeanieUserDatabase(User, OAuthAccount)
|