add dep user in db, fix typecheck
This commit is contained in:
@@ -1,14 +1,20 @@
|
||||
from typing import Annotated
|
||||
from typing import Annotated, AsyncGenerator
|
||||
|
||||
from fastapi import Depends
|
||||
from fastapi_users_db_sqlalchemy import SQLAlchemyUserDatabase
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from src.db.database import async_session_maker
|
||||
from src.models import UsersORM
|
||||
|
||||
|
||||
async def get_db():
|
||||
async def get_db() -> AsyncGenerator[AsyncSession, None]:
|
||||
async with async_session_maker as db:
|
||||
yield db
|
||||
|
||||
|
||||
DBDep = Annotated[AsyncSession, Depends(get_db)]
|
||||
|
||||
|
||||
async def get_user_db(session: DBDep):
|
||||
yield SQLAlchemyUserDatabase(session, UsersORM)
|
||||
|
||||
Reference in New Issue
Block a user