EmoteManager/bot.py

#!/usr/bin/env python3
# encoding: utf-8

import logging
import traceback

import discord
from discord.ext import commands

logging.basicConfig(level=logging.WARNING)
logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)

class Bot(commands.AutoShardedBot):
	def __init__(self, **kwargs):
		with open('config.py') as f:
			self.config = eval(f.read(), {})

		super().__init__(
			command_prefix=commands.when_mentioned,
			description=self.config.get('description', ''),
			help_command=commands.MinimalHelpCommand(),
			**kwargs)

		self._setup_success_emojis()

		for cog in self.config['cogs']:
			self.load_extension(cog)

	def _setup_success_emojis(self):
		"""Load the emojis from the config to be used when a command fails or succeeds
		We do it this way so that they can be used anywhere instead of requiring a bot instance.
		"""
		import utils.misc
		default = ('❌', '✅')
		utils.SUCCESS_EMOJIS = utils.misc.SUCCESS_EMOJIS = (
			self.config.get('response_emojis', {}).get('success', default))

	def run(self):
		super().run(self.config['tokens'].pop('discord'))

	async def on_ready(self):
		logger.info('Logged on as {0} (ID: {0.id})'.format(self.user))

	# https://github.com/Rapptz/RoboDanny/blob/ca75fae7de132e55270e53d89bc19dd2958c2ae0/bot.py#L77-L85
	async def on_command_error(self, context, error):
		if isinstance(error, commands.NoPrivateMessage):
			await context.author.send('This command cannot be used in private messages.')
		elif isinstance(error, commands.DisabledCommand):
			message = 'Sorry. This command is disabled and cannot be used.'
			try:
				await context.author.send(message)
			except discord.Forbidden:
				await context.send(message)
		elif isinstance(error, commands.UserInputError):
			await context.send(error)
		elif isinstance(error, commands.NotOwner):
			logger.error('%s tried to run %s but is not the owner', context.author, context.command.name)
		elif isinstance(error, commands.CommandInvokeError):
			await context.send('An internal error occured while trying to run that command.')

			logger.error('In %s:', context.command.qualified_name)
			logger.error(''.join(traceback.format_tb(error.original.__traceback__)))
			# pylint: disable=logging-format-interpolation
			logger.error('{0.__class__.__name__}: {0}'.format(error.original))

bot = Bot()

if __name__ == '__main__':
	bot.run()