python - Discord.py 重写冷却命令

Question

我正在使用 Discord.py Rewrite 来制作 Discord 机器人。我有一个名为 work 的命令，其中包含以下代码:

import discord
from discord.ext import commands

client = commands.Bot(command_prefix="c!")
# on_ready command

@client.command()
async def work(ctx):
    author_id = ctx.author.id
    currency_dict[author_id] += 50
    await ctx.send("You Gained 50 Coins")
# currency_dict is a dictionary that is already defined

有没有办法让用户每30秒只能使用一次该命令？谢谢!

Answer 1

首先，您必须导入from discord.ext.commands.cooldowns import BucketType。这将帮助你冷却。下面是可用于此导入的 cooldown 检查以及 max_concurrency 检查。

from discord.ext.commands.cooldowns import BucketType
# BucketType can be BucketType.default, member, user, guild, role, or channel
@commands.cooldown(rate,per,BucketType) 
# Limit how often a command can be used, (num per, seconds, BucketType)

@commands.max_concurrency(number, per=BucketType.default, *, wait=False)
# Limit how many instances of the command can be running at the same time.
# Setting wait=True will queue up additional commands. False will raise MaxConcurrencyReached

# Checks can be stacked, and will Raise a CheckFailure if any check fails.

就您而言，您需要使用commands.cooldown(rate,per,BucketType)。

import discord
from discord.ext import commands
from discord.ext.commands.cooldowns import BucketType

client = commands.Bot(command_prefix="c!")
# on_ready command

@client.command()
@commands.cooldown(1,30,commands.BucketType.user) # one command, every 30 seconds, per user
async def work(ctx):
    author_id = ctx.author.id
    currency_dict[author_id] += 50
    await ctx.send("You Gained 50 Coins")
# currency_dict is a dictionary that is already defined

# cooldown error-handling
@work.error
async def work_error(ctx, error):
    if isinstance(error, commands.CommandOnCooldown):
        await ctx.send(f'This command is on cooldown, you can use it in {round(error.retry_after, 2)} seconds')

---

引用文献:

• docs: discord.ext.commands.cooldown
• StackOverflow: Command cooldown in discord.py
• GitHub Gist: Built-in Checks for the commands extension of discord py