为多个discord.py bot命令设置相同的冷却时间？

问题描述

这是用discord.py编写的。

我有多个类似以下内容的命令：

@bot.command(name ="hi")
async def hi(ctx):
    link = ["https://google.com", "https://youtube.com"]
    chosen = random.choice(link)
    url = chosen
    embed = discord.Embed(title="Your Link", description=f"[Click Here]({url})", color=0x00ff00)
    if ctx.message.guild == None:
        await ctx.author.send('You can not use this command in your DM!')
        pass
    else:
        await ctx.author.send(embed=embed)

如果有人使用其中一个命令，则应对所有命令实施冷却(例如，如果使用!hi，则!hi和!bye都将应用冷却。

我知道您可以使用@commands.cooldown(1, 600, commands.BucketType.user)，但这只对当前命令应用冷却。

---

解决方案

代码中的cooldowns修饰符是这样定义的：

def cooldown(rate, per, type=BucketType.default):
    def decorator(func):
        if isinstance(func, Command):
            func._buckets = CooldownMapping(Cooldown(rate, per, type))
        else:
            func.__commands_cooldown__ = Cooldown(rate, per, type)
        return func
    return decorator

我们可以对此进行修改，以便只创建一个在命令之间共享的Cooldown对象：

def shared_cooldown(rate, per, type=BucketType.default):
    cooldown = Cooldown(rate, per, type=type)
    def decorator(func):
        if isinstance(func, Command):
            func._buckets = CooldownMapping(cooldown)
        else:
            func.__commands_cooldown__ = cooldown
        return func
    return decorator

我们将通过调用它来获取装饰器，然后将其应用于命令：

my_cooldown = shared_cooldown(1, 600, commands.BucketType.user)

@bot.command()
@my_cooldown
async def hi(ctx):
    await ctx.send("Hi")


@bot.command()
@my_cooldown
async def bye(ctx):
    await ctx.send("Bye")