update_status.py

from common import *
from utils.check_channel_access import access_check

# change_presence functions
async def game_func(name):
  await bot.change_presence(activity=discord.Game(name))

async def listen_func(name):
  await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.listening, name=name))

async def watch_func(name):
  await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.watching, name=name))

change_presence_funcs = {
  "playing": game_func,
  "listening": listen_func,
  "watching": watch_func
}

change_presence_prefixes = {
  "playing": "Playing",
  "listening": "Listening to",
  "watching": "Watching"
}

@bot.command()
@commands.check(access_check)
async def update_status(ctx, *, arg):
  """
  This function is the main entrypoint of the !update_status command
  The user must provide the `<type>` of status:
    * playing
    * listening
    * watching
  The remainder of the message will be used for the status text
  """
  logger.info(f"{Fore.LIGHTGREEN_EX}Updating Status! Requested by {Style.BRIGHT}{ctx.author.display_name}{Fore.RESET}")
  argument_list = arg.split()

  if len(argument_list) < 2:
    await ctx.reply(embed=discord.Embed(
      title="Usage:",
      description="`!update_status [playing|listening|watching] <status>`",
      color=discord.Color.blue()
    ))
    return
  else:
    type = argument_list[0]
    status = ctx.message.content.replace(f"!update_status {type} ", "")

    if type not in ['playing', 'listening', 'watching']:
      await ctx.reply(embed=discord.Embed(
        title="Invalid <type> Provided",
        description="Must provide one of: `playing`, `listening`, or `watching`",
        color=discord.Color.red()
      ))
      logger.info(f"{Fore.RED}Unable to update status. Invalid type for status request: {Style.BRIGHT}{type}{Fore.RESET}")
      return

    await change_presence_funcs[type](status)
    await ctx.reply(embed=discord.Embed(
      title="Status Updated Successfully!",
      color=discord.Color.green()
    ))

    logger.info(f"{Fore.CYAN}Status updated to: {Style.BRIGHT}{change_presence_prefixes[type]} {status}{Fore.RESET}")