import discord from discord.ext import commands from discord import app_commands class ReactionRoles(commands.Cog): def __init__(self, bot): self.bot = bot self.reaction_messages = {} self.role_mapping = { '🟪': 1485181957754060970, '🟧': 1485165429172338769, '🟩': 1490946660036251740, '🟨': 1485535319754408018, '🎉': 1490948733402349690, '📅': 1503366374054170674, '🎁': 1490947061552513106, '🤝': 1490947062798225468, '📦': 1490948680163790990, '✋': 1490947063305736332, '➕': 1503368229391958086 } @app_commands.command(name="reactionrole", description="Create a reaction role message") async def reactionrole(self, interaction: discord.Interaction): embed = discord.Embed( title="🎭Reaction Role", description=( "Choose what roles you would like to get pinged for!\n\n" f"<@&1485181957754060970>\n" f"React to 🟪to receive the <@&1485181957754060970> role!\n\n" f"<@&1485165429172338769>\n" f"React to 🟧to receive the <@&1485165429172338769> role!\n\n" f"<@&1490946660036251740>\n" f"React to 🟩to receive the <@&1490946660036251740> role!\n\n" f"<@&1485535319754408018>\n" f"React to 🟨to receive the <@&1485535319754408018> role!\n\n" f"<@&1503366527850778714>\n" f"React to 🎉to receive the <@&1503366527850778714> role\n\n" f"<@&1503366374054170674>\n" f"React to 📅to receive the <@&1503366374054170674> role!\n\n" f"<@&1490947061552513106>\n" f"React to 🎁to receive the <@&1490947061552513106> role!\n\n" f"<@&1490947062798225468>\n" f"React to 🤝to receive the <@&1490947062798225468> role!\n\n" f"<@&1490948680163790990>\n" f"React to 📦to receive the <@&1490948680163790990> role!\n\n" f"<@&1490947063305736332>\n" f"React to ✋to receive the <@&1490947063305736332> role!\n\n" f"<@&1503368229391958086>\n" f"React to ➕to receive the <@&1503368229391958086> role" ), color=discord.Color.green() ) await interaction.response.send_message(embed=embed) message = await interaction.original_response() self.reaction_messages[message.id] = True for emoji in self.role_mapping.keys(): await message.add_reaction(emoji) @commands.Cog.listener() async def on_raw_reaction_add(self, payload): if payload.user_id == self.bot.user.id: return if payload.message_id not in self.reaction_messages: return emoji = str(payload.emoji) if emoji not in self.role_mapping: return guild = self.bot.get_guild(payload.guild_id) if not guild: return role_id = self.role_mapping[emoji] role = guild.get_role(role_id) if not role: return member = guild.get_member(payload.user_id) if not member: return try: await member.add_roles(role) except discord.Forbidden: pass @commands.Cog.listener() async def on_raw_reaction_remove(self, payload): if payload.user_id == self.bot.user.id: return if payload.message_id not in self.reaction_messages: return emoji = str(payload.emoji) if emoji not in self.role_mapping: return guild = self.bot.get_guild(payload.guild_id) if not guild: return role_id = self.role_mapping[emoji] role = guild.get_role(role_id) if not role: return member = guild.get_member(payload.user_id) if not member: return try: await member.remove_roles(role) except discord.Forbidden: pass async def setup(bot): await bot.add_cog(ReactionRoles(bot))