Update doc for SQLALchemy2 adapter

docs/src/db_sqlalchemy.py

@@ -1,12 +1,14 @@
-import databases
-import sqlalchemy
+from typing import AsyncGenerator
+
+from fastapi import Depends
 from fastapi_users.db import SQLAlchemyBaseUserTable, SQLAlchemyUserDatabase
+from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
 from sqlalchemy.ext.declarative import DeclarativeMeta, declarative_base
+from sqlalchemy.orm import sessionmaker
 
 from .models import UserDB
 
-DATABASE_URL = "sqlite:///./test.db"
-database = databases.Database(DATABASE_URL)
+DATABASE_URL = "sqlite+aiosqlite:///./test.db"
 Base: DeclarativeMeta = declarative_base()
 
 
@@ -14,13 +16,19 @@ class UserTable(Base, SQLAlchemyBaseUserTable):
     pass
 
 
-engine = sqlalchemy.create_engine(
-    DATABASE_URL, connect_args={"check_same_thread": False}
-)
-Base.metadata.create_all(engine)
-
-users = UserTable.__table__
+engine = create_async_engine(DATABASE_URL, connect_args={"check_same_thread": False})
+async_session_maker = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
 
 
-async def get_user_db():
-    yield SQLAlchemyUserDatabase(UserDB, database, users)
+async def create_db_and_tables():
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+
+
+async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
+    async with async_session_maker() as session:
+        yield session
+
+
+async def get_user_db(session: AsyncSession = Depends(get_async_session)):
+    yield SQLAlchemyUserDatabase(UserDB, session, UserTable)