from async_factory_boy.factory.sqlalchemy import AsyncSQLAlchemyFactory
from sqlalchemy.orm import scoped_session

from tests.conftest import async_session_factory

session = scoped_session(async_session_factory)


class FactoryBase(AsyncSQLAlchemyFactory):
    class Meta:
        sqlalchemy_session = session