run.py

import asyncio
import sys
import traceback
from concurrent import futures
from blumpkin import QuantumJumpBot, BotState
from lib.config import Configuration


async def start(executor, bot, loop, count=0):
    loop.run_in_executor(executor, bot.process_input, loop)
    try:
        await bot.run()
    except Exception as e:
        # print the caught exception so it doesnt get lost in narnia
        print(e)
        traceback.print_exc(file=sys.stdout)

        bot.state = BotState.EXCEPTION
        await asyncio.sleep(5)
        if bot.botconfig.restart_on_error and count <= bot.botconfig.restart_attempts:
            count += 1
            await start(executor, bot, loop, count)


def load_config():
    try:
        return Configuration("config.toml")
    except FileNotFoundError:
        from lib.config import generate_config, write_config
        generated = generate_config()
        towrite = write_config(generated, "config.toml")
        if towrite:
            return Configuration("config.toml")
        else:
            sys.exit("Couldn't load the configuration")


executor = futures.ThreadPoolExecutor(max_workers=2, )
bot = QuantumJumpBot(load_config())
loop = asyncio.get_event_loop()
loop.run_until_complete(start(executor, bot, loop))