Discord机器人calc命令调用WolframAlpha API触发StopIteration错误如何解决

错误原因 #

• 你调用client.query()时传的是wait_for拿到的Discord消息对象，不是用户输入的文本内容，WolframAlpha API识别不了你要查询的内容，返回的结果集是空的，调用next()读取空迭代器就会抛出StopIteration报错。
• 代码没有做异常兜底，就算传对了内容，也存在WolframAlpha没有对应查询结果的场景，直接next()也会触发相同报错。

修复后的代码 #

import discord
import wolframalpha as wf

from discord.ext import commands

bot = commands.Bot(command_prefix='}')

@bot.command()
async def hello(ctx):
    await ctx.send('hi!!!!')


@bot.command()
async def hru(ctx):
    await ctx.send('Im doing great, hbu?')


# 修复后的calc命令
@bot.command()
async def calc(ctx):
    # 优化check逻辑：只接收当前触发命令的用户、在当前频道发送的非空内容，避免其他消息干扰
    def check(m):
        return len(m.content) >= 1 and m.author == ctx.author and m.channel == ctx.channel

    await ctx.send("请输入你要计算的内容：")
    wftoken = '替换为你的wolframalpha app id'
    client = wf.Client(wftoken)
    query_msg = await bot.wait_for("message", check=check)
    # 传入消息的content文本内容，而非整个消息对象
    res = client.query(query_msg.content)
    try:
        answer = next(res.results).text
        await ctx.send(answer)
    except StopIteration:
        # 无结果时返回友好提示
        await ctx.send("没有查询到对应结果，请检查输入内容是否规范。")


bot.run('替换为你的机器人token')

其他注意事项 #

• 检查你的WolframAlpha App ID是否有效，是否超出了单日调用次数限制，权限异常也会导致API返回空结果。
• 如果需要支持更复杂的数学表达式输入，可以对用户输入的内容做预处理，过滤多余的换行、特殊符号，提升API识别成功率。

内容的提问来源于stack exchange，提问作者Maxi Prad