async def sqlalchemy_user_db() -> AsyncGenerator[SQLAlchemyUserDatabase, None]:
Base: DeclarativeMeta = declarative_base()
class User(SQLAlchemyBaseUserTable, Base):
pass
DATABASE_URL = "sqlite:///./test.db"
database = Database(DATABASE_URL)
engine = sqlalchemy.create_engine(
DATABASE_URL, connect_args={"check_same_thread": False})
Base.metadata.create_all(engine)
await database.connect()
yield SQLAlchemyUserDatabase(database, User.__table__)
Base.metadata.drop_all(engine)
async def sqlalchemy_user_db_oauth(
) -> AsyncGenerator[SQLAlchemyUserDatabase, None]:
Base: DeclarativeMeta = declarative_base()
class User(SQLAlchemyBaseUserTable, Base):
first_name = Column(String, nullable=True)
class OAuthAccount(SQLAlchemyBaseOAuthAccountTable, Base):
pass
DATABASE_URL = "sqlite:///./test-sqlalchemy-user-oauth.db"
database = Database(DATABASE_URL)
engine = sqlalchemy.create_engine(
DATABASE_URL, connect_args={"check_same_thread": False})
Base.metadata.create_all(engine)
await database.connect()
yield SQLAlchemyUserDatabase(UserDBOAuth, database, User.__table__,
OAuthAccount.__table__)
Base.metadata.drop_all(engine)