add tasks repo update uow and manager
This commit is contained in:
@@ -1,3 +1,4 @@
|
|||||||
|
from src.repository.tasks import TasksRepo
|
||||||
from src.repository.users import UsersRepo
|
from src.repository.users import UsersRepo
|
||||||
|
|
||||||
|
|
||||||
@@ -8,6 +9,7 @@ class DBManager:
|
|||||||
async def __aenter__(self):
|
async def __aenter__(self):
|
||||||
self.session = self.session_factory()
|
self.session = self.session_factory()
|
||||||
self.user = UsersRepo(self.session)
|
self.user = UsersRepo(self.session)
|
||||||
|
self.task = TasksRepo(self.session)
|
||||||
return self
|
return self
|
||||||
|
|
||||||
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||||
|
|||||||
@@ -1,10 +1,12 @@
|
|||||||
from typing import Protocol
|
from typing import Protocol
|
||||||
|
|
||||||
|
from src.repository.tasks import TasksRepo
|
||||||
from src.repository.users import UsersRepo
|
from src.repository.users import UsersRepo
|
||||||
|
|
||||||
|
|
||||||
class IUOWDB(Protocol):
|
class IUOWDB(Protocol):
|
||||||
user: UsersRepo
|
user: UsersRepo
|
||||||
|
task: TasksRepo
|
||||||
|
|
||||||
async def __aenter__(self) -> "IUOWDB": ...
|
async def __aenter__(self) -> "IUOWDB": ...
|
||||||
|
|
||||||
|
|||||||
5
src/repository/tasks.py
Normal file
5
src/repository/tasks.py
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
from src.repository.base import BaseRepo
|
||||||
|
from src.models.tasks import TasksORM
|
||||||
|
|
||||||
|
class TasksRepo(BaseRepo):
|
||||||
|
model = TasksORM
|
||||||
Reference in New Issue
Block a user