diff options
Diffstat (limited to 'discord')
| -rw-r--r-- | discord/ext/commands/cooldowns.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/discord/ext/commands/cooldowns.py b/discord/ext/commands/cooldowns.py index cb0f75cf..d7fe404f 100644 --- a/discord/ext/commands/cooldowns.py +++ b/discord/ext/commands/cooldowns.py @@ -140,7 +140,7 @@ class CooldownMapping: self._type = type def copy(self): - ret = CooldownMapping(self._cooldown) + ret = CooldownMapping(self._cooldown, self._type) ret._cache = self._cache.copy() return ret @@ -150,7 +150,7 @@ class CooldownMapping: @classmethod def from_cooldown(cls, rate, per, type): - return cls(Cooldown(rate, per, type)) + return cls(Cooldown(rate, per), type) def _bucket_key(self, msg): return self._type(msg) @@ -192,6 +192,11 @@ class DynamicCooldownMapping(CooldownMapping): super().__init__(None, type) self._factory = factory + def copy(self): + ret = DynamicCooldownMapping(self._factory, self._type) + ret._cache = self._cache.copy() + return ret + @property def valid(self): return True |