1
0
Fork 0
mirror of synced 2024-05-17 19:12:33 +12:00
Bonfire/bot.py

126 lines
4.4 KiB
Python
Raw Normal View History

2016-07-08 01:05:42 +12:00
#!/usr/local/bin/python3.5
2016-07-09 11:26:43 +12:00
import discord
import traceback
2016-07-31 02:59:35 +12:00
import logging
import datetime
import pendulum
import os
os.chdir(os.path.dirname(os.path.realpath(__file__)))
2016-07-09 11:26:43 +12:00
from discord.ext import commands
from cogs.utils import config
2016-08-31 10:33:46 +12:00
opts = {'command_prefix': config.command_prefix,
'description': config.bot_description,
2016-08-31 07:16:24 +12:00
'pm_help': None,
'shard_count': config.shard_count,
'shard_id': config.shard_id,
'command_not_found': ''}
2016-08-31 07:16:24 +12:00
bot = commands.Bot(**opts)
logging.basicConfig(level=logging.INFO, filename='bonfire.log')
2016-07-08 10:10:24 +12:00
2016-07-31 12:20:55 +12:00
2016-07-08 01:05:42 +12:00
@bot.event
async def on_ready():
2016-07-08 10:10:24 +12:00
# Change the status upon connection to the default status
await bot.change_presence(game=discord.Game(name=config.default_status, type=0))
if not hasattr(bot, 'uptime'):
bot.uptime = pendulum.utcnow()
2016-07-10 04:40:27 +12:00
@bot.event
async def on_message(message):
if message.author.bot:
return
await bot.process_commands(message)
2016-07-08 10:10:24 +12:00
2016-09-24 14:38:58 +12:00
@bot.event
async def on_command_completion(command, ctx):
2016-09-29 12:52:33 +13:00
# There's no reason to continue waiting for this to complete, so lets immediately launch this in a new future
bot.loop.create_task(process_command(ctx))
2016-09-24 14:38:58 +12:00
async def process_command(ctx):
2016-09-29 12:52:33 +13:00
author = ctx.message.author
server = ctx.message.server
command = ctx.command
2016-09-29 12:52:33 +13:00
r_filter = {'command': command.qualified_name}
command_usage = await config.get_content('command_usage', r_filter)
if command_usage is None:
command_usage = {'command': command.qualified_name}
2016-09-29 12:52:33 +13:00
else:
command_usage = command_usage[0]
# Add one to the total usage for this command, basing it off 0 to start with (obviously)
total_usage = command_usage.get('total_usage', 0) + 1
command_usage['total_usage'] = total_usage
# Add one to the author's usage for this command
total_member_usage = command_usage.get('member_usage', {})
member_usage = total_member_usage.get(author.id, 0) + 1
total_member_usage[author.id] = member_usage
command_usage['member_usage'] = total_member_usage
# Add one to the server's usage for this command
total_server_usage = command_usage.get('server_usage', {})
server_usage = total_server_usage.get(server.id, 0) + 1
total_server_usage[server.id] = server_usage
command_usage['server_usage'] = total_server_usage
# Save all the changes
if not await config.update_content('command_usage', command_usage, r_filter):
await config.add_content('command_usage', command_usage, r_filter)
@bot.event
async def on_command_error(error, ctx):
2016-09-18 13:29:48 +12:00
if isinstance(error, commands.CommandNotFound):
return
try:
if isinstance(error.original, discord.Forbidden):
return
elif isinstance(error.original, discord.HTTPException) and 'empty message' in str(error.original):
return
except AttributeError:
pass
2016-08-02 01:10:52 +12:00
if isinstance(error, commands.BadArgument):
fmt = "Please provide a valid argument to pass to the command: {}".format(error)
2016-07-12 00:40:03 +12:00
await bot.send_message(ctx.message.channel, fmt)
elif isinstance(error, commands.CheckFailure):
fmt = "You can't tell me what to do!"
await bot.send_message(ctx.message.channel, fmt)
elif isinstance(error, commands.CommandOnCooldown):
m, s = divmod(error.retry_after, 60)
fmt = "This command is on cooldown! Hold your horses! >:c\nTry again in {} minutes and {} seconds" \
.format(round(m), round(s))
await bot.send_message(ctx.message.channel, fmt)
elif isinstance(error, commands.NoPrivateMessage):
fmt = "This command cannot be used in a private message"
await bot.send_message(ctx.message.channel, fmt)
elif isinstance(error, commands.MissingRequiredArgument):
await bot.send_message(ctx.message.channel, error)
else:
2016-08-07 00:16:49 +12:00
now = datetime.datetime.now()
with open("error_log", 'a') as f:
print("In server '{0.message.server}' at {1}\nFull command: `{0.message.content}`".format(ctx, str(now)),
file=f)
try:
traceback.print_tb(error.original.__traceback__, file=f)
print('{0.__class__.__name__}: {0}'.format(error.original), file=f)
except:
traceback.print_tb(error.__traceback__, file=f)
print('{0.__class__.__name__}: {0}'.format(error), file=f)
2016-07-31 12:20:55 +12:00
2016-07-09 13:27:19 +12:00
if __name__ == '__main__':
bot.remove_command('help')
for e in config.extensions:
2016-07-09 13:27:19 +12:00
bot.load_extension(e)
bot.run(config.bot_token)