admin.py

import discord
from discord.ext import commands
import bb_config
import aiohttp, asyncio

# Self made check since is_owner() doesnt appear to be working and includes server owner
# For Myself and the Server Owner
def is_super(ctx):
	return (ctx.message.author.id == bb_config.owner_id) or (ctx.message.author == ctx.message.guild.owner) or (discord.utils.get(ctx.message.author.roles, name=bb_config.super_role) != None)

# Async method to load the bytes of a file and return bytes
async def downloadBytes(session: aiohttp.ClientSession, url: str):
	async with session.get(url) as response:
		assert response.status == 200
		return await response.read()

# Helper Method to search for a user
def search_user(bot, who):
    try:
        # If Mention or ID is passed (approach the substring from the right, since it can include !@ or just @)
        if (who[:1] == "<"):
            x = bot.get_user(int(who[-19:-1]))
            return x
        else :
            x = bot.get_user(int(who))
            return x
    except:
        # If Name or Nick is passed
        for x in bot.get_all_members():
            if (x.name == who) or (x.nick == who):
                return x
        return None

class AdminCog(commands.Cog):
	'''Majora/Server Owner Only stuff'''

	def __init__(self, bot):
		self.bot = bot

	@commands.command(name="say", aliases=['echo'], hidden=True)
	@commands.check(is_super)
	async def echo(self, ctx, *, message: str):
		await ctx.send(message)
		print(message)

	@commands.command(name="exit", hidden=True)
	@commands.check(is_super)
	async def exit_bot(self, ctx):
		await ctx.send("Shutting down...")
		exit()

	# Cog Management
	@commands.command(name='modules', aliases=['extensions'], hidden=True)
	@commands.check(is_super)
	async def list_extensions(self, ctx):
		"""Command which Lists loaded Modules."""
		tmp = "```\n"
		for item in self.bot.extensions:
			tmp += item + "\n"
		tmp += "```"
		await ctx.send(tmp)

	@commands.command(name='load', hidden=True)
	@commands.check(is_super)
	async def extension_load(self, ctx, *, cog: str):
		"""Command which Loads a Module. Remember to use dot path. e.g: cogs.owner"""
		try:
			await self.bot.load_extension(cog)
		except Exception as e:
			await ctx.send('**`ERROR: %s`**' % e)
		else:
			await ctx.send('**`SUCCESS`**')

	@commands.command(name='unload', hidden=True)
	@commands.check(is_super)
	async def extension_unload(self, ctx, *, cog: str):
		"""Command which Unloads a Module. Remember to use dot path. e.g: cogs.owner"""
		try:
			await self.bot.unload_extension(cog)
		except Exception as e:
			await ctx.send('**`ERROR: %s`**' % e)
		else:
			await ctx.send('**`SUCCESS`**')

	@commands.command(name='reload', hidden=True)
	@commands.check(is_super)
	async def extension_reload(self, ctx, *, cog: str):
		"""Command which Reloads a Module. Remember to use dot path. e.g: cogs.owner"""
		try:
			await self.bot.unload_extension(cog)
			await self.bot.load_extension(cog)
		except Exception as e:
			await ctx.send('**`ERROR: %s`**' % e)
		else:
			await ctx.send('**`SUCCESS`**')

	# Server Channel Specific
	@commands.command(name='yell', hidden=True)
	@commands.check(is_super)
	async def yell(self, ctx, channel_id, *, message: str):
		channel = self.bot.get_channel(int(channel_id))
		await channel.send(message)

	# User Specific
	@commands.command(name='tell', hidden=True)
	@commands.check(is_super)
	async def tell(self, ctx, who, *, message: str):
		user = search_user(self.bot, who)
		if user.dm_channel == None:
			await user.create_dm()
		await user.dm_channel.send(message)

	@commands.command(aliases=['remove', 'delete'], hidden=True)
	@commands.check(is_super)
	async def purge(self, ctx, num : int):
		if not num > 30:
			messages = [message async for message in ctx.history(limit=num+1)]
			await ctx.channel.delete_messages(messages)
			done_message = await ctx.send("Purged %s messages..." % num)
			await asyncio.sleep(7)
			await done_message.delete()
		else : 
			done_message = await ctx.send("Will not exceed 30")
			await asyncio.sleep(7)
			await done_message.delete()
		
	@commands.command(name='pfp', hidden=True)
	@commands.check(is_super)
	async def update_profile_picture(self, ctx):
		try:
			url = ctx.message.attachments[0].url
			session = aiohttp.ClientSession()
			item = await downloadBytes(session, url)
			await session.close()
			await self.bot.user.edit(avatar=item)
		except Exception as e:
			await ctx.send('**`ERROR: %s`**' % e)
		else:
			await ctx.send('**`SUCCESS`**')

	@commands.command(name='name', hidden=True)
	@commands.check(is_super)
	async def update_profile_name(self, ctx, *, newname: str):
		try:
			await self.bot.user.edit(password=None, username=newname)
		except Exception as e:
			await ctx.send('**`ERROR: %s`**' % e)
		else:
			await ctx.send('**`SUCCESS`**')

	@commands.command(aliases=['about'])
	async def credits(self, ctx):
		'''Show credits.'''
		await ctx.send("`BB created by TheDerpySage.\nHosted on vinny.thederpysage.com.\nQuestions/Concerns? Contact via Discord.\n@TheDerpySage#2049`")

async def setup(bot):
	await bot.add_cog(AdminCog(bot))