NewUser(BaseFilter):
async def __call__(self, message: Message, session: AsyncSession) -> bool:
print(message)
print(session)
return await session.get(UserModel, message.from_user.id) is not None
https://groosha.2038.io/telegram-tutorial-2/filters-and-middlewares/#filters
А возможно как-то прокинуть сессию БД в фильтр, как она прокидаеться в хендлер через мидлеварь?
Не подскажу сейчас, попробуй
Это нужно сделать как тут? Добавить аргумент session и указать тип?
можно. допустим у тебя мидлварь подмешивает сессию в контекст с именем session, значит у фильтра аналогичным образом можно принять аргументом: class MyFilter(BaseFilter): foo: str async def __call__(self, message, session): result = await session.do_something_with_message(message) return result
Фильтр: class NewUser(BaseFilter): async def __call__(self, message, session): print(message) print(session) Мидлеварь: class DbSessionMiddleware(BaseMiddleware): def __init__(self, session_pool): super().__init__() self.session_pool = session_pool async def __call__( self, handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]], event: TelegramObject, data: Dict[str, Any], ) -> Any: async with self.session_pool() as session: data["session"] = session return await handler(event, data) Хендлер: @router.message(NewUser(), commands=["start"], state="*") Таким образом вылазит ошибка: TypeError: NewUser.call() missing 1 required positional argument: 'session'
покажи как зарегистрирована мидлварь (на какй слой - inner/outer)
Вот так было: dp.message.middleware(DbSessionMiddleware(db_pool))
Если оставить на сообщении то: dp.message.outer_middleware(...) если перенести на апдейт: dp.update.outer_middleware(...) или dp.update.middleware(...) у апдейта можно и на внутренний слой, потому что обработка сообщений выполняется после прохождения через: update outer middlewares update filters update middlewares update handler message outer middlewares message filters message middlewares message handler
Обсуждают сегодня