Python cog 中每个命令的 discord.py 错误处理

发布于03月15日

我想在cog中添加一些错误处理，因为所有命令都需要用户的参数.然而，我发现实现这一点的唯一方法是每个命令都有自己的错误句柄，这非常耗时，因为我有50多个命令都使用这种格式.

如果可能的话，我希望能够创建一个错误处理程序，它只适用于该cog中的命令.我该怎么办？

命令示例:

  @commands.command()
  async def poke(self, ctx, tag: Optional[discord.Member]=None, *args):
    self.poke = False
    await ctx.send(f"{tag.mention} Has been poked by {ctx.author.mention}")

  @commands.command()
  async def fight(self, ctx, tag: Optional[discord.Member]=None, *args):
    self.fight = False
    await ctx.send(f"{tag.mention} Has been fought by {ctx.author.mention}")

推荐答案

Cog类中重写cog_command_error可以完成Cog特定的错误处理:

class Test(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.command()
    async def poke(self, ctx, tag: Optional[discord.Member] = None, *args):
        self.poke = False
        await ctx.send(f"{tag.mention} Has been poked by {ctx.author.mention}")

    @commands.command()
    async def fight(self, ctx, tag: Optional[discord.Member]=None, *args):
        self.fight = False
        await ctx.send(f"{tag.mention} Has been fought by {ctx.author.mention}")

    # Cog error handler
    async def cog_command_error(self, ctx, error):
        await ctx.send(f"An error occurred in the Test cog: {error}")