bot.py

from discord.ext.commands import Bot
from settings import Setting
from datetime import datetime


class AnnivBot(Bot):
    def __init__(self, conf: Setting):
        self.conf = conf.conf

        self._a_bot = self.conf.get("Bot", {})
        self._a_token = self._a_bot.get("token")
        self._a_name = self._a_bot.get("name", self.__class__.__name__)
        self._a_command_prefix = self._a_bot.get("prefix", ";")
        self._a_command_prefix += " " if self._a_command_prefix[-1] != " " else ""

        self._a_event = self.conf.get("Event", {})
        self._a_migung = self.conf.get("MigungGame", {})

        self.boot_time = datetime.now()

        self._whitelist = {
            "channels": self._a_bot.get("whitelist_channels", []),
            "guilds": self._a_bot.get("whitelist_guilds", []),
            }

        self.whitelist = {
            "channels": [],
            "guilds": [],
            }

        super().__init__(command_prefix=self._a_command_prefix)

    # Property for prior structure support (Rewriting)
    @property
    def conf_bot(self):
        return self._a_bot

    # Property for prior structure support (Rewriting)
    @property
    def conf_event(self):
        return self._a_event

    @property
    def conf_migung(self):
        return self._a_migung

    # Property for prior structure support (Rewriting)
    @property
    def name(self):
        return self._a_name

    def kick(self):
        self.run(self._a_token)

    async def on_ready(self):
        if not isinstance(self._whitelist["guilds"], list):
            self._whitelist["guilds"] = [self._whitelist["guilds"]]

        if not isinstance(self._whitelist["channels"], list):
            self._whitelist["channels"] = [self._whitelist["channels"]]

        for g in self._whitelist["guilds"]:
            guild = self.get_guild(g)
            if guild:
                self.whitelist["guilds"].append(guild)

        for c in self._whitelist["channels"]:
            channel = self.get_channel(c)
            if channel:
                self.whitelist["channels"].append(channel)

    async def on_message(self, message):
        if message.guild.id in self._whitelist["guilds"] or \
                message.channel.id in self._whitelist["channels"]:
            await self.process_commands(message)