如何实现TeleBot仅允许触发命令的用户点击对应内联按钮

核心方案 #

不用全局变量存储用户ID，而是将发起命令的用户ID直接嵌入到按钮的callback_data中，在回调处理时验证点击者ID与嵌入的ID是否匹配，以此实现权限控制。

完整代码实现 #

@bot.message_handler(commands=['start'])
def start(message):
    markup = types.InlineKeyboardMarkup(row_width=1)
    # 将发起命令的用户ID拼接到callback_data中，格式为"指令_用户ID"
    b1 = types.InlineKeyboardButton(text='First option', callback_data=f'option1_{message.from_user.id}')
    markup.add(b1)
    bot.send_message(message.chat.id, 'Hello! Choose an option', reply_markup=markup)


@bot.callback_query_handler(lambda x: x.data.startswith('option1_'))
def parsing(callback):
    # 拆分callback_data，取出存储的用户ID
    try:
        _, stored_user_id = callback.data.split('_', 1)
        stored_user_id = int(stored_user_id)
    except (ValueError, IndexError):
        # 处理格式错误的情况，直接回复提示并结束
        bot.answer_callback_query(callback.id, text='无效的操作请求')
        return

    # 验证点击者ID与存储的用户ID是否一致
    if callback.from_user.id == stored_user_id:
        # 验证通过，执行后续操作
        bot.send_message(callback.message.chat.id, '操作成功！')
        bot.answer_callback_query(callback.id)
    else:
        # 验证失败，提示无权限
        bot.answer_callback_query(callback.id, text='你没有权限执行此操作')

代码说明 #

• 命令处理阶段：生成内联按钮时，把发起/start命令的用户ID拼接到callback_data中，确保每个按钮的回调数据都与发起用户绑定。
• 回调验证阶段：
  1. 通过split('_', 1)拆分回调数据，避免用户ID包含下划线导致的拆分错误
  2. 转换存储的用户ID为整数类型，避免字符串比对的问题
  3. 对比点击者的ID与存储的ID，一致则执行操作，否则返回无权限提示
• 异常处理：针对格式错误的回调数据做兜底处理，避免程序崩溃

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