aixing_bot.py

# aixing_bot.py
# A discord bot created by Mikey San
# This is mostly a tutorial project for use on my discord server.

import os
import logging
import datetime
from itertools import cycle

import discord
from discord.ext import commands, tasks
from discord.ext.commands import Context, CommandError

from dotenv import load_dotenv
load_dotenv()

TOKEN = os.getenv('DISCORD_TOKEN')
GUILD = os.getenv('DISCORD_GUILD')


# Setup logging to a file called discord.log.
logger = logging.getLogger('discord')
logger.setLevel(logging.DEBUG)
handler = logging.FileHandler(filename='discord.log', encoding='utf-8', mode='w')
handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s: %(message)s'))
logger.addHandler(handler)


# We are using the Bot API to interact with discord.
# Assign "bot" to commands.Bot and set the commnd prefix to look for.
# We have also set our commands to be case insensitive. this means $help or
# $Help or even $helP will trigger the bot.
bot = commands.Bot(command_prefix='$',
                   description="A support Bot for NLB Clan", 
                   case_insensitive=True)


# Create a cycle of status changes.
# This doesn't do much. It's just fun to have; something to play with in future
status = cycle(['hating on D2', 'Thinking...', 'bathroom break', 'dumping on Apex', '$help'])

# Create an event that takes the on_ready function
# This will do a few things once our bot goes live
@bot.event
async def on_ready():
    '''
        Description: Gives the status of aixingBot when it becomes ready
        and loads a footer block with additional notes and URL to gitHub
    '''
    change_status.start()
    print("Bot is ready.")

    # Check that we are in the expected server.
    for guild in bot.guilds:
        if guild.name == GUILD:
            break

        # Print to terminal (log file) when we make a connection.
        # Also confirm the server name and ID we're connected to.
        print(
            f'{bot.user} is connected to the following guild:\n'
            f'{guild.name}(id: {guild.id})'
        )
        # Send a message to the channel "chat" once we are connected.
        # So we can see that we are live there too.
        channel = discord.utils.get(guild.channels, name="chat")
        # wave = ":wave:"


        # Create a discord embed instance.
        # Set title, colour and timestamp. ps. don't forget to import datetime module
        embed = discord.Embed(
            title = f"{bot.user.name} Online!",
            colour = discord.Colour.from_rgb(255,191,0),
            url = "https://github.com/mikeysan/aixingBot",
            timestamp = datetime.datetime.now(datetime.timezone.utc)
        )
        # Set a footer using the embed instance.
        embed.set_footer(
            text = "I am Open Source. I am Skynet."
        )
        # Send our embeded content to the channel.
        await channel.send(embed = embed)


async def on_command_error(self, ctx: Context, exception: CommandError) -> None:
        """Fired when exception happens."""
        logger.error(
            "Exception happened while executing command",
            exc_info=(type(exception), exception, exception.__traceback__)
        )


# Change status task
@tasks.loop(hours=2)
async def change_status():
        # Let's pretend the bot is playing the game of $help
        game = discord.Game(next(status))
        await bot.change_presence(status=discord.Status.idle, activity = game)


# Reload cogs
@bot.command()
@commands.is_owner()
async def reload(ctx, cog):
    '''
        Description: Reloads all Cog files
    '''
    try:
        bot.unload_extension(f"cogs.{cog}")
        bot.load_extension(f"cogs.{cog}")
        ctx.send(f"{cog} reloaded successfully")
    except Exception as e:
        print(f"{cog} can not be loaded:")
        raise e

# Load cogs
cogPath = "./cogs/"
for cogFile in os.listdir(cogPath):
    if cogFile.endswith(".py"):
        try:
            cogFile = f"cogs.{cogFile.replace('.py', '')}"
            bot.load_extension(cogFile)
        except Exception as e:
            print(f"{cogFile} can not be loaded:")
            raise e

# Finally, authenticate with discord and let's get cracking.
bot.run(TOKEN)