Moved telegram part to telegram package

This commit is contained in:
Alexey Sokolov
2022-02-10 15:56:57 +03:00
parent 80b9d35d8f
commit 4739b8963a
5 changed files with 27 additions and 23 deletions

View File

@@ -1,37 +1,32 @@
import logging
from aiogram import Bot, Dispatcher
from aiogram.dispatcher.webhook import configure_app
from aiogram.contrib.middlewares.logging import LoggingMiddleware
from aiohttp import web
from config import config
from telegram import telegram_bot, telegram_dispatcher
WEBAPP_PORT = 3001
TELEGRAM_WEBHOOK_URL = f'{config.base_url}{config.api_token}'
TELEGRAM_WEBHOOK_URL = f'{config.base_url}/{config.teletoken}'
logging.basicConfig(level=logging.INFO)
bot = Bot(token=config.api_token)
dp = Dispatcher(bot)
dp.middleware.setup(LoggingMiddleware())
async def test_route(_):
return web.json_response({'test': 'passed'}, status=200)
async def on_startup(_):
await bot.set_webhook(TELEGRAM_WEBHOOK_URL)
await telegram_bot.set_webhook(TELEGRAM_WEBHOOK_URL)
async def on_shutdown(_):
logging.warning('Shutting down')
await bot.delete_webhook()
await dp.storage.close()
await dp.storage.wait_closed()
await bot.session.close()
await telegram_bot.delete_webhook()
await telegram_dispatcher.storage.close()
await telegram_dispatcher.storage.wait_closed()
session = await telegram_bot.get_session()
await session.close()
logging.warning('Bye!..')
@@ -39,7 +34,7 @@ app = web.Application()
app.add_routes([web.get('/test', test_route)])
app.on_startup.append(on_startup)
app.on_shutdown.append(on_shutdown)
configure_app(dp, app, '/'+config.api_token)
configure_app(telegram_dispatcher, app, '/'+config.teletoken)
if __name__ == '__main__':
web.run_app(app, port=WEBAPP_PORT)