Category rendering, DB columns refined.
This commit is contained in:
@@ -1,12 +1,14 @@
|
||||
from aiogram import BaseMiddleware
|
||||
from typing import Any, Callable
|
||||
|
||||
from services.categories import CategoryService
|
||||
from services.orders import OrderService
|
||||
|
||||
|
||||
class DIMiddleware(BaseMiddleware):
|
||||
def __init__(self, order_service: OrderService):
|
||||
def __init__(self, order_service: OrderService, category_service: CategoryService):
|
||||
self._order_service = order_service
|
||||
self._category_service = category_service
|
||||
|
||||
async def __call__(
|
||||
self,
|
||||
@@ -15,5 +17,6 @@ class DIMiddleware(BaseMiddleware):
|
||||
data: dict[str, Any],
|
||||
) -> Any:
|
||||
data["order_service"] = self._order_service
|
||||
data["category_service"] = self._category_service
|
||||
|
||||
return await handler(event, data)
|
||||
|
||||
@@ -1,9 +1,10 @@
|
||||
from aiogram import BaseMiddleware
|
||||
|
||||
from repositories.orders import OrderRepository
|
||||
from repositories import OrderRepository, CategoriesRepository
|
||||
|
||||
|
||||
class RepositoryMiddleware(BaseMiddleware):
|
||||
async def __call__(self, handler, event, data):
|
||||
data["order_repo"] = OrderRepository()
|
||||
data["categories_repo"] = CategoriesRepository()
|
||||
return await handler(event, data)
|
||||
|
||||
Reference in New Issue
Block a user