diff --git a/src/core/db_manager.py b/src/core/db_manager.py index 286db3c..85f6897 100644 --- a/src/core/db_manager.py +++ b/src/core/db_manager.py @@ -1,3 +1,4 @@ +from src.repository.tasks import TasksRepo from src.repository.users import UsersRepo @@ -8,6 +9,7 @@ class DBManager: async def __aenter__(self): self.session = self.session_factory() self.user = UsersRepo(self.session) + self.task = TasksRepo(self.session) return self async def __aexit__(self, exc_type, exc_val, exc_tb): diff --git a/src/core/interfaces.py b/src/core/interfaces.py index d1c4682..d863fe3 100644 --- a/src/core/interfaces.py +++ b/src/core/interfaces.py @@ -1,10 +1,12 @@ from typing import Protocol +from src.repository.tasks import TasksRepo from src.repository.users import UsersRepo class IUOWDB(Protocol): user: UsersRepo + task: TasksRepo async def __aenter__(self) -> "IUOWDB": ... diff --git a/src/repository/tasks.py b/src/repository/tasks.py new file mode 100644 index 0000000..4709aaa --- /dev/null +++ b/src/repository/tasks.py @@ -0,0 +1,5 @@ +from src.repository.base import BaseRepo +from src.models.tasks import TasksORM + +class TasksRepo(BaseRepo): + model = TasksORM