[commands] Added a method to reset command cooldown.

This commit is contained in:
Dan Hess
2016-09-06 15:03:47 -05:00
committed by Rapptz
parent a23e51f6c4
commit b7ffbca0c7
2 changed files with 17 additions and 0 deletions

View File

@ -43,6 +43,7 @@ class Cooldown:
self.type = type
self._window = 0.0
self._tokens = self.rate
self._last = 0.0
if not isinstance(self.type, BucketType):
raise TypeError('Cooldown type must be a BucketType')
@ -72,6 +73,10 @@ class Cooldown:
if self._tokens == 0:
self._window = current
def reset(self):
self._tokens = self.rate
self._last = 0.0
def copy(self):
return Cooldown(self.rate, self.per, self.type)