CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-06 11:18:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

[V3 Cleanup] Cleanup commands clean up after themselves (#1602)

Browse Source 
Resolves #1572
This commit is contained in:
Tobotimus 2018-05-14 09:51:50 +10:00 committed by palmtree5
parent 2b35d9f012
commit 8f74e4dd31
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
redbot/cogs/cleanup/cleanup.py
View File

@ -21,17 +21,24 @@ class Cleanup:
@staticmethod @staticmethod
async def check_100_plus(ctx: commands.Context, number: int) -> bool: async def check_100_plus(ctx: commands.Context, number: int) -> bool:
""" """
Called when trying to delete more than 100 messages at once Called when trying to delete more than 100 messages at once.
Prompts the user to choose whether they want to continue or not Prompts the user to choose whether they want to continue or not.
Tries its best to cleanup after itself if the response is positive.
""" """
def author_check(message): def author_check(message):
return message.author == ctx.author return message.author == ctx.author
await ctx.send(_('Are you sure you want to delete {} messages? (y/n)').format(number)) prompt = await ctx.send(_('Are you sure you want to delete {} messages? (y/n)').format(number))
response = await ctx.bot.wait_for('message', check=author_check) response = await ctx.bot.wait_for('message', check=author_check)
if response.content.lower().startswith('y'): if response.content.lower().startswith('y'):
await prompt.delete()
try:
await response.delete()
except:
pass
return True return True
else: else:
await ctx.send(_('Cancelled.')) await ctx.send(_('Cancelled.'))
@ -138,6 +145,7 @@ class Cleanup:
cleanup user @\u200bTwentysix 2 cleanup user @\u200bTwentysix 2
cleanup user Red 6""" cleanup user Red 6"""
member = None
try: try:
member = await commands.converter.MemberConverter().convert(ctx, user) member = await commands.converter.MemberConverter().convert(ctx, user)
except commands.BadArgument: except commands.BadArgument:
@ -243,6 +251,7 @@ class Cleanup:
ctx, channel, number, limit=1000, before=ctx.message, ctx, channel, number, limit=1000, before=ctx.message,
delete_pinned=delete_pinned delete_pinned=delete_pinned
) )
to_delete.append(ctx.message)
reason = "{}({}) deleted {} messages in channel {}."\ reason = "{}({}) deleted {} messages in channel {}."\
"".format(author.name, author.id, "".format(author.name, author.id,
@ -292,6 +301,7 @@ class Cleanup:
ctx, channel, number, check=check, limit=1000, before=ctx.message, ctx, channel, number, check=check, limit=1000, before=ctx.message,
delete_pinned=delete_pinned delete_pinned=delete_pinned
) )
to_delete.append(ctx.message)
reason = "{}({}) deleted {} "\ reason = "{}({}) deleted {} "\
" command messages in channel {}."\ " command messages in channel {}."\