[commands] Allow subtracting multiple tokens from cooldown

3 years ago · 8401316e48
1 changed files with 2 additions and 2 deletions
--- a/discord/ext/commands/cooldowns.py
+++ b/discord/ext/commands/cooldowns.py
@ -253,9 +253,9 @@ class CooldownMapping:

        return bucket

-    def update_rate_limit(self, message: Message, current: Optional[float] = None) -> Optional[float]:
+    def update_rate_limit(self, message: Message, current: Optional[float] = None, tokens: int = 1) -> Optional[float]:
        bucket = self.get_bucket(message, current)
-        return bucket.update_rate_limit(current)
+        return bucket.update_rate_limit(current, tokens=tokens)


 class DynamicCooldownMapping(CooldownMapping):