[commands] Allow passing reference time to update_rate_limit

6 years ago · ec7a701ceb
1 changed files with 2 additions and 2 deletions
--- a/discord/ext/commands/cooldowns.py
+++ b/discord/ext/commands/cooldowns.py
@ -61,8 +61,8 @@ class Cooldown:
            tokens = self.rate
        return tokens

-    def update_rate_limit(self):
-        current = time.time()
+    def update_rate_limit(self, current=None):
+        current = current or time.time()
        self._last = current

        self._tokens = self.get_tokens(current)