[MANAGECHAT] Cleaned up code

This commit is contained in:
Alexis LEBEL
2023-03-31 20:46:08 +02:00
parent 367a3b6082
commit c5c12991d7

View File

@@ -17,7 +17,8 @@ class ManageChat(discord.Cog):
f"The user {ctx.author} ran the cancel command in the channel {ctx.channel} of the guild {ctx.guild}, named {ctx.guild.name}" 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 # check if the guild is in the database
curs_data.execute("SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,)) curs_data.execute(
"SELECT * FROM data WHERE guild_id = ?", (ctx.guild.id,))
if curs_data.fetchone() is None: if curs_data.fetchone() is None:
await ctx.respond( await ctx.respond(
"This server is not setup, please run /setup", ephemeral=True "This server is not setup, please run /setup", ephemeral=True
@@ -87,32 +88,34 @@ class ManageChat(discord.Cog):
last_message: discord.Message = await ctx.channel.fetch_message( last_message: discord.Message = await ctx.channel.fetch_message(
ctx.channel.last_message_id ctx.channel.last_message_id
) )
new_file_name = f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt"
# 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 # 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( os.rename(
"transcript.txt", "transcript.txt",
f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt", new_file_name,
) )
# send the file in a private message to the user who ran the command # send the file in a private message to the user who ran the command
# TODO: rework so as to give the choice of a private send or a public send
if channel_send is None: if channel_send is None:
await ctx.respond( await ctx.respond(
file=discord.File( file=discord.File(
f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt" new_file_name
), ),
ephemeral=True, ephemeral=True,
) )
else: else:
await channel_send.send( await channel_send.send(
file=discord.File( file=discord.File(
f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt" new_file_name
) )
) )
await ctx.respond("Transcript sent!", ephemeral=True, delete_after=5) await ctx.respond("Transcript sent!", ephemeral=True, delete_after=5)
await ctx.author.send( await ctx.author.send(
file=discord.File( file=discord.File(
f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt" new_file_name
) )
) )
# delete the file # delete the file
os.remove( os.remove(
f"transcript_{ctx.guild.name}_{ctx.channel.name}_{last_message.created_at.strftime('%d-%B-%Y')}.txt" new_file_name
) )