flifloo/FurrMula-Bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Rework of reaction message command, add of reaction list command and some fix

Browse source
This commit is contained in:
Ethanell 2019-08-10 21:12:17 +02:00
parent beab3ef0c3
commit 8358da4e53
1 changed files with 132 additions and 66 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

198
bot.py
View file

@ -1,17 +1,40 @@
import shelve import shelve
from discord import NotFound, InvalidArgument, Embed from discord import NotFound, InvalidArgument, HTTPException, Embed
from discord.ext import commands from discord.ext import commands
bot = commands.Bot(command_prefix="%") # Set bot object and command prefix bot = commands.Bot(command_prefix="%") # Set bot object and command prefix
bot.remove_command("help") # Override default help command bot.remove_command("help") # Override default help command
async def find_message(guild, message_id):
"""guild: guild object, messgae_id: (in) message id
Find a message on a guild"""
message = None
for c in guild.text_channels:
try: # Check message id
message = await c.fetch_message(int(message_id))
except NotFound:
pass
else:
break
return message
async def clean_reaction(message, emoji):
"""message: message object, emoji, target emoji
Clean all reaction of a specific message and emoji"""
for r in message.reactions:
if str(r) == emoji:
async for u in r.users():
await r.remove(u)
def set_config(guild, conf): def set_config(guild, conf):
"""guild : guild object, conf: shelve object """guild : guild object, conf: shelve object
Set configuration for server on configuration file""" Set configuration for server on configuration file"""
print(f"Set config for {guild.name}") print(f"Set config for {guild.name}")
conf[str(guild.id)] = dict() # Create default dictionary conf[str(guild.id)] = dict() # Create default dictionary
for opt in [["default_role_id", ""], ["reaction_messages", dict()]]: # Set up each option in configuration file for opt in [["default_role_id", ""], ["reaction", dict()]]: # Set up each option in configuration file
conf[str(guild.id)][opt[0]] = opt[1] conf[str(guild.id)][opt[0]] = opt[1]
@ -39,7 +62,7 @@ async def on_member_join(member):
"""member: member object """member: member object
If available, add a default role to new members of guild""" If available, add a default role to new members of guild"""
with shelve.open("config.conf") as conf: with shelve.open("config.conf") as conf:
if "default_role_id" in conf[str(member.guild.id)] and conf[str(member.guild.id)]["default_role_id"]: if conf[str(member.guild.id)]["default_role_id"]:
role = member.guild.get_role(conf[str(member.guild.id)]["default_role_id"][0]) # Get the role from guild role = member.guild.get_role(conf[str(member.guild.id)]["default_role_id"][0]) # Get the role from guild
await member.add_roles(role) await member.add_roles(role)
@ -69,18 +92,19 @@ async def reaction_role(message_id, guild, emoji, member, state):
with shelve.open("config.conf") as conf: with shelve.open("config.conf") as conf:
# Avoid bot user and check if message and reaction are in configuration # Avoid bot user and check if message and reaction are in configuration
if not member.bot and \ if not member.bot and \
"reaction_messages" in conf[str(guild.id)] and \ "reaction" in conf[str(guild.id)] and \
message_id in conf[str(guild.id)]["reaction_messages"] and \ message_id in conf[str(guild.id)]["reaction"] and \
emoji in conf[str(guild.id)]["reaction_messages"][message_id]: emoji in conf[str(guild.id)]["reaction"][message_id]:
# If member has default new role, give him default role # If member has default new role, give him default role
def_new_role = guild.get_role(conf[str(guild.id)]["default_role_id"][0]) if conf[str(member.guild.id)]["default_role_id"]:
if def_new_role in member.roles: def_new_role = guild.get_role(conf[str(guild.id)]["default_role_id"][0])
def_role = guild.get_role(conf[str(guild.id)]["default_role_id"][1]) if def_new_role in member.roles:
await member.remove_roles(def_new_role) def_role = guild.get_role(conf[str(guild.id)]["default_role_id"][1])
await member.add_roles(def_role) await member.remove_roles(def_new_role)
await member.add_roles(def_role)
role = guild.get_role(conf[str(guild.id)]["reaction_messages"][message_id][emoji]) # Get the target role role = guild.get_role(conf[str(guild.id)]["reaction"][message_id][emoji]) # Get the target role
# State-dependent action # State-dependent action
if state: if state:
@ -94,15 +118,16 @@ async def reaction_role(message_id, guild, emoji, member, state):
async def help_cmd(ctx): async def help_cmd(ctx):
embed = Embed(title="Help", description="", color=0xffff00) embed = Embed(title="Help", description="", color=0xffff00)
embed.add_field(name="Set default role", value="``set_default_roles @new_default @default`` to set new member " embed.add_field(name="Set default role", value="``set_default_roles @new_default @default`` to set new member "
"default role and default role, if any role mentioned this disable " "default role and default role, if no role mentioned this disable "
"the option") "the option")
embed.add_field(name="Default roles", value="``default_role`` show the default roles") embed.add_field(name="Default roles", value="``default_role`` show the default roles")
embed.add_field(name="Reaction message", value="""``reaction_message <action> <message id> <...>`` embed.add_field(name="Reaction", value="""``reaction <action> <...>``
**__actions :__** **__actions :__**
``set <message id>`` to set a reaction message ``add <message id> <reaction emoji> @role`` to add/edit a reaction on message
``unset <message id>`` to unset a reaction message ``remove <message id> <reaction emoji>`` to remove a reaction on message
``add <message id> <reaction emoji> @role`` to add a reaction on set message ``remove-all <message id>`` to remove all reactions of a message""")
``remove <message id> <reaction emoji>`` to remove a reaction on set message""") embed.add_field(name="Reaction list", value="``reaction_list`` show the list of message with role reaction on the "
"guild ")
await ctx.send(embed=embed) await ctx.send(embed=embed)
@ -166,62 +191,65 @@ async def default_role_error(ctx, error):
@bot.command() @bot.command()
@commands.guild_only() @commands.guild_only()
@commands.has_permissions(administrator=True) @commands.has_permissions(administrator=True)
async def reaction_message(ctx, action=None, message_id=None, reaction=None): async def reaction(ctx, action, message_id, emoji=None):
"""ctx: context object, action: (str) command action, message_id: (str) the message id, """ctx: context object, action: (str) command action, message_id: (str) the message id,
reaction: (str) reaction emoji emoji: (str) reaction emoji
Set or unset a reaction message and add reaction emoji link with roles""" Set or unset a reaction message and add reaction emoji link with roles"""
if action not in ["set", "unset", "add", "remove"]: # Check if action is correct if action not in ["add", "remove", "remove-all"]: # Check if action is correct
raise commands.BadArgument("argument") raise commands.BadArgument("argument")
else: else:
try: # Check message id try:
message = await ctx.fetch_message(int(message_id)) message_id = int(message_id)
except (TypeError, ValueError, NotFound): message = await find_message(ctx.guild, message_id)
if not message:
raise ValueError
except ValueError:
raise commands.BadArgument("message id") raise commands.BadArgument("message id")
else:
with shelve.open("config.conf", writeback=True) as conf: with shelve.open("config.conf", writeback=True) as conf:
if action == "set": # Set a reaction message if message.id not in conf[str(ctx.guild.id)]["reaction"]: # Check if not already set
if message.id in conf[str(ctx.guild.id)]["reaction_messages"]: # Check if not already set conf[str(ctx.guild.id)]["reaction"][message.id] = dict() # Add to configuration
raise commands.BadArgument("already set") if action == "add": # Add reaction
else: if len(ctx.message.role_mentions) != 1: # Check if correct mentioned role
conf[str(ctx.guild.id)]["reaction_messages"][message.id] = dict() # Add to configuration raise commands.BadArgument("role")
await ctx.send("Message set :white_check_mark:") else:
elif action == "unset": # Unset a reaction message try: # Add reaction to message and check if given emoji is correct
try: # Check if the message is set when remove await message.add_reaction(emoji)
del conf[str(ctx.guild.id)]["reaction_messages"][message.id] except (InvalidArgument, NotFound, HTTPException):
except KeyError: raise commands.BadArgument("reaction")
raise commands.BadArgument("not set") else: # Add reaction and role to configuration
else: conf[str(ctx.guild.id)]["reaction"][message.id][emoji] =\
await ctx.send("Message remove :wastebasket:") ctx.message.role_mentions[0].id
else: # Add and Remove reaction actions await ctx.send("Reaction add :white_check_mark:")
if message.id not in conf[str(ctx.guild.id)]["reaction_messages"]: # Check if message is set
raise commands.BadArgument("not set") elif action == "remove": # Remove a reaction
else: try: # Remove reaction from message and check if given emoji is correct
if action == "add": # Add reaction await message.remove_reaction(emoji, bot.user)
if len(ctx.message.role_mentions) != 1: # Check if correct mentioned role await clean_reaction(message, emoji)
raise commands.BadArgument("role") except (InvalidArgument, NotFound, HTTPException):
else: raise commands.BadArgument("reaction")
try: # Add reaction to message and check if given emoji is correct else: # Delete reaction from configuration
await message.add_reaction(reaction) del conf[str(ctx.guild.id)]["reaction"][message.id][emoji]
except InvalidArgument: if len(conf[str(ctx.guild.id)]["reaction"][message.id]) == 0: # Clean if no reaction left
raise commands.BadArgument("reaction") del conf[str(ctx.guild.id)]["reaction"][message.id]
else: # Add reaction and role to configuration await ctx.send("Reaction remove :wastebasket:")
conf[str(ctx.guild.id)]["reaction_messages"][message.id][reaction] =\
ctx.message.role_mentions[0].id elif action == "remove-all": # Remove all reactions
await ctx.send("Reaction add :white_check_mark:") for r in conf[str(ctx.guild.id)]["reaction"][message.id]:
if action == "remove": # Remove a reaction try: # Remove all reaction from message
try: # Remove reaction from message and check if given emoji is correct await message.remove_reaction(r, bot.user)
await message.remove_reaction(reaction, bot.user) await clean_reaction(message, r)
except (InvalidArgument, NotFound): except (InvalidArgument, NotFound, HTTPException):
raise commands.BadArgument("reaction") pass
else: # Delete reaction from configuration del conf[str(ctx.guild.id)]["reaction"][message.id]
del conf[str(ctx.guild.id)]["reaction_messages"][message.id][reaction] await ctx.send("All reactions remove :wastebasket:")
await ctx.send("Reaction remove :wastebasket:")
@reaction_message.error @reaction.error
async def reaction_message_error(ctx, error): async def reaction_error(ctx, error):
print(error)
"""ctx: context object, error: raised error """ctx: context object, error: raised error
Manage reaction_message command errors""" Manage reaction command errors"""
err = {"argument": "Invalid arguments !", "message id": "Invalid message id", "already set": "Message already set", err = {"argument": "Invalid arguments !", "message id": "Invalid message id", "already set": "Message already set",
"not set": "Message not set", "role": "Invalid mentioned role", "reaction": "Invalid reaction"} # Database "not set": "Message not set", "role": "Invalid mentioned role", "reaction": "Invalid reaction"} # Database
if str(error) in err: if str(error) in err:
@ -230,6 +258,44 @@ async def reaction_message_error(ctx, error):
await ctx.send("You are missing Administrator permission to run this command ! :no_entry:") await ctx.send("You are missing Administrator permission to run this command ! :no_entry:")
@bot.command()
@commands.guild_only()
@commands.has_permissions(administrator=True)
async def reaction_list(ctx):
"""ctx: context object
Show a list of all message with reaction role on the guild"""
embed = Embed(title="Reaction list", description="", color=0xffff00)
with shelve.open("config.conf", writeback=True) as conf:
for m in conf[str(ctx.guild.id)]["reaction"]: # All message in configuration
message = await find_message(ctx.guild, m)
if not message: # Clean if message not found
del conf[str(ctx.guild.id)]["reaction"][m]
else:
reactions = str()
for r in conf[str(ctx.guild.id)]["reaction"][m]: # All reaction of message
role = ctx.guild.get_role(conf[str(ctx.guild.id)]['reaction'][m][r])
if not role: # Clean if can't get role
await clean_reaction(message, r)
del conf[str(ctx.guild.id)]['reaction'][m][r]
else:
reactions += f"{r} - ``{role.name}``\n"
if len(conf[str(ctx.guild.id)]["reaction"][m]) == 0: # Clean if any reaction left
del conf[str(ctx.guild.id)]["reaction"][m]
else:
embed.add_field(name=f"{m} [{message.channel.name}]", value=reactions, inline=False)
if len(embed.fields) == 0:
embed.add_field(name="Empty", value="No message with reaction on this guild :/")
await ctx.send(embed=embed)
@reaction_list.error
async def reaction_list_error(ctx, error):
"""ctx: context object, error: raised error
Manage reaction_list command errors"""
if isinstance(error, commands.errors.MissingPermissions):
await ctx.send("You are missing Administrator permission to run this command ! :no_entry:")
@bot.command() @bot.command()
@commands.is_owner() @commands.is_owner()
async def shutdown(ctx): async def shutdown(ctx):