it-volunteers-for-peace/bot.py

#!/usr/bin/python
import logging
import os

import asyncio
from aiogram import Bot, Dispatcher
from aiogram.contrib.fsm_storage.memory import MemoryStorage
import asyncpg

from middlewares.db import DbMiddleware
from handlers.commands import register_command_handlers
from handlers.projects import register_projects_handlers


async def main():
    """Configures and runs the bot, connects to the db"""
    logging.basicConfig(level=logging.INFO)
    DB_URL = os.getenv("DB_URL")
    API_TOKEN = os.getenv("API_TOKEN")

    pool = await asyncpg.create_pool(DB_URL)

    bot = Bot(API_TOKEN)
    dp = Dispatcher(bot, storage=MemoryStorage())

    # Register handlers
    register_command_handlers(dp)
    register_projects_handlers(dp)

    # Register middlewares
    dp.middleware.setup(DbMiddleware(pool))

    try:
        await dp.start_polling()
    finally:
        await dp.storage.close()
        bot_session = await bot.get_session()
        await bot_session.close()


if __name__ == "__main__":
    asyncio.run(main())