aboutsummaryrefslogtreecommitdiff
path: root/discord/ext/commands/cooldowns.py
diff options
context:
space:
mode:
Diffstat (limited to 'discord/ext/commands/cooldowns.py')
-rw-r--r--discord/ext/commands/cooldowns.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/discord/ext/commands/cooldowns.py b/discord/ext/commands/cooldowns.py
index 08f36b60..c530bb12 100644
--- a/discord/ext/commands/cooldowns.py
+++ b/discord/ext/commands/cooldowns.py
@@ -91,6 +91,15 @@ class Cooldown:
tokens = self.rate
return tokens
+ def get_retry_after(self, current=None):
+ current = current or time.time()
+ tokens = self.get_tokens(current)
+
+ if tokens == 0:
+ return self.per - (current - self._window)
+
+ return 0.0
+
def update_rate_limit(self, current=None):
current = current or time.time()
self._last = current