mirror of
https://github.com/koloideal/DutyLog.git
synced 2026-06-10 10:25:29 +03:00
94 lines
3.5 KiB
Python
94 lines
3.5 KiB
Python
from collections.abc import AsyncIterable
|
|
|
|
from dishka import Provider, Scope, provide
|
|
from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, async_sessionmaker
|
|
|
|
from dutylog.infrastructure.database.config import create_engine, create_session_maker
|
|
from dutylog.infrastructure.database.dao.users_dao import UsersDAO
|
|
from dutylog.infrastructure.database.dao.hours_transactions_dao import HoursTransactionsDAO
|
|
from dutylog.infrastructure.database.dao.rooms_dao import RoomsDAO
|
|
from dutylog.infrastructure.database.dao.residents_dao import ResidentsDAO
|
|
from dutylog.infrastructure.database.dao.floors_dao import FloorsDAO
|
|
from dutylog.infrastructure.database.repositories.users_repository import UsersRepository
|
|
from dutylog.infrastructure.database.repositories.hours_transactions_repository import HoursTransactionsRepository
|
|
from dutylog.infrastructure.database.repositories.rooms_repository import RoomsRepository
|
|
from dutylog.infrastructure.database.repositories.residents_repository import ResidentsRepository
|
|
from dutylog.infrastructure.database.repositories.floors_repository import FloorsRepository
|
|
from dutylog.infrastructure.utils.config import Config, load_config
|
|
|
|
|
|
class ConfigProvider(Provider):
|
|
@provide(scope=Scope.APP)
|
|
def get_config(self) -> Config:
|
|
return load_config()
|
|
|
|
|
|
class DatabaseProvider(Provider):
|
|
@provide(scope=Scope.APP)
|
|
def get_engine(self, config: Config) -> AsyncEngine:
|
|
return create_engine(config.database.url)
|
|
|
|
@provide(scope=Scope.APP)
|
|
def get_session_maker(self, engine: AsyncEngine) -> async_sessionmaker[AsyncSession]:
|
|
return create_session_maker(engine)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
async def get_session(
|
|
self, session_maker: async_sessionmaker[AsyncSession]
|
|
) -> AsyncIterable[AsyncSession]:
|
|
async with session_maker() as session:
|
|
yield session
|
|
|
|
|
|
class DAOProvider(Provider):
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_users_dao(self, session: AsyncSession) -> UsersDAO:
|
|
return UsersDAO(session)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_hours_transactions_dao(self, session: AsyncSession) -> HoursTransactionsDAO:
|
|
return HoursTransactionsDAO(session)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_rooms_dao(self, session: AsyncSession) -> RoomsDAO:
|
|
return RoomsDAO(session)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_residents_dao(self, session: AsyncSession) -> ResidentsDAO:
|
|
return ResidentsDAO(session)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_floors_dao(self, session: AsyncSession) -> FloorsDAO:
|
|
return FloorsDAO(session)
|
|
|
|
|
|
class RepositoryProvider(Provider):
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_users_repository(self, users_dao: UsersDAO) -> UsersRepository:
|
|
return UsersRepository(users_dao)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_hours_transactions_repository(
|
|
self,
|
|
transactions_dao: HoursTransactionsDAO,
|
|
residents_dao: ResidentsDAO,
|
|
) -> HoursTransactionsRepository:
|
|
return HoursTransactionsRepository(transactions_dao, residents_dao)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_rooms_repository(self, rooms_dao: RoomsDAO) -> RoomsRepository:
|
|
return RoomsRepository(rooms_dao)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_residents_repository(self, residents_dao: ResidentsDAO) -> ResidentsRepository:
|
|
return ResidentsRepository(residents_dao)
|
|
|
|
@provide(scope=Scope.REQUEST)
|
|
def get_floors_repository(self, floors_dao: FloorsDAO) -> FloorsRepository:
|
|
return FloorsRepository(floors_dao)
|
|
|
|
|
|
|
|
|
|
|