mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
3.6 KiB
Python
83 lines
3.6 KiB
Python
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))
|