bot.py

#!/usr/bin/env python
"""Main bot file."""
import os

import discord

from discord.ext import commands
from dotenv import load_dotenv

from cogs.utils.checks import check_admin
from cogs.utils.utils import get_logger, get_prefix

__version__ = "1.2.0-dev"
DOCS_URL = "placeholder"


class MyBot(commands.Bot):
    """Bot class."""

    def __init__(self, command_prefix, intents, version, test_guild=None):
        """Initialise the bot."""
        super().__init__(command_prefix=command_prefix, intents=intents)
        self.initial_extensions = [
            "cogs.initial",
            "cogs.admin",
            "cogs.any_raids_filter",
            "cogs.fc_filter",
            "cogs.join_name_filter",
            "cogs.misc",
            "cogs.time_channel",
            "cogs.schedules",
        ]
        self.help_command.add_check(check_admin)
        self.my_version = version
        self.test_guild = test_guild
        self.docs = DOCS_URL

    async def setup_hook(self):
        """Load the initial extensions."""
        for ext in self.initial_extensions:
            await self.load_extension(ext)

    async def on_ready(self):
        """Print a message to the console when the bot is ready."""
        logger.info("Bot is ready!")


# LOADS THE .ENV FILE THAT RESIDES ON THE SAME LEVEL AS THE SCRIPT.
load_dotenv()

# GRAB THE API TOKEN FROM THE .ENV FILE.
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
# Set the TEST_GUILD ID
try:
    TEST_GUILD = int(os.getenv("TEST_GUILD"))
    if len(str(TEST_GUILD)) < 17:
        raise ValueError(f"{TEST_GUILD} is not a valid guild ID!")
except Exception:
    TEST_GUILD = None

intents = discord.Intents.default()
intents.members = True
intents.message_content = True

logger = get_logger(logfile="snorlax.log")
logger.info("Starting bot...")

bot = MyBot((get_prefix), intents, __version__, test_guild=TEST_GUILD)
bot.run(DISCORD_TOKEN, log_handler=None)