Fixed type hints

This commit is contained in:
Alex Root Junior 2021-08-17 00:34:56 +03:00
parent c3bf6c2670
commit 6477ac0550
2 changed files with 5 additions and 3 deletions

View file

@ -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)

View file

@ -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)