每个用户的Discord.py不同的冷却时间 [英] Discord.py different cooldown per user

问题描述

如果用户不在列表中，我想创建一个仅具有冷却时间的命令这是我的代码:

I want to make a command that have only a cooldown if the user is not in a list This is my code:

@commands.command(name='cooldown')
@commands.cooldown(1, 300, commands.BucketType.user)
async def cdown(self, ctx):
  await ctx.send("Executed")

这应该是这样的:

@commands.command(name='cooldown')
if not user in no_cooldown_users:
  @commands.cooldown(1, 300, commands.BucketType.user)
async def cdown(self, ctx):
  await ctx.send("Executed")

是否有可能代替配置使用此功能?

Is there any possibility to make this instead of using a config?

推荐答案

您必须进行自定义冷却，这是一个非常简单的示例

You have to make a custom cooldown, here a very simple example

from discord.ext import commands

class CustomCooldown:
    def __init__(self, rate, per, alter_rate, alter_per, bucket, *, elements):
        self.elements = elements
        # Default mapping is the default cooldown
        self.default_mapping = commands.CooldownMapping.from_cooldown(rate, per, bucket)
        # Alter mapping is the alternative cooldown
        self.alter_mapping = commands.CooldownMapping.from_cooldown(alter_rate, alter_per, bucket)
        # Copy of the original BucketType
        self._bucket_type = bucket

    def __call__(self, ctx):
        key = self.alter_mapping ._bucket_key(ctx.message)

        if self._bucket_type is commands.BucketType.member: # `BucketType.member` returns a tuple
            key = key[1] # The second (last) value is the actual key, the first one is the guild ID

        if key in self.elements:
            # If the key is in the elements, the bucket will be taken from the alternative cooldown
            bucket = self.alter_mapping.get_bucket(ctx.message)
        else:
            # If not, from the default cooldown
            bucket = self.default_mapping.get_bucket(ctx.message)

        # Getting the ratelimit left (can be None)
        retry_after = bucket.update_rate_limit()

        if retry_after: # If the command is on cooldown, raising the error
            raise commands.CommandOnCooldown(bucket, retry_after)
        return True

要使用它:

@command.commands()
@commands.check(CustomCooldown(1, 300, 1, 0, commands.BucketType.user, elements=[list of ids]))
async def foo(self, ctx):
    ...

前两个值是速率和默认冷却时间的每一个，后两个值是针对特殊"冷却时间的.冷却.接下来是存储桶类型，最后是元素列表(标识).

The first two values are the rate and per of the default cooldown, the next two values are for the "special" cooldown. Next there's the bucket type, and lastly the list of elements (ids).

您还可以使用此方法对每个角色/行业/类别/渠道使用不同的冷却时间...

You can also use this method for different cooldown per roles/guilds/categories/channels...

这篇关于每个用户的Discord.py不同的冷却时间的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！