From 6477ac05507b8c268e2a8feea011ae185cb903bc Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Tue, 17 Aug 2021 00:34:56 +0300 Subject: [PATCH] Fixed type hints --- aiogram/dispatcher/event/telegram.py | 4 +++- aiogram/dispatcher/router.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/aiogram/dispatcher/event/telegram.py b/aiogram/dispatcher/event/telegram.py index e52013f4..424ffeb3 100644 --- a/aiogram/dispatcher/event/telegram.py +++ b/aiogram/dispatcher/event/telegram.py @@ -148,7 +148,9 @@ class TelegramEventObserver: middleware = functools.partial(m, middleware) return middleware - def wrap_outer_middleware(self, callback: Any, event, data) -> Any: + def wrap_outer_middleware( + self, callback: Any, event: TelegramObject, data: Dict[str, Any] + ) -> Any: wrapped_outer = self._wrap_middleware(self._resolve_middlewares(outer=True), callback) return wrapped_outer(event, data) diff --git a/aiogram/dispatcher/router.py b/aiogram/dispatcher/router.py index 1d18e80e..b776bcdf 100644 --- a/aiogram/dispatcher/router.py +++ b/aiogram/dispatcher/router.py @@ -95,9 +95,9 @@ class Router: kwargs.update(event_router=self) observer = self.observers[update_type] - async def _wrapped(e, **data): + async def _wrapped(telegram_event: TelegramObject, **data: Any) -> Any: return await self._propagate_event( - observer=observer, update_type=update_type, event=e, **data + observer=observer, update_type=update_type, event=telegram_event, **data ) return await observer.wrap_outer_middleware(_wrapped, event=event, data=kwargs)