diff --git a/app/dragon-bot.py b/app/dragon-bot.py index ca28d18f..7c107bb4 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -162,7 +162,14 @@ async def on_message(message): await client.send_message(message.channel, random.choice(excuses)) if message.content.startswith('!purge'): - await client.purge_from(message.channel, limit=20, check=is_me) + num = 20 + if len(message.content.split()) > 1: + num = message.content.split()[1] + try: + await client.purge_from(message.channel, limit=num, check=is_me) + except ValueError: + await client.send_message(message.channel, 'You want me to purge what?') + # await client.send_message(message.channel, 'Deleted {} message(s)'.format(len(deleted))) if message.content.startswith('!decide'):