如何将Python Telebot的照片上传与/foto命令绑定？

要实现只有用户输入/foto命令后才能上传照片的需求，核心是跟踪用户的交互状态，避免全局监听照片消息。以下是修正后的实现方案：

实现思路 #

• 用一个字典维护用户的当前状态，标记用户是否处于等待上传照片的状态
• 仅当用户状态为「等待照片」时，才处理其上传的照片
• 修正原代码中register_next_step_handler的调用错误（原代码错误地调用了photo()而非传入函数对象）

修正后的完整代码 #

import telebot
import logging

# 初始化bot（替换为你的token）
bot = telebot.TeleBot("YOUR_BOT_TOKEN")

# 配置日志
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)

# 存储用户状态：key为用户ID，value为当前状态（比如"waiting_for_photo"）
user_states = {}

@bot.message_handler(commands=['foto'])
def prompt_photo_upload(message):
    chat_id = message.chat.id
    # 设置用户状态为等待上传照片
    user_states[chat_id] = "waiting_for_photo"
    bot.send_message(chat_id, 'Subí tu foto.')

@bot.message_handler(content_types=["photo"])
def handle_photo(message):
    chat_id = message.chat.id
    # 检查用户是否处于等待上传照片的状态
    if user_states.get(chat_id) != "waiting_for_photo":
        # 未触发/foto命令，提示用户先执行命令
        bot.send_message(chat_id, '请先输入/foto命令，再上传照片哦！')
        return
    
    # 处理照片逻辑
    fileID = message.photo[-1].file_id
    logger.info("User_id, fileID: %s, %s", message.from_user.id, fileID)
    file_info = bot.get_file(fileID)
    downloaded_file = bot.download_file(file_info.file_path)
    
    # 保存照片
    with open(f"./images/{fileID}.jpg", 'wb') as new_file:
        new_file.write(downloaded_file)
    
    bot.send_message(chat_id, 'Foto recibida, gracias!')
    # 重置用户状态，避免重复处理
    del user_states[chat_id]

# 启动bot
bot.polling()

关键代码说明 #

1. 状态字典user_states：
   用来记录每个用户的交互状态，当用户输入/foto时，将其ID对应的状态设为waiting_for_photo；处理完照片后删除该状态，确保用户下次上传需要重新触发命令。

2. /foto命令处理：
   简化了原函数的冗余参数，仅完成状态设置和提示消息发送。

3. 照片处理逻辑：
   先校验用户状态，只有状态匹配时才执行保存操作；否则直接提示用户先触发命令。处理完成后删除用户状态，保证状态的准确性。

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