|
|
@ -139,6 +139,24 @@ def send_modmail(self, msg, author):
|
|
|
|
return embed
|
|
|
|
return embed
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def wait_for_msg(self, check, user_channel):
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
Method to check if the user actually types in a message
|
|
|
|
|
|
|
|
If not, delete the channel
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
|
|
|
# Wait for the message from the author
|
|
|
|
|
|
|
|
mod_message = await self.bot.wait_for('message', check=check, timeout=300.0)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Delete channel if user does not send a message within 5 minutes
|
|
|
|
|
|
|
|
except asyncio.TimeoutError:
|
|
|
|
|
|
|
|
await user_channel.delete()
|
|
|
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
return mod_message
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Set up the Cog
|
|
|
|
# Set up the Cog
|
|
|
|
class Guild(Cog):
|
|
|
|
class Guild(Cog):
|
|
|
|
"""Modmail System!"""
|
|
|
|
"""Modmail System!"""
|
|
|
@ -498,16 +516,13 @@ class Guild(Cog):
|
|
|
|
f"\nDo **{ctx.prefix}help modmail** to find out more!"
|
|
|
|
f"\nDo **{ctx.prefix}help modmail** to find out more!"
|
|
|
|
await self.bot.generate_embed(ctx, desc=text)
|
|
|
|
await self.bot.generate_embed(ctx, desc=text)
|
|
|
|
|
|
|
|
|
|
|
|
# Setting up Listener to listen for reactions within the modmail channel created
|
|
|
|
|
|
|
|
@Cog.listener()
|
|
|
|
@Cog.listener()
|
|
|
|
async def on_raw_reaction_add(self, payload):
|
|
|
|
async def on_raw_reaction_add(self, payload):
|
|
|
|
# Don't count reactions that are made by the bot
|
|
|
|
"""Listen for reactions for modmail channel"""
|
|
|
|
if payload.user_id == self.bot.user.id:
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Don't count reactions that are made by the bot
|
|
|
|
# Don't count other reactions other than ✅ and ❌
|
|
|
|
# Don't count other reactions other than ✅ and ❌
|
|
|
|
elif payload.user_id:
|
|
|
|
if payload.member.bot or str(payload.emoji) not in ['✅', '❌']:
|
|
|
|
if str(payload.emoji) not in ['✅', '❌']:
|
|
|
|
|
|
|
|
return
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
# Find a role corresponding to the Emoji name.
|
|
|
|
# Find a role corresponding to the Emoji name.
|
|
|
@ -583,7 +598,6 @@ class Guild(Cog):
|
|
|
|
def emoji_check(reaction, user):
|
|
|
|
def emoji_check(reaction, user):
|
|
|
|
return user == member and str(reaction.emoji) in ['✅', '❌']
|
|
|
|
return user == member and str(reaction.emoji) in ['✅', '❌']
|
|
|
|
|
|
|
|
|
|
|
|
# Surround with try/except to catch any exceptions that may occur
|
|
|
|
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
# Wait for the user to add a reaction
|
|
|
|
# Wait for the user to add a reaction
|
|
|
|
reaction, user = await self.bot.wait_for('reaction_add', check=emoji_check, timeout=60.0)
|
|
|
|
reaction, user = await self.bot.wait_for('reaction_add', check=emoji_check, timeout=60.0)
|
|
|
@ -598,6 +612,7 @@ class Guild(Cog):
|
|
|
|
def check(m):
|
|
|
|
def check(m):
|
|
|
|
return m.author == payload.member and user_channel.id == instructions.channel.id
|
|
|
|
return m.author == payload.member and user_channel.id == instructions.channel.id
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Checking if user wants to be Anonymous or not
|
|
|
|
if str(reaction.emoji) == "✅":
|
|
|
|
if str(reaction.emoji) == "✅":
|
|
|
|
self.anon = True
|
|
|
|
self.anon = True
|
|
|
|
|
|
|
|
|
|
|
@ -610,15 +625,16 @@ class Guild(Cog):
|
|
|
|
# Tell the user to type their mail into the chat
|
|
|
|
# Tell the user to type their mail into the chat
|
|
|
|
instructions = await user_channel.send(embed=send_instructions(self, member))
|
|
|
|
instructions = await user_channel.send(embed=send_instructions(self, member))
|
|
|
|
|
|
|
|
|
|
|
|
# Wait for the message from the author
|
|
|
|
msg = await wait_for_msg(self, check, user_channel)
|
|
|
|
msg = await self.bot.wait_for('message', check=check)
|
|
|
|
if msg is None: return
|
|
|
|
|
|
|
|
|
|
|
|
# Making sure that the message is below 50 characters and the message was sent in the channel
|
|
|
|
# Making sure that the message is below 50 characters and the message was sent in the channel
|
|
|
|
while len(msg.content) <= 50 and msg.channel == user_channel:
|
|
|
|
while len(msg.content) <= 50 and msg.channel == user_channel:
|
|
|
|
await user_channel.send(embed=error_handling(self, member))
|
|
|
|
await user_channel.send(embed=error_handling(self, member))
|
|
|
|
|
|
|
|
|
|
|
|
# Wait for the message from the author
|
|
|
|
# Wait for the message from the author
|
|
|
|
msg = await self.bot.wait_for('message', check=check)
|
|
|
|
msg = await wait_for_msg(self, check, user_channel)
|
|
|
|
|
|
|
|
if msg is None: return
|
|
|
|
|
|
|
|
|
|
|
|
# As long as the message is above 50 characters and in the correct channel
|
|
|
|
# As long as the message is above 50 characters and in the correct channel
|
|
|
|
if len(msg.content) > 50 and msg.channel == user_channel:
|
|
|
|
if len(msg.content) > 50 and msg.channel == user_channel:
|
|
|
@ -632,7 +648,6 @@ class Guild(Cog):
|
|
|
|
text_bytes = str.encode(text)
|
|
|
|
text_bytes = str.encode(text)
|
|
|
|
|
|
|
|
|
|
|
|
file = io.BytesIO(text_bytes)
|
|
|
|
file = io.BytesIO(text_bytes)
|
|
|
|
|
|
|
|
|
|
|
|
file_name = "Anon.txt" if self.anon else f"{member.name}.txt"
|
|
|
|
file_name = "Anon.txt" if self.anon else f"{member.name}.txt"
|
|
|
|
|
|
|
|
|
|
|
|
# Send the message to the modmail channel
|
|
|
|
# Send the message to the modmail channel
|
|
|
|