mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-04-20 07:54:36 +00:00
Code Issues Projects Releases Wiki Activity

[commands] Fix errors when DynamicCooldown returns None

Browse Source
This commit is contained in:
z03h 2022-05-17 00:37:31 -07:00 committed by GitHub
parent dbb00bd361
commit 2be65e5874
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -125,9 +125,9 @@ class CooldownMapping:
def create_bucket(self, message: Message) -> Cooldown:
return self._cooldown.copy() # type: ignore
def get_bucket(self, message: Message, current: Optional[float] = None) -> Cooldown:
def get_bucket(self, message: Message, current: Optional[float] = None) -> Optional[Cooldown]:
if self._type is BucketType.default:
return self._cooldown # type: ignore
return self._cooldown
self._verify_cache_integrity(current)
key = self._bucket_key(message)
@ -142,6 +142,8 @@ class CooldownMapping:
def update_rate_limit(self, message: Message, current: Optional[float] = None, tokens: int = 1) -> Optional[float]:
bucket = self.get_bucket(message, current)
if bucket is None:
return None
return bucket.update_rate_limit(current, tokens=tokens)

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

@ -930,6 +930,8 @@ class Command(_BaseCommand, Generic[CogT, P, T]):
return False
bucket = self._buckets.get_bucket(ctx.message)
if bucket is None:
return False
dt = ctx.message.edited_at or ctx.message.created_at
current = dt.replace(tzinfo=datetime.timezone.utc).timestamp()
return bucket.get_tokens(current) == 0
@ -948,7 +950,8 @@ class Command(_BaseCommand, Generic[CogT, P, T]):
"""
if self._buckets.valid:
bucket = self._buckets.get_bucket(ctx.message)
bucket.reset()
if bucket is not None:
bucket.reset()
def get_cooldown_retry_after(self, ctx: Context[BotT], /) -> float:
"""Retrieves the amount of seconds before this command can be tried again.
@ -972,6 +975,8 @@ class Command(_BaseCommand, Generic[CogT, P, T]):
"""
if self._buckets.valid:
bucket = self._buckets.get_bucket(ctx.message)
if bucket is None:
return 0.0
dt = ctx.message.edited_at or ctx.message.created_at
current = dt.replace(tzinfo=datetime.timezone.utc).timestamp()
return bucket.get_retry_after(current)