36 lines
956 B
Python
36 lines
956 B
Python
|
from discord.ext import commands
|
||
|
|
from BotErrors import *
|
||
|
|
from DiscordHelperFunctions import *
|
||
|
|
from Geoffrey import bot_commands
|
||
|
|
|
||
|
|
|
||
|
|
class Admin_Commands:
|
||
|
|
'''
|
||
|
|
Commands for cool people only.
|
||
|
|
'''
|
||
|
|
|
||
|
|
def __init__(self, bot):
|
||
|
|
self.bot = bot
|
||
|
|
|
||
|
|
def check_mod(self, user):
|
||
|
|
try:
|
||
|
|
if ("admin" in [y.name.lower() for y in user.roles]) | \
|
||
|
|
("mod" in [y.name.lower() for y in user.roles]):
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
except AttributeError:
|
||
|
|
raise NotOnServerError
|
||
|
|
|
||
|
|
@commands.command(pass_context=True)
|
||
|
|
async def test(self, ctx):
|
||
|
|
'''
|
||
|
|
Checks if the bot is alive.
|
||
|
|
'''
|
||
|
|
if self.check_mod(ctx.message.author):
|
||
|
|
await self.bot.say('I\'m here you ding dong')
|
||
|
|
else:
|
||
|
|
raise NoPermissionError
|
||
|
|
|
||
|
|
def setup(bot):
|
||
|
|
bot.add_cog(Admin_Commands(bot))
|