mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-05-16 18:59:09 +00:00
Code Issues Projects Releases Wiki Activity

[commands] fix BucketType.default bypassing dynamic_cooldown

Browse Source
This commit is contained in:
Mikey 2022-04-30 15:46:37 -07:00 committed by GitHub
parent 8e9e25246e
commit 165e14063f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 2 deletions
discord/ext/commands
core.py
docs/ext/commands
api.rst

5
discord/ext/commands/core.py

@ -2315,7 +2315,7 @@ def cooldown(
def dynamic_cooldown(
cooldown: Union[BucketType, Callable[[Message], Any]],
type: BucketType = BucketType.default,
type: BucketType,
) -> Callable[[T], T]:
"""A decorator that adds a dynamic cooldown to a :class:`.Command`
@ -2348,6 +2348,9 @@ def dynamic_cooldown(
if not callable(cooldown):
raise TypeError("A callable must be provided")
if type is BucketType.default:
raise ValueError('BucketType.default cannot be used in dynamic cooldowns')
def decorator(func: Union[Command, CoroFunc]) -> Union[Command, CoroFunc]:
if isinstance(func, Command):
func._buckets = DynamicCooldownMapping(cooldown, type)

2
docs/ext/commands/api.rst

@ -356,7 +356,7 @@ Checks
.. autofunction:: discord.ext.commands.cooldown(rate, per, type=discord.ext.commands.BucketType.default)
:decorator:
.. autofunction:: discord.ext.commands.dynamic_cooldown(cooldown, type=BucketType.default)
.. autofunction:: discord.ext.commands.dynamic_cooldown(cooldown, type)
:decorator:
.. autofunction:: discord.ext.commands.max_concurrency(number, per=discord.ext.commands.BucketType.default, *, wait=False)