bot.py

import asyncio
import json
import os
from datetime import datetime

import asyncpg
import discord
from discord.ext import commands


class FlandersBOT(commands.AutoShardedBot):
    def __init__(self, intents):
        super().__init__(command_prefix=self.get_default_prefixes, case_insensitive=True, intents=intents)

        # Remove default help command
        self.remove_command('help')

        # Default configuration with cache
        self.cached_screencaps = {}
        self.reminders = []
        self.uptime = datetime.utcnow()
        self.db = None

        # Load config file
        with open('settings/config.json', 'r') as config_file:
            self.config = json.load(config_file)

        # Configure debug mode
        self.debug_mode = self.config['debug_mode']

    async def setup_hook(self):
        # Establish connection to PostgreSQL database
        try:
            self.db = await asyncpg.create_pool(**self.config['db_credentials'])

        except Exception as e:
            exc = f'{type(e).__name__}: {e}'
            print(f'Failed to connect PostgreSQL. Terminating.\n{exc}')
            exit()

        # Load all bot extensions from cogs folder
        for file in os.listdir("cogs"):
            if file.endswith(".py") and not file.startswith('_'):
                extension = file[:-3]

                try:
                    await self.load_extension(f"cogs.{extension}")

                except Exception as e:
                    exc = f'{type(e).__name__}: {e}'
                    print(f'Failed to load extension {extension}\n{exc}')

    # Default get prefixes method, only supports mentions without message content privileges
    async def get_default_prefixes(self, bot, message):
        return commands.when_mentioned(self, message)


# Runs FlandersBOT
def run_bot():
    # Requires members intents for leaderboard username display
    intents = discord.Intents.default()

    # Initialise bot with db connection
    bot = FlandersBOT(intents)

    # Load config file for token
    with open('settings/config.json', 'r') as conf:
        config = json.load(conf)

    # Run FlandersBOT
    bot.run(config['bot_token'])


run_bot()