mirror of
https://github.com/Paillat-dev/Botator.git
synced 2026-01-02 01:06:19 +00:00
74 lines
4.6 KiB
Python
74 lines
4.6 KiB
Python
import discord
|
|
import re
|
|
import os
|
|
from config import debug, c
|
|
|
|
class ManageChat (discord.Cog):
|
|
def __init__(self, bot: discord.Bot) -> None:
|
|
super().__init__()
|
|
self.bot = bot
|
|
|
|
@discord.slash_command(name="cancel", description="Cancel the last message sent into a channel")
|
|
async def cancel(self, ctx: discord.ApplicationContext):
|
|
debug(f"The user {ctx.author} ran the cancel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
|
#check if the guild is in the database
|
|
c.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
|
|
if c.fetchone() is None:
|
|
await ctx.respond("This server is not setup, please run /setup", ephemeral=True)
|
|
return
|
|
#get the last message sent by the bot in the cha where the command was sent
|
|
last_message = await ctx.channel.fetch_message(ctx.channel.last_message_id)
|
|
#delete the message
|
|
await last_message.delete()
|
|
await ctx.respond("The last message has been deleted", ephemeral=True)
|
|
|
|
#add a slash command called "clear" that deletes all the messages in the channel
|
|
@discord.slash_command(name="clear", description="Clear all the messages in the channel")
|
|
async def clear(self, ctx: discord.ApplicationContext):
|
|
debug(f"The user {ctx.author.display_name} ran the clear command command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
|
await ctx.respond("messages deleted!", ephemeral=True)
|
|
return await ctx.channel.purge()
|
|
|
|
@discord.slash_command(name="transcript", description="Get a transcript of the messages that have been sent in this channel intoa text file")
|
|
@discord.option(name="channel_send", description="The channel to send the transcript to", required=False)
|
|
async def transcript(self, ctx: discord.ApplicationContext, channel_send: discord.TextChannel = None):
|
|
debug(f"The user {ctx.author.display_name} ran the transcript command command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}")
|
|
#save all the messages in the channel in a txt file and send it
|
|
messages = await ctx.channel.history(limit=None).flatten()
|
|
messages.reverse()
|
|
transcript = ""
|
|
#defer the response
|
|
await ctx.defer() #defer the response so that the bot doesn't say that it's thinking
|
|
for msg in messages:
|
|
if msg.author.bot:
|
|
transcript += f"Botator: {msg.content}\n"
|
|
else:
|
|
mentions = re.findall(r"<@!?\d+>", msg.content)
|
|
#then replace each mention with the name of the user
|
|
for mention in mentions:
|
|
#get the user id
|
|
id = mention[2:-1]
|
|
#get the user
|
|
user = await self.bot.fetch_user(id)
|
|
#replace the mention with the name
|
|
msg.content = msg.content.replace(mention, msg.guild.get_member(user.id).display_name)
|
|
transcript += f"{msg.author.display_name}: {msg.content}\n"
|
|
#save the transcript in a txt file called transcript.txt. If the file already exists, delete it and create a new one
|
|
#check if the file exists
|
|
if os.path.exists("transcript.txt"):
|
|
os.remove("transcript.txt")
|
|
f = open("transcript.txt", "w")
|
|
f.write(transcript)
|
|
f.close()
|
|
last_message: discord.Message = await ctx.channel.fetch_message(ctx.channel.last_message_id)
|
|
#rename the file with the name of the channel and the date in this format: transcript_servername_channelname_dd-month-yyyy.txt ex : transcript_Botator_Testing_12-may-2021.txt
|
|
os.rename("transcript.txt", f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt")
|
|
#send the file in a private message to the user who ran the command
|
|
if channel_send is None:
|
|
await ctx.respond(file=discord.File(f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt"), ephemeral=True)
|
|
else:
|
|
await channel_send.send(file=discord.File(f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt"))
|
|
await ctx.respond("Transcript sent!", ephemeral=True, delete_after=5)
|
|
await ctx.author.send(file=discord.File(f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt"))
|
|
#delete the file
|
|
os.remove(f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt") |