aiogram/examples/echo_fastapi_based_bot.py
2022-05-13 22:56:34 +10:00

45 lines
1 KiB
Python

from aiogram import Bot, Dispatcher, types
from fastapi import FastAPI
import logging
import asyncio
API_TOKEN = "<TOKEN>"
logging.basicConfig(level=logging.DEBUG)
class AsyncioBot:
# Initialize bot and dispatcher
def __init__(self) -> None:
self.bot = Bot(token=API_TOKEN)
self.dp = Dispatcher(self.bot)
self.dp.register_message_handler(self.on_message)
async def on_message(self, message: types.Message):
print(message.get_command())
await message.answer(f"Asnwer: {message.text}")
async def run(self):
# await self.dp.start_polling() # will be blocking main :on_start: FastAPI event
asyncio.create_task(self.dp.start_polling())
async def close(self):
bot = self.bot.get_session()
bot.close()
app = FastAPI()
bot = AsyncioBot()
@app.on_event("startup")
async def on_startup():
await bot.run()
@app.get("/")
async def root():
return {"data": "Hello World"}
@app.on_event("shutdown")
async def on_shutdown():
await bot.close() # closing bot without error