21 lines
469 B
Python
21 lines
469 B
Python
"""FastAPI dependencies."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Annotated
|
|
|
|
from fastapi import Depends, Request
|
|
from sqlalchemy.orm import Session
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Iterator
|
|
|
|
|
|
def get_db(request: Request) -> Iterator[Session]:
|
|
"""Get database session from app state."""
|
|
with Session(request.app.state.engine) as session:
|
|
yield session
|
|
|
|
|
|
DbSession = Annotated[Session, Depends(get_db)]
|