bot.py

"""
PO(I)TD

A Discord bot that sends a random OI problem every day
"""

import os
import logging

from discord import Activity, ActivityType
from discord.ext.commands import Bot, when_mentioned_or

from cogs.bot_info import BotInfo
from cogs.problems import Problems
from cogs.analytics import Analytics
from cogs.top_gg import TopGG

logging.basicConfig(level=logging.INFO)

prefix = '$'
if 'BOT_PREFIX' in os.environ:
    prefix = os.environ['BOT_PREFIX']

bot = Bot(
    command_prefix=when_mentioned_or(prefix),
    help_command=None,
    activity=Activity(type=ActivityType.watching, name='you solve problems')
)

bot.add_cog(BotInfo(bot))
bot.add_cog(Problems(bot))

if 'PRISMA_TOKEN' in os.environ:
    bot.add_cog(Analytics(bot))

if 'TOP_GG_TOKEN' in os.environ:
    bot.add_cog(TopGG(bot))

bot.run(os.environ['DISCORD_TOKEN'])