Adding setup command for first time setup of modmail

cogs/modmail/setup.py

@@ -0,0 +1,82 @@
+import datetime
+
+from discord import Embed
+from discord.ext import commands
+from discord.ext.commands import command, cooldown, BucketType
+
+import db
+from settings import enso_embedmod_colours, blank_space
+
+
+# Set up the Cog
+class SetupModmail(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @command(name="mmsetup")
+    @cooldown(1, 1, BucketType.user)
+    async def _setup(self, ctx, *args):
+        """Allows the bot to setup a channel for users to react to for sending modmail"""
+
+        # Make sure the first two arguments are set and modmail
+        if args[0] == "set":
+            if args[1] == "modmail":
+
+                # Retrieve a list of channel id's in the guild
+                channels = [channel.id for channel in ctx.guild.channels]
+
+                # Ask for the channel ID that the modmail should be logged to
+                await ctx.send("`Please enter the ID of the channel you want your modmail to be sent`")
+
+                # Check the response is from the author and from the same channel as the previous message
+                def check(m):
+                    return m.author == ctx.author and m.channel == ctx.channel
+
+                # Wait for the message from the author
+                msg = await self.bot.wait_for('message', check=check)
+
+                # As long as the channel exists within the guild
+                if int(msg.content) in channels:
+
+                    # Set up embed to let the user how to start sending modmail
+                    ModMail = Embed(title="**Welcome to Modmail!**",
+                                    colour=enso_embedmod_colours,
+                                    timestamp=datetime.datetime.utcnow())
+
+                    ModMail.set_thumbnail(url=self.bot.user.avatar_url)
+
+                    # Define fields to be inserted into the embed
+                    fields = [
+                        (blank_space, "**React to this message if you want to send a message to the Staff Team!**",
+                         False),
+                        ("**React with ✅**",
+                         "We encourage all suggestions/thoughts and opinions on the server! As long as it is **valid** criticism."
+                         "Purely negative feedback will not be considered.", False)]
+
+                    # Add the fields to the embed
+                    for name, value, inline in fields:
+                        ModMail.add_field(name=name, value=value, inline=inline)
+
+                    # Get the channel object from the channelID input by the user
+                    channel = ctx.author.guild.get_channel(int(args[2]))
+                    modmailchannelID = await channel.send(embed=ModMail)
+                    # Auto add the ✅ reaction
+                    await modmailchannelID.add_reaction('✅')
+
+                    # Store the information within the database
+                    with db.connection() as conn:
+                        # Define the insert statement that will insert information about the modmail channel
+                        insert_query = """INSERT INTO moderatormail (guildID, channelID, messageID, modmailChannelID) VALUES (?, ?, ?, ?)"""
+                        vals = ctx.author.guild.id, args[2], modmailchannelID.id, int(msg.content),
+                        cursor = conn.cursor()
+
+                        # Execute the SQL Query
+                        cursor.execute(insert_query, vals)
+                else:
+                    # Send error message if the channel ID is not recognised
+                    await ctx.send("`Invalid Channel ID. Aborting Process...`")
+                    return
+
+
+def setup(bot):
+    bot.add_cog(SetupModmail(bot))