* docs(examples): full examples use lifespan instead of on_startup for database initialization * Update beanie.md Replaced the annotations on the notes that I missed
2.4 KiB
Beanie
FastAPI Users provides the necessary tools to work with MongoDB databases using the Beanie ODM.
Setup database connection and collection
The first thing to do is to create a MongoDB connection using mongodb/motor (automatically installed with Beanie).
--8<-- "docs/src/db_beanie.py"
You can choose any name for the database.
Create the User model
As for any Beanie ODM model, we'll create a User model.
--8<-- "docs/src/db_beanie.py"
As you can see, FastAPI Users provides a base class that will include base fields for our User table. You can of course add you own fields there to fit to your needs!
!!! info
The base class is configured to automatically create a unique index on id and email.
!!! info
If you want to add your own custom settings to your User document model - like changing the collection name - don't forget to let your inner Settings class inherit the pre-defined settings from BeanieBaseUser like this: class Settings(BeanieBaseUser.Settings): # ...! See Beanie's documentation on Settings for details.
Create the database adapter
The database adapter of FastAPI Users makes the link between your database configuration and the users logic. It should be generated by a FastAPI dependency.
--8<-- "docs/src/db_beanie.py"
Notice that we pass a reference to the User model we defined above.
Initialize Beanie
When initializing your FastAPI app, it's important that you initialize Beanie so it can discover your models. We can achieve this using Lifespan Events on the FastAPI app:
from contextlib import asynccontextmanager
from beanie import init_beanie
@asynccontextmanager
async def lifespan(app: FastAPI):
await init_beanie(
database=db, # (1)!
document_models=[
User, # (2)!
],
)
yield
app = FastAPI(lifespan=lifespan)
-
This is the
dbMotor database instance we defined above. -
This is the Beanie
Usermodel we defined above. Don't forget to also add your very own models!