From 74574b4b5f5cf89e1d8b03da09b56537eb36c990 Mon Sep 17 00:00:00 2001 From: Forevka <32968153+Forevka@users.noreply.github.com> Date: Sat, 3 Jul 2021 12:25:03 +0000 Subject: [PATCH] static type checking --- aiogram/utils/handlers_in_use.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/aiogram/utils/handlers_in_use.py b/aiogram/utils/handlers_in_use.py index b50e89fa..53e8e63d 100644 --- a/aiogram/utils/handlers_in_use.py +++ b/aiogram/utils/handlers_in_use.py @@ -1,19 +1,21 @@ +from aiogram.dispatcher.router import Router from itertools import chain -from typing import List +from typing import List, cast from aiogram.dispatcher.dispatcher import Dispatcher AIOGRAM_INTERNAL_HANDLERS = ['update', 'error', ] def get_handlers_in_use(dispatcher: Dispatcher, handlers_to_skip: List[str] = AIOGRAM_INTERNAL_HANDLERS) -> List[str]: - handlers_in_use = [] + handlers_in_use: List[str] = [] for router in [dispatcher.sub_routers, dispatcher]: if (isinstance(router, list)): if (router): handlers_in_use.extend(chain(*list(map(get_handlers_in_use, router)))) else: + router = cast(Router, router) for update_name, observer in router.observers.items(): if (observer.handlers and update_name not in [*handlers_to_skip, *handlers_in_use]): handlers_in_use.append(update_name)