如何使用 discord.py 制作冷却系统？

Question

假设我有两个命令...

• hi 1 向用户发送一次 hi 并开始 500 秒的冷却时间
• hi 2 向用户发送两次 hi 并开始 1000 秒的冷却

现在当我输入hi 1 时，它不应该在我输入hi 2 时响应我。 类似于共享冷却系统的东西！

我该怎么做？

我目前正在使用这个：

@commands.cooldown(1, 5, commands.BucketType.user)

这让我可以使用多个命令，而无需停止前一个命令的冷却时间。

Answer 1

您可以使用custom check 来确保命令在冷却时不会被使用。 ./q63262849/cooldowns.py

import datetime
from discord.ext import commands

on_cooldown = {}  # A dictionary mapping user IDs to cooldown ends


def cooldown(seconds):
    def predicate(context):
        if (cooldown_end := on_cooldown.get(context.author.id)) is None or cooldown_end < datetime.datetime.now():  # If there's no cooldown or it's over
            if context.valid and context.invoked_with in (*context.command.aliases, context.command.name):  # If the command is being run as itself (not by help, which runs checks and would end up creating more cooldowns if this didn't exist)
                on_cooldown[context.author.id] = datetime.datetime.now() + datetime.timedelta(seconds=seconds)  # Add the datetime of the cooldown's end to the dictionary
            return True  # And allow the command to run
        else:
            raise commands.CommandOnCooldown(commands.BucketType.user, (cooldown_end - datetime.datetime.now()).seconds)  # Otherwise, raise the cooldown error to say the command is on cooldown

    return commands.check(predicate)

然后可以导入

./main.py

from q63262849 import cooldowns

并用作命令的装饰器

./main.py

@bot.command()
@cooldowns.cooldown(10)
async def cooldown1(ctx):
    await ctx.send("Ran cooldown 1")


@bot.command()
@cooldowns.cooldown(20)
async def cooldown2(ctx):
    await ctx.send("Ran cooldown 2")

值得注意的是，这种方法仍然存在一些问题，特别是如果稍后的另一个检查失败，此检查仍会将命令置于冷却状态，但是可以通过放置此检查以使其在所有其他检查之后运行来解决这些问题.