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 5 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 mariadb
from discord import Embed
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
from settings import enso_embedmod_colours, blank_space
@ -14,69 +15,138 @@ class SetupModmail(commands.Cog):
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"""
# 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]
# Retrieve a list of channel id's in the guild
channels = [channel.id for channel in ctx.guild.channels]
# 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()
# 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`")
# Execute the SQL Query
cursor.execute(select_query, val)
result = cursor.fetchone()
# 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
# Throw error if the guild already exists and then stop the function
if result[0] is not None:
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
msg = await self.bot.wait_for('message', check=check)
# Allows the user to setup modmail for the first time
if args[0] == "set":
if args[1] == "modmail":
# 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
ModMail = Embed(title="**Welcome to Modmail!**",
colour=enso_embedmod_colours,
timestamp=datetime.datetime.utcnow())
# As long as the channel exists within the guild
if int(args[2]) in channels:
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
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)]
# Execute the SQL Query
cursor.execute(update_query, vals)
conn.commit()
# Add the fields to the embed
for name, value, inline in fields:
ModMail.add_field(name=name, value=value, inline=inline)
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!")
# 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(
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":
pass
def setup(bot):
bot.add_cog(SetupModmail(bot))

Write Preview
Loading…
Cancel
Save