run.py

""" Entry point for python application. Run this file to start the bot. """

import sys
import discord
import logging
import models

logging.basicConfig(level=logging.INFO, filename='discord.log',
                    filemode='w', format='date+time:%(asctime)s | %(message)s')
models.load_dot_env()

DEBUG_GUILDS = models.get_env_safe(models.ENVs.DEBUG_GUILDS).split(',')
bot = discord.Bot(debug_guilds=DEBUG_GUILDS)
extensions = [
    "test",
    "roles",
]

@bot.event
async def on_ready():
    print(f'\n{bot.user} is ready and online!')
    print(f'Connected to {len(bot.guilds)} Guilds:')
    print('\n'.join([f'  - {guild} ({guild.id})' for guild in bot.guilds]))
    logging.info(f'{bot.user} is ready and online!')
    logging.info(f'Connected to {len(bot.guilds)} Guilds:')
    for guild in bot.guilds:
        logging.info(f'{guild}|{guild.id}')


@bot.slash_command(guild_ids=DEBUG_GUILDS)
async def refresh(ctx):
    reload_extensions()
    await ctx.respond("Refreshed extensions!")


def load_extensions():
    for ext in extensions:
        bot.load_extension(f'cogs.{ext}')


def unload_extensions():
    for ext in extensions:
        bot.unload_extension(f'cogs.{ext}')


def reload_extensions():
    for ext in extensions:
        bot.reload_extension(f'cogs.{ext}')


def main():
    # Whether to deploy to testing bot or main bot (defaults to testing)
    token = models.ENVs.DEBUG_TOKEN
    if len(sys.argv) > 1 and sys.argv[1] == 'deploy':
        token = models.ENVs.TOKEN

    # Run Pycord Bot until keyboard interrupt
    logging.info(f'Starting Pycord Bot...')
    load_extensions()
    bot.run(models.get_env_safe(token, accept_empty=False))
    unload_extensions()

    # Safely shut down connections and save data
    print(f'\nShutting down...')
    logging.info(f'Shutting down...')


if __name__ == '__main__':
    main()