mirror of
https://github.com/fastapi-users/fastapi-users.git
synced 2025-08-15 03:04:27 +08:00
25 lines
611 B
Python
25 lines
611 B
Python
from typing import List
|
|
|
|
import motor.motor_asyncio
|
|
from beanie import PydanticObjectId
|
|
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[PydanticObjectId]):
|
|
oauth_accounts: List[OAuthAccount] = Field(default_factory=list)
|
|
|
|
|
|
async def get_user_db():
|
|
yield BeanieUserDatabase(User, OAuthAccount)
|