index.py

import os

from utils import default
from utils.data import (
    Bot,
    HelpFormat,
)
from multiprocessing import Process
import discord


def main():
    config = default.get_from_env("CONFIG")
    if config is None:
        exit(3)

    print("機器人登入中 ...")
    intents = discord.Intents.default()
    intents.members = True
    bot = Bot(
        command_prefix=config.prefix,
        prefix=config.prefix,
        command_attrs=dict(hidden=True),
        help_command=HelpFormat(),
        intents=intents
    )

    for file in os.listdir("cogs"):
        if file.endswith(".py"):
            name = file[:-3]
            bot.load_extension(f"cogs.{name}")

    token = os.environ.get("TOKEN")
    if token is None:
        exit(2)
    bot.run(token)


if __name__ == '__main__':
    while True:
        p = Process(target=main)
        p.start()
        p.join()
        if p.exitcode == 100:
            print('準備重啟 ...')
            continue
        else:
            exit(p.exitcode)