1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
from discord.ext import commands
import discord
import importlib
import utils
importlib.reload(utils)
class RoleManager(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.extension_name = "[Role Manager] "
self.valid_roles = []
self.role_names = []
self.example_role = ''
@commands.bot_has_permissions(embed_links=True)
@commands.command(aliases=["color"], application_command_meta=commands.ApplicationCommandMeta(options=[discord.ApplicationCommandOption(
name='role_name',
description='Choose a role name from the list',
type=discord.ApplicationCommandOptionType.string,
required=False)]))
async def colour(self, ctx: discord.ext.commands.Context, *, role_name=""):
"""Gives users the option to pick a colour for their name"""
await self.get_roles(ctx)
if role_name.casefold() in self.role_names:
for role in ctx.author.roles:
if role.name[1:-1].casefold() in self.role_names:
await ctx.author.remove_roles(role)
guild_role = discord.utils.get(ctx.guild.roles, name=f"[{role_name.lower()}]")
await ctx.author.add_roles(guild_role)
emb = await utils.embed(ctx, "Completed!", "")
await ctx.send(embed=emb)
# Send log message
emb = await utils.notice("Colour changed",
f"[COLOURS] Changed colour of {ctx.author} to {guild_role}",
colour="green")
await utils.log(ctx.guild.id, self.bot, emb=emb)
elif len(self.valid_roles) == 0:
emb = await utils.embed(ctx, "No roles found.",
"Roles are found by naming them with [square] brackets. \n" +
"For example, a role named `[red]` will allow it to be used in this command.")
await ctx.send(embed=emb)
# elif arg.lower().casefold() not in guild_roles:
# emb = await utils.embed(ctx, "Available colours:", f"`{'`, `'.join(guild_roles)}`")
# await ctx.send(embed=emb)
else:
strout = ""
for role in self.valid_roles:
strout += f"\n{role.name[1:-1]} ({role.mention})"
# valid_roles = '\n'.join(guild_roles)
emb = await utils.embed(ctx, "Available colours:", f"{strout}\n\nFor example, `/colour {self.example_role}` will change your colour")
await ctx.send(embed=emb)
self.valid_roles = []
self.role_names = []
self.example_role = ''
async def get_roles(self, ctx):
all_roles = ctx.guild.roles
for role in all_roles:
if role.name.startswith("[") and role.name.endswith("]"):
self.valid_roles.append(role)
self.role_names.append(role.name[1:-1])
self.example_role = role if not self.example_role else False
def setup(bot):
print("INFO: Loading [ColouredRoles]... ", end="")
bot.add_cog(RoleManager(bot))
print("Done!")
def teardown(bot):
print("INFO: Unloading [ColouredRoles]")
|
