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] 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
2 changed files with 10 additions and 3 deletions
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)