Skip to main content

Adds simple SQLModel support to FastAPI

Project description

FastAPI-SQLModel provides a simple integration between FastAPI and SQLModel in your application. It gives access to useful helpers to facilitate the completion of common tasks.


Install and update using pip:

$ pip install fastapi-sqlmodel


Usage inside of a route

from fastapi import FastAPI
from fastapi_sqlmodel import DBSessionMiddleware  # middleware helper
from fastapi_sqlmodel import db  # an object to provide global access to a database session

from app.models import User

app = FastAPI()

app.add_middleware(DBSessionMiddleware, db_url="sqlite://")

# once the middleware is applied, any route can then access the database session
# from the global ``db``

def get_users():
    users = db.session.query(User).all()

    return users

Note that the session object provided by db.session is based on the Python3.7+ ContextVar. This means that each session is linked to the individual request context in which it was created.

Usage outside of a route

Sometimes it is useful to be able to access the database outside the context of a request, such as in scheduled tasks which run in the background:

import pytz
from apscheduler.schedulers.asyncio import AsyncIOScheduler  # other schedulers are available
from fastapi import FastAPI
from fastapi_sqlmodel import db

from app.models import User, UserCount

app = FastAPI()

app.add_middleware(DBSessionMiddleware, db_url="sqlite://")

async def startup_event():
    scheduler = AsyncIOScheduler(timezone=pytz.utc)
    scheduler.add_job(count_users_task, "cron", hour=0)  # runs every night at midnight

def count_users_task():
    """Count the number of users in the database and save it into the user_counts table."""

    # we are outside of a request context, therefore we cannot rely on ``DBSessionMiddleware``
    # to create a database session for us. Instead, we can use the same ``db`` object and
    # use it as a context manager, like so:

    with db():
        user_count = db.session.query(User).count()


    # no longer able to access a database session once the db() context manager has ended

    return users

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

FastAPI-SQLModel-0.0.1.tar.gz (5.1 kB view hashes)

Uploaded source

Built Distribution

FastAPI_SQLModel-0.0.1-py3-none-any.whl (5.4 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page