hammy
/
Enso-Bot
mirror of https://github.com/sgoudham/Enso-Bot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added Error Handling

Browse Source 
Added Permissions Check
Adding Updating ChannelID
pull/8/head
sgoudham 4 years ago
parent 14a36158b3
commit 1a8803dbd9
1 changed files with 113 additions and 43 deletions
Show Stats Download Patch File Download Diff File

156
cogs/modmail/setup.py
Unescape Escape View File

@ -1,8 +1,9 @@
import datetime import datetime
import mariadb
from discord import Embed from discord import Embed
from discord.ext import commands from discord.ext import commands
from discord.ext.commands import command, cooldown, BucketType from discord.ext.commands import command, cooldown, BucketType, has_permissions
import db import db
from settings import enso_embedmod_colours, blank_space from settings import enso_embedmod_colours, blank_space
@ -14,69 +15,138 @@ class SetupModmail(commands.Cog):
self.bot = bot self.bot = bot
@command(name="mmsetup") @command(name="mmsetup")
@has_permissions(manage_messages=True, manage_roles=True, manage_channels=True)
@cooldown(1, 1, BucketType.user) @cooldown(1, 1, BucketType.user)
async def _setup(self, ctx, *args): async def _setup(self, ctx, *args):
"""Allows the bot to setup a channel for users to react to for sending modmail""" """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 # Retrieve a list of channel id's in the guild
if args[0] == "set": channels = [channel.id for channel in ctx.guild.channels]
if args[1] == "modmail":
# Retrieve a list of channel id's in the guild # Checking if the guild already exists within the database
channels = [channel.id for channel in ctx.guild.channels] 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()
# Ask for the channel ID that the modmail should be logged to # Execute the SQL Query
await ctx.send("`Please enter the ID of the channel you want your modmail to be sent`") cursor.execute(select_query, val)
result = cursor.fetchone()
# Check the response is from the author and from the same channel as the previous message # Throw error if the guild already exists and then stop the function
def check(m): if result[0] is not None:
return m.author == ctx.author and m.channel == ctx.channel await ctx.send("Looks like this guild already has a modmail system set up!"
"\nPlease check `~help` for information on how to update/delete existing information")
return
# Wait for the message from the author # Allows the user to setup modmail for the first time
msg = await self.bot.wait_for('message', check=check) if args[0] == "set":
if args[1] == "modmail":
# As long as the channel exists within the guild # As long as the channel exists within the guild
if int(msg.content) in channels: 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)
except mariadb.IntegrityError as err:
print(err)
await ctx.send("Looks like this guild already has a modmail system set up!"
"\nPlease check `~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":
# Set up embed to let the user how to start sending modmail # As long as the channel exists within the guild
ModMail = Embed(title="**Welcome to Modmail!**", if int(args[2]) in channels:
colour=enso_embedmod_colours,
timestamp=datetime.datetime.utcnow())
ModMail.set_thumbnail(url=self.bot.user.avatar_url) 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()
# Define fields to be inserted into the embed # Execute the SQL Query
fields = [ cursor.execute(update_query, vals)
(blank_space, "**React to this message if you want to send a message to the Staff Team!**", conn.commit()
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 except mariadb.Error as err:
for name, value, inline in fields: print(err)
ModMail.add_field(name=name, value=value, inline=inline) await ctx.send("Looks like something went wrong during the update!"
"\nMake sure that the Channel ID is correct!")
# Get the channel object from the channelID input by the user
channel = ctx.author.guild.get_channel(int(args[2])) channel = ctx.author.guild.get_channel(int(args[2]))
modmailchannelID = await channel.send(embed=ModMail) await ctx.send(
# Auto add the ✅ reaction f"**The channel has been updated! Your new modmail will be sent to** {channel.mention}")
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: else:
# Send error message if the channel ID is not recognised # Send error message if the channel ID is not recognised
await ctx.send("`Invalid Channel ID. Aborting Process...`") await ctx.send("`Invalid Channel ID. Aborting Process...`")
return return
# Allows the user to completely erase the modmail system currently set up
if args[0] == "delete":
if args[1] == "modmail":
pass
def setup(bot): def setup(bot):
bot.add_cog(SetupModmail(bot)) bot.add_cog(SetupModmail(bot))

Write Preview
Loading…
Cancel
Save