Aiogram 多聊天机器人状态管理技巧

本文档旨在解决在使用 Aiogram 构建多聊天 Telegram 机器人时，由于不恰当的状态管理导致后续聊天无法使用机器人功能的问题。通过分析问题代码，明确状态设置的必要性，并提供修改后的代码示例，帮助开发者避免类似问题，提升机器人用户体验。

在使用 Aiogram 构建 Telegram 机器人时，状态管理是一个重要的概念，它允许机器人记住用户的上下文信息，并根据不同的状态执行不同的操作。然而，不恰当的状态管理可能会导致意想不到的问题，例如，多个聊天室中的用户在使用相同命令时，机器人可能会进入错误的状态，导致后续交互异常。

问题分析

原始代码中，在 /help 命令的处理函数 help 中，使用了 await AnswerState.helpInfo.set() 来设置状态。这个状态的目的是什么？从后续代码来看，这个状态似乎并没有实际用途，因为在 help_handler_answer 函数中，只是简单地更新了状态数据，并没有依赖这个状态进行任何逻辑判断。

更重要的是，这个状态的设置会影响到其他聊天室的用户。当一个聊天室的用户触发了 /help 命令后，机器人会进入 AnswerState.helpInfo 状态。如果此时另一个聊天室的用户也尝试使用 /help 命令，由于机器人已经处于 AnswerState.helpInfo 状态，这个新的 /help 命令将不会被 help 函数处理，而是会被状态处理器处理，导致功能失效。

解决方案

解决这个问题的方法很简单：移除 await AnswerState.helpInfo.set() 这行代码。因为如上文分析，这个状态的设置并没有实际用途。

修改后的 help 函数如下：

help_cb = CallbackData("help",'SenderId', "SenderName", 'action')
appeal_cb = CallbackData("appeal",'SenderId', 'action')

@dp.message_handler(commands="help")
async def help(message: types.Message):
    print("help")
    result = await db.check_if_allow(message.chat.id)
    if not result:
        await message.answer("Поки-що вам не дозволено відправляти запити про допогу. Спробуйте пізніше")
    else:
        if len(message.text) < 3:
            await message.answer("Опишіть вашу проблему детальніше.")
        else:
            forms, form = await db.get_chats_name(message.chat.id)
            await db.update_cooldown(message.chat.id)

            markup = InlineKeyboardMarkup()
            markup.add(InlineKeyboardButton("? Відповісти", callback_data=help_cb.new(SenderId = message.chat.id, SenderName = form[0],  action='answer')))
            markup.add(InlineKeyboardButton("?️ Видалити", callback_data=help_cb.new(SenderId = message.chat.id, SenderName = form[0], action='delete')))
            markup.add(InlineKeyboardButton("⛔ Поскаржитися", callback_data=help_cb.new(SenderId = message.chat.id, SenderName = form[0], action='appeal')))

            await message.answer("Запит на допомогу успішно надіслана")

            text = f"<b>? Новий запит про допомогу!</b>\nКлас: {form[0]}\nЗапитання: {message.text}"

            for form in forms:
                await bot.send_message(form[0], text=text, reply_markup = markup)

            # await AnswerState.helpInfo.set() # 移除这行代码

进一步优化

虽然移除 await AnswerState.helpInfo.set() 可以解决问题，但仍然可以对代码进行一些优化，使其更清晰易懂。例如，可以将 helpInfo 数据直接存储在 help_handler_answer 函数中，而不是通过状态来传递。

修改后的 help_handler_answer 函数如下：

@dp.callback_query_handler(help_cb.filter(action='answer'))
async def help_handler_answer(query: CallbackQuery, callback_data: dict, state: FSMContext):
    print("answer")
    # await state.update_data(helpInfo = [callback_data["SenderId"], callback_data["SenderName"]]) # 移除这行代码
    text = f"<b>Введіть відповідь на запитання</b>"
    await query.message.edit_reply_markup(None)
    await query.answer(text=text)
    # await AnswerState.text.set() # 移除这行代码
    await state.update_data(sender_id=callback_data["SenderId"], sender_name=callback_data["SenderName"])
    await AnswerState.text.set()

并修改 process_title 函数：

@dp.message_handler(state=AnswerState.text)
async def process_title(message: Message, state: FSMContext):
    print("text")
    if len(message.text) < 3:
        await message.answer("В пояснені напишіть більше 2 слів.")
    else:
        await state.update_data(text = message.text)
        data = await state.get_data()

        await message.answer(f"<b>{data['sender_name']} щиро вдячний за допомогу!</b>")
        await state.finish()
        await bot.send_message(chat_id=data["sender_id"], text=f"<b>? Відповідь отримана!</b>\n\n{message.text}")

注意事项：

• 确保理解 Aiogram 的状态管理机制，避免滥用状态。
• 仔细分析代码逻辑，明确每个状态的用途。
• 在多聊天机器人开发中，要特别注意状态隔离，避免不同聊天室之间的状态冲突。

总结：

通过本文的分析和修改，可以解决在使用 Aiogram 构建多聊天 Telegram 机器人时，由于不恰当的状态管理导致后续聊天无法使用机器人功能的问题。理解状态管理的原理，并根据实际需求合理使用状态，是构建稳定可靠的 Telegram 机器人的关键。在调试过程中，可以利用 print 语句或日志记录来跟踪状态变化，帮助定位问题。