diff --git a/telegram/__init__.py b/telegram/__init__.py index 1c9f102..28194e6 100644 --- a/telegram/__init__.py +++ b/telegram/__init__.py @@ -4,16 +4,14 @@ from aiogram.contrib.middlewares.logging import LoggingMiddleware from config import Config +from .handlers import register_handlers + config = Config() telegram_bot = Bot(token=config.teletoken) telegram_dispatcher = Dispatcher(telegram_bot) telegram_dispatcher.middleware.setup(LoggingMiddleware()) - - -@telegram_dispatcher.message_handler(commands=['test']) -async def test_telegram(message): - return SendMessage(message.chat.id, 'passed') +register_handlers(telegram_dispatcher) __all__ = ('telegram_bot', 'telegram_dispatcher') \ No newline at end of file diff --git a/telegram/handlers.py b/telegram/handlers.py new file mode 100644 index 0000000..e58350d --- /dev/null +++ b/telegram/handlers.py @@ -0,0 +1,11 @@ +from aiogram import Dispatcher +from aiogram.dispatcher.webhook import SendMessage +from aiogram.types import Message + + +async def telegram_test(message: Message) -> SendMessage: + return SendMessage(message.chat.id, 'passed') + + +def register_handlers(dispatcher: Dispatcher): + dispatcher.register_message_handler(telegram_test, commands=['test'])