db.py

from typing import Callable, Any, Awaitable, Dict

from aiogram import BaseMiddleware
from aiogram.types import CallbackQuery

from bot.data.config import Config
from bot.models.base import create_poll


class DataBaseMiddleware(BaseMiddleware):
    def __init__(self, config: Config) -> None:
        self.sessionmaker = create_poll(
            user=config.db.user,
            password=config.db.password,
            host=config.db.host,
            name=config.db.name
        )

    async def __call__(
        self,
        handler: Callable[[CallbackQuery, Dict[str, Any]], Awaitable[Any]],
        event: CallbackQuery,
        data: Dict[str, Any]
    ) -> Any:

        async with self.sessionmaker() as session:
            data['db'] = session
            return await handler(event, data)