GPQBot

1
from pybotx import Bot, IncomingMessage, IncomingMessageHandlerFunc
2

3

4
async def db_session_middleware(
5
    message: IncomingMessage, bot: Bot, call_next: IncomingMessageHandlerFunc
6
) -> None:
7
    session_factory = bot.state.db_session_factory
8

9
    async with session_factory() as db_session:
10
        message.state.db_session = db_session
11

12
        await call_next(message, bot)
13
        await db_session.commit()