129 lines
4.5 KiB
Python
129 lines
4.5 KiB
Python
from discord.ext import commands
|
|
from BotErrors import *
|
|
from BotConfig import bot_config
|
|
from bot import bot_commands
|
|
from discord import Game
|
|
|
|
|
|
def check_mod(user):
|
|
try:
|
|
for role in user.roles:
|
|
if role.id == bot_config.bot_mod:
|
|
return True
|
|
except AttributeError:
|
|
raise NotOnServerError
|
|
|
|
return False
|
|
|
|
|
|
class Admin_Commands:
|
|
'''
|
|
Commands for cool people only.
|
|
'''
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
async def error(self, error, ctx):
|
|
if isinstance(error, PlayerNotFound):
|
|
error_str = 'that player is not in the database.'
|
|
elif isinstance(error, DeleteEntryError):
|
|
error_str = 'that player does not have a location by that name.'
|
|
else:
|
|
error_str = 'the bot encountered the following error: *{}*'.format(error.__str__())
|
|
|
|
await self.bot.send_message(ctx.message.channel, '{}, {}'.format(ctx.message.author.mention, error_str))
|
|
|
|
@commands.command(pass_context=True)
|
|
async def test(self, ctx):
|
|
'''
|
|
Checks if the bot is alive.
|
|
'''
|
|
if check_mod(ctx.message.author):
|
|
await self.bot.say('I\'m here you ding dong')
|
|
else:
|
|
raise NoPermissionError
|
|
|
|
@commands.group(pass_context=True)
|
|
async def mod(self, ctx):
|
|
'''
|
|
Bot moderation tools.
|
|
'''
|
|
if check_mod(ctx.message.author):
|
|
if ctx.invoked_subcommand is None:
|
|
await self.bot.say('{}, invalid sub-command for command **mod**.'.format(ctx.message.author.mention))
|
|
else:
|
|
raise NoPermissionError
|
|
|
|
@mod.command(pass_context=True)
|
|
async def delete(self, ctx, discord_uuid: str, location_name: str):
|
|
'''
|
|
Deletes a location in the database/
|
|
'''
|
|
bot_commands.delete(location_name, discord_uuid=discord_uuid)
|
|
await self.bot.say('{}, **{}** has been deleted.'.format(ctx.message.author.mention, location_name))
|
|
|
|
@delete.error
|
|
async def delete_error(self, error, ctx):
|
|
await self.error(error, ctx)
|
|
|
|
@mod.command(pass_context=True)
|
|
async def edit_name(self, ctx, discord_uuid: str, new_name: str, current_name: str):
|
|
'''
|
|
Edits the name of a location in the database.
|
|
'''
|
|
bot_commands.edit_name(new_name, current_name, discord_uuid=discord_uuid)
|
|
await self.bot.say('{}, **{}** has been rename to **{}**.'.format(ctx.message.author.mention, current_name,
|
|
new_name))
|
|
|
|
@edit_name.error
|
|
async def edit_error(self, error, ctx):
|
|
await self.error(error, ctx)
|
|
|
|
@mod.command(pass_context=True)
|
|
async def update_mc_uuid(self, ctx, discord_uuid: str, mc_uuid: str):
|
|
'''
|
|
Updates a user's MC UUID
|
|
'''
|
|
bot_commands.update_mc_uuid(discord_uuid, mc_uuid)
|
|
await self.bot.say('{}, **{}** has been updated.'.format(ctx.message.author.mention, discord_uuid))
|
|
|
|
@update_mc_uuid.error
|
|
async def update_mc_uuid_error(self, error, ctx):
|
|
await self.error(error, ctx)
|
|
|
|
@mod.command(pass_context=True)
|
|
async def update_discord_uuid(self, ctx, current_discord_uuid: str, new_discord_uuid: str):
|
|
'''
|
|
Updates a user's Discord UUID
|
|
'''
|
|
bot_commands.update_mc_uuid(current_discord_uuid, new_discord_uuid)
|
|
await self.bot.say('{}, user **{}** has been updated.'.format(ctx.message.author.mention, current_discord_uuid))
|
|
|
|
@update_discord_uuid.error
|
|
async def update_discord_uuid_error(self, error, ctx):
|
|
await self.error(error, ctx)
|
|
|
|
@mod.command(pass_context=True)
|
|
async def update_mc_name(self, ctx, discord_uuid: str):
|
|
'''
|
|
Updates a user's MC name to the current name on the MC UUID
|
|
'''
|
|
bot_commands.update_mc_name(discord_uuid)
|
|
await self.bot.say('{}, user **{}**\'s MC name has update.'.format(ctx.message.author.mention, discord_uuid))
|
|
|
|
@update_mc_name.error
|
|
async def update_mc_name_error(self, error, ctx):
|
|
await self.error(error, ctx)
|
|
|
|
@mod.command(pass_context=True)
|
|
async def status(self, ctx, status: str):
|
|
'''
|
|
Updates playing game status of the bot
|
|
'''
|
|
await self.bot.change_presence(game=Game(name=status))
|
|
await self.bot.say('{}, status has been changed'.format(ctx.message.author.mention))
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Admin_Commands(bot)) |