discord.py-rewrite - 如何在 cogs 中使用自定义检查？

Question

所以，在我的 bot.py 主文件中，我有：

class Bot(commands.Bot):

    # BOT ATTRIBUTES

    class default_cooldown:
        maxcommands = ...
        seconds = ...
        mode = ...

    class my_exception(commmands.CommandError): pass

    def my_check(self):
        def predicate(ctx):
            if ctx.author in a_list: return True
            raise self.my_exception

bot = Bot(...)

现在我还有一个 cog 文件，我想在其中使用 Bot().my_check() 检查和我的 Bot().default_cooldown() 类：

class Cog(commands.Cog):

    def __init__(self, bot):
        self.bot = bot

    @commands.command()
    @self.bot.my_check()
    @commands.cooldown(self.bot.default_cooldown().maxcommands, self.bot.default_cooldown().seconds, self.bot.default_cooldown().mode)
    async def a_command(self, ctx):
        pass

但是我收到一个错误，说我的检查和冷却中没有定义self。谁能帮我解决这个问题？

Answer 1

方法在创建类对象时创建一次，而不是为每个实例单独创建。您有两种选择：

1. 将需要出现在这两个地方的代码分解到仅包含该代码的第三个模块中，然后将其导入到其他两个文件中
2. 将您的 Bot 定义移动到与您的机器人执行不同的模块中，并将 my_check 设置为 staticmethod。然后就可以通过Bot.my_check而不是通过具体实例来访问了。
3. 在setup 中定义你的cog，以便类在创建bot 实例时知道它。