import datetime import mariadb from discord import Embed from discord.ext import commands from discord.ext.commands import command, cooldown, BucketType, has_permissions 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") @has_permissions(manage_messages=True, manage_roles=True, manage_channels=True) @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""" # Retrieve a list of channel id's in the guild channels = [channel.id for channel in ctx.guild.channels] # Allows the user to setup modmail for the first time if args[0] == "set": if args[1] == "modmail": # Checking if the guild already exists within the database with db.connection() as conn: # Get the author's row from the Members Table select_query = """SELECT * FROM moderatormail WHERE guildID = (?)""" val = ctx.author.guild.id, cursor = conn.cursor() # Execute the SQL Query cursor.execute(select_query, val) result = cursor.fetchone() # Throw error if the guild already exists and then stop the function if result is not None: await ctx.send("**Looks like this guild already has a modmail system set up!" + f"\nPlease check `{ctx.prefix}help` for information on how to update/delete existing information**") return # As long as the channel exists within the guild if int(args[2]) in 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) try: # 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) await ctx.send("**Your Modmail system is now successfully set up!" + f"\nPlease refer to `{ctx.prefix}help` for any information**") return except mariadb.IntegrityError as err: print(err) await ctx.send("Looks like this guild already has a modmail system set up!" + f"\nPlease check `{ctx.prefix}help` for information on how to update/delete existing information") else: # Send error message if the channel ID is not recognised await ctx.send("`Invalid Channel ID. Aborting Process...`") return else: # Send error message if the channel ID is not recognised await ctx.send("`Invalid Channel ID. Aborting Process...`") return # Allows the user to update the channel that the modmail can be sent to if args[0] == "update": if args[1] == "modmail": # Checking if the guild already exists within the database with db.connection() as conn: # Get the author's row from the Members Table select_query = """SELECT * FROM moderatormail WHERE guildID = (?)""" vals = ctx.author.guild.id, cursor = conn.cursor() # Execute the SQL Query cursor.execute(select_query, vals) result = cursor.fetchone() # Throw error if the guild already exists and then stop the function if result is None: await ctx.send("**Looks like this guild does not have a modmail system setup!" + f"\nPlease check `{ctx.prefix}help` for information on how to update/delete existing information**") return # As long as the channel exists within the guild if int(args[2]) in channels: try: # Store the information within the database with db.connection() as conn: # Define the insert statement that will insert information about the modmail channel update_query = """UPDATE moderatormail SET modmailChannelID = (?) WHERE guildID = (?)""" vals = args[2], ctx.author.guild.id cursor = conn.cursor() # Execute the SQL Query cursor.execute(update_query, vals) conn.commit() except mariadb.Error as err: print(err) await ctx.send("**Looks like something went wrong during the update!" "\nMake sure that the Channel ID is correct!**") channel = ctx.author.guild.get_channel(int(args[2])) await ctx.send( f"**The channel has been updated! Your new modmail will be sent to** {channel.mention}") else: # Send error message if the channel ID is not recognised await ctx.send("`Invalid Channel ID. Aborting Process...`") return # Allows the user to completely erase the modmail system currently set up if args[0] == "delete": if args[1] == "modmail": # Checking if the guild already exists within the database with db.connection() as conn: # Get the author's row from the Members Table select_query = """SELECT * FROM moderatormail WHERE guildID = (?)""" vals = ctx.author.guild.id, cursor = conn.cursor() # Execute the SQL Query cursor.execute(select_query, vals) result = cursor.fetchone() # Throw error if the guild already exists and then stop the function if result is None: await ctx.send("**Looks like this guild does not have a modmail system setup!" + f"\nPlease check `{ctx.prefix}help` for information on how to update/delete existing information**") return try: # Store the information within the database with db.connection() as conn: # Define the delete statement to remove all information about the guild delete_query = """DELETE FROM moderatormail WHERE guildID = (?)""" vals = ctx.author.guild.id, cursor = conn.cursor() # Execute the SQL Query cursor.execute(delete_query, vals) conn.commit() except mariadb.Error as err: print(err) await ctx.send("**Looks like this guild has not set up the modmail system yet!" + f"\nPlease do `{ctx.prefix}help` to find out how to set it up!**") # Sending confirmation message that the modmail system has been deleted await ctx.send("**Modmail system successfully deleted!" + f"\nPlease do `{ctx.prefix}help` to find out how to set Modmail again!**") def setup(bot): bot.add_cog(SetupModmail(bot))