112 lines
4 KiB
Python
112 lines
4 KiB
Python
#!/usr/local/bin/python3.5
|
|
|
|
import discord
|
|
import traceback
|
|
import logging
|
|
import datetime
|
|
import pendulum
|
|
from discord.ext import commands
|
|
from cogs.utils import config
|
|
|
|
extensions = ['cogs.interaction',
|
|
'cogs.core',
|
|
'cogs.mod',
|
|
'cogs.owner',
|
|
'cogs.stats',
|
|
'cogs.playlist',
|
|
'cogs.twitch',
|
|
'cogs.overwatch',
|
|
'cogs.links',
|
|
'cogs.tags',
|
|
'cogs.roles',
|
|
'cogs.statsupdate']
|
|
|
|
bot = commands.Bot(command_prefix=config.commandPrefix, description=config.botDescription, pm_help=None)
|
|
discord_logger = logging.getLogger('discord')
|
|
discord_logger.setLevel(logging.WARNING)
|
|
|
|
log = logging.getLogger()
|
|
log.setLevel(logging.INFO)
|
|
handler = logging.FileHandler(filename='bonfire.log', encoding='utf-8', mode='a')
|
|
log.addHandler(handler)
|
|
|
|
|
|
# Bot event overrides
|
|
@bot.event
|
|
async def on_ready():
|
|
# Change the status upon connection to the default status
|
|
await bot.change_status(discord.Game(name=config.defaultStatus, type=0))
|
|
channel_id = config.getContent('restart_server')
|
|
if channel_id != 0:
|
|
destination = discord.utils.find(lambda m: m.id == channel_id, bot.get_all_channels())
|
|
await bot.send_message(destination, "I have just finished restarting!")
|
|
config.saveContent('restart_server', 0)
|
|
if not hasattr(bot, 'uptime'):
|
|
bot.uptime = pendulum.utcnow()
|
|
|
|
|
|
@bot.event
|
|
async def on_member_join(member):
|
|
notifications = config.getContent('user_notifications') or {}
|
|
server_notifications = notifications.get(member.server.id)
|
|
if not server_notifications:
|
|
return
|
|
|
|
channel = discord.utils.get(member.server.channels, id=server_notifications)
|
|
await bot.send_message(channel, "Welcome to the '{0.server.name}' server {0.mention}!".format(member))
|
|
|
|
|
|
@bot.event
|
|
async def on_member_remove(member):
|
|
notifications = config.getContent('user_notifications') or {}
|
|
server_notifications = notifications.get(member.server.id)
|
|
if not server_notifications:
|
|
return
|
|
|
|
channel = discord.utils.get(member.server.channels, id=server_notifications)
|
|
await bot.send_message(channel,
|
|
"{0} has left the server, I hope it wasn't because of something I said :c".format(
|
|
member.display_name))
|
|
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
if message.author.bot:
|
|
return
|
|
await bot.process_commands(message)
|
|
|
|
|
|
@bot.event
|
|
async def on_command_error(error, ctx):
|
|
if isinstance(error, commands.BadArgument):
|
|
fmt = "Please provide a valid argument to pass to the command: {}".format(error)
|
|
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):
|
|
fmt = "This command cannot be used in a private message"
|
|
await bot.send_message(ctx.message.channel, fmt)
|
|
elif not isinstance(error, commands.CommandNotFound):
|
|
now = datetime.datetime.now()
|
|
with open("/home/phxntx5/public_html/Bonfire/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)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
for e in extensions:
|
|
bot.load_extension(e)
|
|
bot.run(config.botToken)
|