Files
2022-07-18 15:43:54 +02:00

2.5 KiB

SQLAlchemy

FastAPI Users provides the necessary tools to work with SQL databases thanks to SQLAlchemy ORM with asyncio.

Asynchronous driver

To work with your DBMS, you'll need to install the corresponding asyncio driver. The common choices are:

  • For PostgreSQL: pip install asyncpg
  • For SQLite: pip install aiosqlite

Examples of DB_URLs are:

  • PostgreSQL: engine = create_engine('postgresql+asyncpg://user:password@host:port/name')
  • SQLite: engine = create_engine('sqlite+aiosqlite:///name.db')

For the sake of this tutorial from now on, we'll use a simple SQLite database.

Create the User model

As for any SQLAlchemy ORM model, we'll create a User model.

--8<-- "docs/src/db_sqlalchemy.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!

!!! tip "Primary key is defined as UUID" By default, we use UUID as a primary key ID for your user. If you want to use another type, like an auto-incremented integer, you can use SQLAlchemyBaseUserTable as base class and define your own id column.

```py
class User(SQLAlchemyBaseUserTable[int], Base):
    id = Column(Integer, primary_key=True)
```

Notice that `SQLAlchemyBaseUserTable` expects a generic type to define the actual type of ID you use.

Implement a function to create the tables

We'll now create an utility function to create all the defined tables.

--8<-- "docs/src/db_sqlalchemy.py"

This function can be called, for example, during the initialization of your FastAPI app.

!!! warning In production, it's strongly recommended to setup a migration system to update your SQL schemas. See Alembic.

Create the database adapter dependency

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_sqlalchemy.py"

Notice that we define first a get_async_session dependency returning us a fresh SQLAlchemy session to interact with the database.

It's then used inside the get_user_db dependency to generate our adapter. Notice that we pass it two things:

  • The session instance we just injected.
  • The User class, which is the actual SQLAlchemy model.