matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 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

6
discord/ext/commands/core.py
View File

@@ -438,7 +438,7 @@ class Command:
yield from self._verify_checks(ctx)
if self._buckets.valid:
bucket = self._buckets.get_bucket(ctx)
bucket = self._buckets.get_bucket(ctx.message)
retry_after = bucket.update_rate_limit()
if retry_after:
raise CommandOnCooldown(bucket, retry_after)
@@ -462,7 +462,7 @@ class Command:
if not self._buckets.valid:
return False
bucket = self._buckets.get_bucket(ctx)
bucket = self._buckets.get_bucket(ctx.message)
return bucket.get_tokens() == 0
def reset_cooldown(self, ctx):
@@ -474,7 +474,7 @@ class Command:
The invocation context to reset the cooldown under.
"""
if self._buckets.valid:
bucket = self._buckets.get_bucket(ctx)
bucket = self._buckets.get_bucket(ctx.message)
bucket.reset()
@asyncio.coroutine