from aiogram import BaseMiddleware

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)