[commands] Allow passing reference time to update_rate_limit

2019-04-14 16:57:47 -04:00 · 2019-04-14 16:57:47 -04:00 · ec7a701ceb
commit ec7a701ceb
parent 9b089c9a7b
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)