[commands] Allow passing reference time to update_rate_limit

This commit is contained in:
Rapptz 2019-04-14 16:57:47 -04:00
parent 9b089c9a7b
commit ec7a701ceb

View File

@ -61,8 +61,8 @@ class Cooldown:
tokens = self.rate tokens = self.rate
return tokens return tokens
def update_rate_limit(self): def update_rate_limit(self, current=None):
current = time.time() current = current or time.time()
self._last = current self._last = current
self._tokens = self.get_tokens(current) self._tokens = self.get_tokens(current)