launcher.py

import asyncio
import asyncpg
import logging
import contextlib
import uvloop
import sys

from bot import Cirilla
import config

# Faster asyncio
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())


@contextlib.contextmanager
def setup_logging():
    logging.getLogger("discord").setLevel(logging.INFO)
    logging.getLogger("discord.http").setLevel(logging.WARNING)

    log = logging.getLogger()
    try:
        # __enter__
        log.setLevel(logging.INFO)
        if config.debugging:
            handler = logging.StreamHandler(sys.stdout)
        else:
            handler = logging.FileHandler(
                filename="cirilla.log", encoding="utf-8", mode="w"
            )
        dt_fmt = "%Y-%m-%d %H:%M:%S"
        fmt = logging.Formatter(
            "[{asctime}] [{levelname:<7}] {name}: {message}", dt_fmt, style="{"
        )
        handler.setFormatter(fmt)
        log.addHandler(handler)

        yield
    finally:
        # __exit__
        handlers = log.handlers[:]
        for hdlr in handlers:
            hdlr.close()
            log.removeHandler(hdlr)


async def run_bot():
    log = logging.getLogger()

    # async with asyncpg.create_pool(**config.db, command_timeout=60) as pool:
    #     bot = Cirilla(pool)
    #     async with bot:
    #         await bot.start(config.token)
    #         print("Bot finished running")
    try:
        pool = await asyncpg.create_pool(**config.db, command_timeout=60)
    except:
        log.exception("Failed to initialize Postgres")
        return
    bot = Cirilla(pool)
    async with bot:
        await bot.start(config.token)
    await pool.close()


async def main():
    with open("cirilla_errors.log", "w") as f:
        with contextlib.redirect_stderr(f):
            with setup_logging():
                await run_bot()


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        pass
    except:
        raise