Python Telebot开发实战：构建智能图书推荐与管理机器人

pip install pyTelegramBotAPI

import random
import telebot
from telebot import types
import pickle
import os # 用于检查文件是否存在

# 替换为你的Bot Token
# 建议将Token存储在环境变量中，以提高安全性
BOT_TOKEN = os.getenv("TELEBOT_TOKEN", "YOUR_BOT_TOKEN_HERE")
bot = telebot.TeleBot(BOT_TOKEN)

# 定义数据文件路径
DATA_FILE = "book_list.p"

# 尝试从文件中加载图书列表，如果文件不存在或加载失败，则初始化为空列表
try:
    with open(DATA_FILE, "rb") as file:
        books = pickle.load(file)
    print(f"成功加载图书列表，当前有 {len(books)} 本书。")
except FileNotFoundError:
    books = []
    print("图书列表文件不存在，已初始化为空列表。")
except Exception as e:
    books = []
    print(f"加载图书列表失败: {e}，已初始化为空列表。")

# 保存图书列表到文件
def save_books():
    try:
        with open(DATA_FILE, "wb") as file:
            pickle.dump(books, file)
        print("图书列表已保存。")
    except Exception as e:
        print(f"保存图书列表失败: {e}")

@bot.message_handler(commands=['start', 'help'])
def send_welcome(message):
    markup = types.ReplyKeyboardMarkup(resize_keyboard=True)
    item1 = types.KeyboardButton("选择图书")
    item2 = types.KeyboardButton("清空列表")
    item3 = types.KeyboardButton("添加图书")
    markup.add(item1, item2, item3) # 可以一次添加多个按钮
    bot.send_message(message.chat.id,
                     "欢迎使用图书管理机器人！\n"
                     "点击'选择图书'获取推荐。\n"
                     "点击'添加图书'来增加新书。\n"
                     "点击'清空列表'删除所有图书。",
                     reply_markup=markup)

# 用于处理“选择图书”和“清空列表”的通用文本消息
@bot.message_handler(content_types="text")
def handle_text_messages(message):
    global books # 声明全局变量以修改

    if message.text == "选择图书":
        if books:
            bot.send_message(message.chat.id, random.choice(books))
        else:
            bot.send_message(message.chat.id, "图书列表为空，请先添加图书。")
    elif message.text == "清空列表":
        books.clear()
        save_books() # 调用保存函数
        bot.send_message(message.chat.id, "图书列表已清空。")
    elif message.text == "添加图书":
        bot.send_message(message.chat.id, "请输入您想添加的图书名称：")
        # 注册下一步处理函数，等待用户输入图书名称
        bot.register_next_step_handler(message, add_book_step)
    else:
        # 对于不匹配任何已知命令的文本，可以给出提示
        bot.send_message(message.chat.id, "抱歉，我无法理解您的指令。请使用键盘按钮或输入 /start。")

# 原始的、有问题的逻辑示例
# if message.text == "Add":
#     bot.send_message(message.chat.id, "Type the book name")
# else: # 这个else会捕获所有不匹配前述条件的文本，包括用户在“Add”之后输入的书名，但也包括任何其他不相关的消息。
#     new_book = message.text
#     books.append(new_book)
#     # ...

# 标记词解决方案示例 (简单但有局限性)
# elif "book" in message.text.lower(): # 假设用户会输入 "book 你的书名"
#     book_name = message.text.lower().replace("book", "").strip()
#     if book_name:
#         books.append(book_name)
#         save_books()
#         bot.send_message(message.chat.id, f"图书 '{book_name}' 已添加。")
#     else:
#         bot.send_message(message.chat.id, "请输入有效的图书名称。")

def add_book_step(message):
    global books
    new_book = message.text.strip() # 清除首尾空白

    if new_book: # 确保书名不为空
        books.append(new_book)
        save_books() # 调用保存函数
        bot.send_message(message.chat.id, f"图书 '{new_book}' 已成功添加到列表。")
    else:
        bot.send_message(message.chat.id, "图书名称不能为空，请重新点击'添加图书'并输入。")

    # 如果需要，可以在这里重新发送键盘或提示用户下一步操作
    send_welcome(message) # 重新显示键盘

if __name__ == '__main__':
    print("Bot is running...")
    # 持续监听消息，none_stop=True 确保机器人不会在连接中断时停止
    bot.polling(none_stop=True, interval=0)