From a4abf84de2bb1dd110bfcd751cc19523ccc0a3b7 Mon Sep 17 00:00:00 2001
From: Ava <ava@yert.pink>
Date: Fri, 15 Feb 2019 21:31:46 +0200
Subject: Refine love calculator command

---
 bot/bot.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'bot/bot.py')

diff --git a/bot/bot.py b/bot/bot.py
index 2df550dc..12291a5f 100644
--- a/bot/bot.py
+++ b/bot/bot.py
@@ -5,6 +5,7 @@ from typing import List
 
 from aiohttp import AsyncResolver, ClientSession, TCPConnector
 from discord import Embed
+from discord.ext import commands
 from discord.ext.commands import Bot
 
 from bot import constants
@@ -61,3 +62,10 @@ class SeasonalBot(Bot):
         embed.set_author(name=title, icon_url=icon)
 
         await devlog.send(embed=embed)
+
+    async def on_command_error(self, context, exception):
+        # Don't punish the user for getting the arguments wrong
+        if isinstance(exception, commands.UserInputError):
+            context.command.reset_cooldown(context)
+        else:
+            super().on_command_error(context, exception)
-- 
cgit v1.2.3