PDBA/bot-base
8
2
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[roles mod] Implemented

Browse Source
This commit is contained in:
Suwako Moriya 2020-04-05 02:39:46 +02:00
parent 5585984192
commit 626dc50a33
Signed by: SuwakoMmh
GPG Key ID: A27482B806F13CD5
2 changed files with 100 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
main.py
View File

@ -395,17 +395,33 @@ class ClientById:
channel = self.client.get_channel(id_)
return channel.send(*args, **kwargs)
async def get_role(self, id_=None, name=None):
async def get_role(self, id_=None, name=None, guild=None, case_sensitive=True):
guilds = self.client.guilds
if guild is not None:
guilds = [guild]
if id_ is not None:
for guild in self.client.guilds:
for guild in guilds:
role = discord.utils.get(guild.roles, id=id_)
if role:
return role
if name is not None:
for guild in self.client.guilds:
role = discord.utils.get(guild.roles, name=name)
if role:
return role
if case_sensitive:
for guild in guilds:
role = discord.utils.get(guild.roles, name=name)
if role:
return role
else:
name = name.lower()
role = None
for guild in guilds:
for role_ in guild.roles:
if role_.name.lower() == name:
role = role_
break
if role is not None:
break
return role
return None

90
modules/roles/__init__.py
View File

@ -11,31 +11,97 @@ class MainClass(BaseClassPython):
"description": "Modulé gérant l'attribution des roles",
"commands": {
"`{prefix}{command} list`": "Liste les roles",
"`{prefix}{command} <role>`": "S'attribuer le role <role>"
"`{prefix}{command} <role> [role] ...`": "S'attribuer (ou désattribuer) le(s) rôles <role> ([role]...)",
}
}
command_text = "roles"
def __init__(self, client):
super().__init__(client)
self.config.init({"roles": {}})
self.config.init({"guild":297780867286433792,
"roles": {"435559220860157952":"Rôle mentionné lors des jeux."}})
async def com_list(self, message, args, kwargs):
response = discord.Embed(title="Roles disponibles", color=self.color)
for id_ in self.config.roles.keys():
role = await self.client.id.get_role(int(id_))
response.add_field(name=role.name, value=self.config.roles[id_], inline=True)
print(id_,type(id_))
role = message.guild.get_role(int(id_))
if role is not None:
response.add_field(name=role.name, value=f" -> `{self.config.roles[id_]}`", inline=True)
await message.channel.send(embed=response)
async def com_add(self, message, args, kwargs):
guild = self.client.get_guild(self.config.guild)
member = guild.get_member(message.author.id)
if len(args) <= 1:
await message.channel.send("Il manque des arguments à la commande")
for role in args[0:]:
drole = await self.client.id.get_role(name="Invité")
if drole is None or str(drole.id) not in self.config.roles.keys():
await message.channel.send(f"Le role {role} n'est pas disponible.")
for role_ in args[1:]:
role = await self.client.id.get_role(name=role_, guild=guild, case_sensitive=False)
if role is None or str(role.id) not in self.config.roles.keys():
await message.channel.send(f"Le role {role_} n'est pas disponible.")
else:
try:
await message.author.add_roles(drole)
except discord.errors.Forbidden:
await message.channel.send(f"Je n'ai pas la permission d'attribuer le role {role}.")
await self.tryaddrole(message, member, role)
async def com_remove(self, message, args, kwargs):
guild = self.client.get_guild(self.config.guild)
member = guild.get_member(message.author.id)
if len(args) <= 1:
await message.channel.send("Il manque des arguments à la commande")
for role_ in args[1:]:
role = await self.client.id.get_role(name=role_, guild=guild, case_sensitive=False)
if role is None or str(role.id) not in self.config.roles.keys():
await message.channel.send(f"Le role {role_} n'est pas disponible.")
else:
await self.tryremoverole(message, member, role)
async def com_toggle(self, message, args, kwargs):
guild = self.client.get_guild(self.config.guild)
member = guild.get_member(message.author.id)
if len(args) <= 1:
await message.channel.send("Il manque des arguments à la commande")
for role_ in args[1:]:
role = await self.client.id.get_role(name=role_, guild=guild, case_sensitive=False)
if role is None or str(role.id) not in self.config.roles.keys():
await message.channel.send(f"Le role {role_} n'est pas disponible.")
else:
await self.trytogglerole(message, member, role)
async def command(self, message, args, kwargs):
guild = self.client.get_guild(self.config.guild)
member = guild.get_member(message.author.id)
if len(args) < 1:
await message.channel.send("Il manque des arguments à la commande")
for role_ in args:
role = await self.client.id.get_role(name=role_, guild=guild, case_sensitive=False)
if role is None or str(role.id) not in self.config.roles.keys():
await message.channel.send(f"Le role {role_} n'est pas disponible.")
else:
await self.trytogglerole(message, member, role)
async def trytogglerole(self, message, member, role):
if role in member.roles:
await self.tryremoverole(message, member, role)
else:
await self.tryaddrole(message, member, role)
async def tryaddrole(self, message, member, role):
if role in member.roles:
await message.channel.send(f"Vous avez déjà le rôle {role}.")
return
try:
await member.add_roles(role, reason="Auto-attribution")
except discord.errors.Forbidden:
await message.channel.send(f"Je n'ai pas la permission de vous attribuer le rôle {role}.")
else:
await message.channel.send(f"Vous avez reçu le rôle {role}.")
async def tryremoverole(self, message, member, role):
if not role in member.roles:
await message.channel.send(f"Vous n'avez pas le rôle {role}.")
return
try:
await member.remove_roles(role, reason="Auto-désattribution")
except discord.errors.Forbidden:
await message.channel.send(f"Je n'ai pas la permission de vous retirer le rôle {role}.")
else:
await message.channel.send(f"Vous avez perdu le rôle {role}.")