commit

src/dutylog/infrastructure/ioc.py

@@ -8,10 +8,12 @@ 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
 
 
@@ -55,6 +57,10 @@ class DAOProvider(Provider):
     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)
@@ -77,6 +83,11 @@ class RepositoryProvider(Provider):
     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)
+
+