mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-09-04 17:06:21 +00:00
Code Issues Projects Releases Wiki Activity

[commands] Make CooldownMapping.get_bucket take Message instead.

Browse Source 
Requiring a full blown Context might be a bit overkill considering
we only use a single attribute from it.
This commit is contained in:
Rapptz
2017-10-08 07:52:56 -04:00
parent d53cada91e
commit 1bb7b6ff2d
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
discord/ext/commands/cooldowns.py
View File

@ -103,8 +103,7 @@ class CooldownMapping:
def from_cooldown(cls, rate, per, type):
return cls(Cooldown(rate, per, type))
def _bucket_key(self, ctx):
msg = ctx.message
def _bucket_key(self, msg):
bucket_type = self._cooldown.type
if bucket_type is BucketType.user:
return msg.author.id
@ -122,12 +121,12 @@ class CooldownMapping:
for k in dead_keys:
del self._cache[k]
def get_bucket(self, ctx):
def get_bucket(self, message):
if self._cooldown.type is BucketType.default:
return self._cooldown
self._verify_cache_integrity()
key = self._bucket_key(ctx)
key = self._bucket_key(message)
if key not in self._cache:
bucket = self._cooldown.copy()
self._cache[key] = bucket