roxbot/cogs/selfassign.py

import discord
from discord.ext import commands

import load_config
from config.server_config import ServerConfig


class SelfAssign():
	def __init__(self, Bot):
		self.bot = Bot
		self.con = ServerConfig()
		self.servers = self.con.servers
		self.embed_colour = load_config.embedcolour

	@commands.command(pass_context=True)
	async def listroles(self, ctx):
		"""
		List's all roles that can be self-assigned on this server.
		Usage:
			{command_prefix}listroles
		"""
		if not self.servers[str(ctx.guild.id)]["self_assign"]["enabled"]:
			embed = discord.Embed(colour=discord.Colour(self.embed_colour),
								  description="SelfAssignable roles are not enabled on this server")
			return await ctx.send(embed=embed)
		roles = []
		for role in self.servers[str(ctx.guild.id)]["self_assign"]["roles"]:
			for serverrole in ctx.guild.roles:
				if role == serverrole.id:
					roles.append("**"+serverrole.name+"**")
		roles = '\n'.join(roles)
		embed = discord.Embed(colour=self.embed_colour,
							  description="The self-assignable roles for this server are: \n"+roles)
		return await ctx.send(embed=embed)

	@commands.command(pass_context=True)
	async def iam(self, ctx, *, role: discord.Role = None):
		"""
		Self-assign yourself a role. Only one role at a time.
		Usage:
			{command_prefix}iam [role]
		Example:
			.iam OverwatchPing
		"""
		if not self.servers[str(ctx.guild.id)]["self_assign"]["enabled"]:
			embed = discord.Embed(colour=discord.Colour(self.embed_colour),
								  description="SelfAssignable roles are not enabled on this server")
			return await ctx.send(embed=embed)

		user = ctx.author
		server = ctx.guild

		if role not in server.roles:
			return await ctx.send("That role doesn't exist. Roles are case sensitive. ")

		if role in user.roles:
			return await ctx.send("You already have that role.")

		if role.id in self.servers[str(ctx.guild.id)]["self_assign"]["roles"]:
			await user.add_roles(role, reason="'iam' command triggered.")
			return await ctx.send("Yay {}! You now have the {} role!".format(user.mention, role.name))
		else:
			return await ctx.send("That role is not self-assignable.")

	@commands.command(pass_context=True)
	async def iamn(self, ctx, *, role: discord.Role = None):
		"""
		Remove a self-assigned role
		Usage:
			{command_prefix}iamn [role]
		Example:
			.iamn OverwatchPing
		"""
		if not self.servers[str(ctx.guild.id)]["self_assign"]["enabled"]:
			embed = discord.Embed(colour=discord.Colour(self.embed_colour),
								  description="SelfAssignable roles are not enabled on this server")
			return await ctx.send(embed=embed)

		user = ctx.author
		server = ctx.guild

		if role in user.roles and role.id in self.servers[str(ctx.guild.id)]["self_assign"]["roles"]:
			await user.remove_roles(role, reason="'iamn' command triggered.")
			return await ctx.send("{} has been successfully removed.".format(role.name))
		elif role not in user.roles and role.id in self.servers[str(ctx.guild.id)]["self_assign"]["roles"]:
			return await ctx.send("You do not have {}.".format(role.name))
		else:
			return await ctx.send("That role is not self-assignable.")

	@iam.error
	async def iam_err(self, ctx, error):
		if isinstance(error, commands.BadArgument):
			return await ctx.send("This role doesn't exist. Reminder, roles are case-sensitive.")

	@iamn.error
	async def iamn_err(self, ctx, error):
		if isinstance(error, commands.BadArgument):
			return await ctx.send("This role doesn't exist. Reminder, roles are case-sensitive.")

def setup(Bot):
	Bot.add_cog(SelfAssign(Bot))