Discord.py 拼写检查命令

2022-01-15 00:00:00 python discord.py

问题描述

最近,我查看了 Stack Overflow,发现了这段代码可以检查潜在的拼写错误:

Recently, I looked up Stack Overflow and found this code which can check for potential typos:

from difflib import SequenceMatcher
SequenceMatcher(None, "help", "hepl").ratio()
# Returns 0.75

这适用于 inside bot 命令的代码.但是,我应该如何做到这一点,如果我在命令名称中打错字,它会更正并执行命令?

This works for code inside the bot command. However, how should I make it so, if I make a typo in the command name, it will correct it and execute the command?


解决方案

from difflib import SequenceMatcher

@bot.event
async def on_command_error(ctx, error):
    if not isinstance(error, commands.CommandNotFound):
        return

    message = ctx.message  # later overwrite the attributes

    used_prefix = ctx.prefix  # the prefix used
    used_command = message.content.split()[0][len(used_prefix):]  # getting the command, `!foo a b c` -> `foo`

    available_commands = [cmd.name for cmd in bot.commands]
    matches = {  # command name: ratio
        cmd: SequenceMatcher(None, cmd, used_command).ratio()
        for cmd in available_commands
    }

    command = max(matches.items(), key=lambda item: item[1])[0]  # the most similar command

    try:
        arguments = message.content.split(" ", 1)[1]
    except IndexError:
        arguments = ""  # command didn't take any arguments

    new_content = f"{used_prefix}{command} {arguments}".strip()
    message.content = new_content  # overwriting the "original" message

    await bot.process_commands(message)  # processing commands with the new, updated message

相关文章