main.py

import asyncio
import traceback
from os import getenv
from typing import Optional

from dotenv import load_dotenv
from tortoise import Tortoise, connections

from utils.client import CryptoBot

load_dotenv(override=True)


client: Optional[CryptoBot] = None


async def init():
    await Tortoise.init(db_url="sqlite://main.db", modules={"models": ["utils.models"]})
    await Tortoise.generate_schemas()

    global client
    client = CryptoBot()
    token = getenv("TOKEN")
    await client.start(token)


if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    try:
        loop.run_until_complete(init())
    except (KeyboardInterrupt, Exception) as e:
        if not isinstance(e, KeyboardInterrupt):
            traceback.print_exc()
        loop.run_until_complete(connections.close_all())
        if client and not client.is_closed():
            loop.run_until_complete(client.close())
    finally:
        if loop.is_running():
            loop.close()