aiogram/aiogram/utils/handlers_in_use.py
2021-07-03 12:07:02 +00:00

21 lines
777 B
Python

from itertools import chain
from typing import List
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 = []
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:
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)
return handlers_in_use