mirror of
https://github.com/Balshgit/mosgortrans.git
synced 2025-09-11 13:00:40 +03:00
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
from http import HTTPStatus
|
|
|
|
from aiogram import Bot, Dispatcher
|
|
from aiogram.types import Update
|
|
from aiogram.utils.executor import start_polling, start_webhook
|
|
from aiohttp import web
|
|
from app.core.bot import bot, dispatcher
|
|
from app.core.logger import logger
|
|
from app.core.scheduler import asyncio_schedule
|
|
from app.settings import (
|
|
START_WITH_WEBHOOK,
|
|
WEBAPP_HOST,
|
|
WEBAPP_PORT,
|
|
WEBHOOK_PATH,
|
|
WEBHOOK_URL,
|
|
)
|
|
|
|
|
|
async def on_startup(dp: Dispatcher) -> None:
|
|
await bot.set_webhook(WEBHOOK_URL)
|
|
asyncio_schedule()
|
|
|
|
|
|
async def on_shutdown(dp: Dispatcher) -> None:
|
|
logger.warning('Shutting down..')
|
|
|
|
# Remove webhook (not acceptable in some cases)
|
|
await bot.delete_webhook()
|
|
|
|
# Close DB connection (if used)
|
|
await dp.storage.close()
|
|
await dp.storage.wait_closed()
|
|
|
|
logger.warning('Bye!')
|
|
|
|
|
|
def bot_polling() -> None:
|
|
logger.info("Start bot in polling mode")
|
|
start_polling(
|
|
dispatcher=dispatcher,
|
|
skip_updates=True,
|
|
on_startup=on_startup,
|
|
on_shutdown=on_shutdown,
|
|
)
|
|
|
|
|
|
def bot_webhook() -> None:
|
|
logger.info("Start bot with webhook")
|
|
start_webhook(
|
|
dispatcher=dispatcher,
|
|
webhook_path=WEBHOOK_PATH,
|
|
on_startup=on_startup,
|
|
on_shutdown=on_shutdown,
|
|
skip_updates=True,
|
|
host=WEBAPP_HOST,
|
|
port=WEBAPP_PORT,
|
|
)
|
|
|
|
|
|
async def webhook(request: web.Request) -> web.Response:
|
|
data = await request.json()
|
|
tg_update = Update(**data)
|
|
logger.info(data)
|
|
Dispatcher.set_current(dispatcher)
|
|
Bot.set_current(dispatcher.bot)
|
|
await dispatcher.process_update(tg_update)
|
|
return web.Response(status=HTTPStatus.OK)
|
|
|
|
|
|
async def on_startup_gunicorn(app: web.Application) -> None:
|
|
await bot.set_webhook(WEBHOOK_URL)
|
|
logger.info(f'Webhook set to {WEBHOOK_URL}')
|
|
|
|
|
|
async def create_app() -> web.Application:
|
|
application = web.Application()
|
|
application.router.add_post(WEBHOOK_PATH, webhook)
|
|
application.on_startup.append(on_startup_gunicorn)
|
|
return application
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if START_WITH_WEBHOOK:
|
|
# bot_webhook() # type: ignore
|
|
app = create_app()
|
|
web.run_app(app=app, host=WEBAPP_HOST, port=WEBAPP_PORT)
|
|
else:
|
|
bot_polling()
|