From 275e0ad7cff47fce5d066d7ddcbcd39b43722c2b Mon Sep 17 00:00:00 2001 From: sgoudham Date: Tue, 21 Jul 2020 11:27:28 +0100 Subject: [PATCH] Inserting verification listener/command Making sure verification check actually works --- cogs/fun/enso.py | 74 ++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 72 insertions(+), 2 deletions(-) diff --git a/cogs/fun/enso.py b/cogs/fun/enso.py index d9a7a0a7..fc7fe522 100644 --- a/cogs/fun/enso.py +++ b/cogs/fun/enso.py @@ -3,12 +3,14 @@ import datetime import random import string +import discord from discord import Embed, Colour from discord.ext import commands -from discord.ext.commands import cooldown, BucketType, command +from discord.ext.commands import cooldown, BucketType, command, is_owner import settings -from settings import colour_list, enso_guild_ID, enso_ensochancommands_Mention, blank_space, enso_embedmod_colours +from settings import colour_list, enso_guild_ID, enso_ensochancommands_Mention, blank_space, enso_embedmod_colours, \ + enso_verification_ID # Error handling function to make sure that the commands only work in "enso-chan-commands" @@ -297,6 +299,74 @@ class Enso(commands.Cog): # Delete the message await message.delete() + @commands.Cog.listener() + async def on_raw_reaction_add(self, payload): + """Verification listener for Ensō""" + + # Get the guild + guild = self.bot.get_guild(enso_guild_ID) + # Get the member + member = guild.get_member(payload.user_id) + # Getting the channel verification by setting it to #verification + channel = guild.get_channel(enso_verification_ID) + + # If the channel is #verification + if payload.channel_id == channel.id: + + # A check that makes sure that the reaction is done by the bot + def check(m): + return m == self.bot.user + + # If the member is not a user, do nothing + if check(payload.member): + return + else: + + # Get the 'Lucid' role and then give it to the user + role = discord.utils.get(guild.roles, name='Lucid') + + # if the emoji that was reacted is the tick mark. + if payload.emoji.name == "✅": + await member.add_roles(role) + + # Set hamothyID equal to my id in discord + hamothyID = '<@&715412394968350756>' + + # Set the channel id to "general" + general = guild.get_channel(663651584399507481) + + # String for welcoming people in the #general channel + general_welcome = f"Welcome to the server! {member.mention} I hope you enjoy your stay here " \ + f"\nPlease go into <#722347423913213992> to choose some ping-able roles for events! " \ + f"\nPlease ping {hamothyID} for any questions about the server and of course, the other staff members!" + + # Send welcome message to #general + await general.send(general_welcome) + + # Allowing people to get ping-able self roles + @command(name="verification", hidden=True) + @is_owner() + async def verification(self, ctx): + # Set up embed to let the user know that they have to react with ✅ + embed = Embed(title="**Verification**", + colour=Colour(0xFF69B4), + timestamp=datetime.datetime.utcnow()) + + embed.set_thumbnail(url=ctx.guild.icon_url) + embed.set_author(name=ctx.author, icon_url=ctx.author.avatar_url) + embed.add_field( + name="Remember to read the rules!", + value="React with ✅ to gain access to the rest of the server!", + inline=False) + + # Send embed to the channel it was called in and automatically add the reaction ✅ + # verif = await ctx.send(embed=embed) + # await verif.add_reaction('✅') + + # Edit the Embed And Update it + verif = await ctx.fetch_message(728424149692842115) + await verif.edit(embed=embed) + def setup(bot): bot.add_cog(Enso(bot))