在Python Telegram Bot的开发场景中，机器人重启后用户状态丢失是常见的问题。默认情况下机器人运行时的用户会话数据存储在内存中，一旦进程重启内存数据就会被清空，用户之前的操作进度、临时配置等信息都会消失。要解决这个问题，核心是将用户状态数据从内存转移到可持久化的存储介质中，重启后从存储介质重新加载数据即可。
常见状态保持方案对比
目前主流的实现方案有三种，各自的适用场景和优缺点如下：
| 方案类型 | 存储位置 | 优点 | 缺点 | 适用场景 |
|---|---|---|---|---|
| 内存存储 | 程序运行内存 | 读写速度快，实现简单 | 重启后数据丢失，无法多进程共享 | 临时测试，无状态要求的简单机器人 |
| 文件存储 | 本地JSON/DB文件 | 实现简单，不需要额外服务 | 并发读写容易出错，大数据量性能差 | 个人小型机器人，用户量少的场景 |
| 数据库存储 | SQLite/MySQL/Redis等 | 支持并发读写，数据可靠，支持多进程 | 需要额外配置存储服务，实现稍复杂 | 生产环境，用户量较多的正式机器人 |
基于SQLite的持久化状态实现
SQLite是轻量级的文件数据库，不需要单独部署服务，非常适合中小型Telegram Bot的状态存储需求。我们可以通过Python内置的sqlite3模块实现用户状态的增删改查。
1. 初始化数据库
首先创建存储用户状态的表，包含用户ID、状态键、状态值、更新时间等字段：
import sqlite3
from datetime import datetime
def init_db(db_path="bot_state.db"):
# 连接数据库，文件不存在会自动创建
conn = sqlite3.connect(db_path)
cursor = conn.cursor()
# 创建用户状态表
cursor.execute("""
CREATE TABLE IF NOT EXISTS user_state (
user_id INTEGER NOT NULL,
state_key TEXT NOT NULL,
state_value TEXT,
update_time TEXT NOT NULL,
PRIMARY KEY (user_id, state_key)
)
""")
conn.commit()
conn.close()
# 初始化数据库
init_db()
2. 状态读写工具类
封装状态读写的通用方法，方便在机器人处理逻辑中调用：
import json
import sqlite3
from datetime import datetime
class StateManager:
def __init__(self, db_path="bot_state.db"):
self.db_path = db_path
def _get_conn(self):
return sqlite3.connect(self.db_path)
def set_state(self, user_id, key, value):
# 存储用户状态，支持存储字典等复杂类型，转成JSON字符串存储
conn = self._get_conn()
cursor = conn.cursor()
state_value = json.dumps(value, ensure_ascii=False)
update_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
cursor.execute("""
INSERT OR REPLACE INTO user_state (user_id, state_key, state_value, update_time)
VALUES (?, ?, ?, ?)
""", (user_id, key, state_value, update_time))
conn.commit()
conn.close()
def get_state(self, user_id, key, default=None):
# 获取用户状态，反序列化JSON字符串
conn = self._get_conn()
cursor = conn.cursor()
cursor.execute("""
SELECT state_value FROM user_state WHERE user_id = ? AND state_key = ?
""", (user_id, key))
result = cursor.fetchone()
conn.close()
if result:
return json.loads(result[0])
return default
def delete_state(self, user_id, key):
# 删除指定用户状态
conn = self._get_conn()
cursor = conn.cursor()
cursor.execute("""
DELETE FROM user_state WHERE user_id = ? AND state_key = ?
""", (user_id, key))
conn.commit()
conn.close()
3. 在Bot逻辑中使用状态管理
结合python-telegram-bot库的处理逻辑，在用户发送消息时读取和恢复状态：
from telegram import Update
from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters, ContextTypes
from state_manager import StateManager
# 初始化状态管理器
state_manager = StateManager()
# 处理/start命令
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
user_id = update.effective_user.id
# 设置用户初始状态
state_manager.set_state(user_id, "step", "init")
await update.message.reply_text("欢迎使用机器人，请发送任意内容开始流程")
# 处理普通消息
async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
user_id = update.effective_user.id
# 获取用户当前步骤状态
current_step = state_manager.get_state(user_id, "step", "init")
user_text = update.message.text
if current_step == "init":
# 第一步处理
state_manager.set_state(user_id, "step", "step1")
state_manager.set_state(user_id, "input_data", user_text)
await update.message.reply_text(f"你输入的内容是：{user_text}，请继续操作")
elif current_step == "step1":
# 第二步处理，读取之前存储的输入数据
prev_data = state_manager.get_state(user_id, "input_data")
await update.message.reply_text(f"之前输入的内容：{prev_data}，当前输入：{user_text}")
# 流程结束，清除状态
state_manager.delete_state(user_id, "step")
state_manager.delete_state(user_id, "input_data")
def main():
# 替换为你的Bot Token
app = ApplicationBuilder().token("YOUR_BOT_TOKEN").build()
app.add_handler(CommandHandler("start", start))
app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
app.run_polling()
if __name__ == "__main__":
main()
注意事项
- 存储复杂类型数据时，建议使用JSON序列化后再存储，读取时反序列化，避免类型丢失
- 如果机器人使用多进程部署，不建议使用SQLite，优先选择Redis或MySQL这类支持多连接的数据库
- 定期清理过期状态数据，避免无用数据占用存储空间，比如可以只保留7天内的用户状态
- 状态键的命名建议加上模块前缀，避免不同功能的状态键冲突，比如
order_step、user_config
总结
实现Python Telegram Bot重启后保持用户状态的核心是将状态数据持久化存储，根据项目规模选择合适的存储方案即可。小型项目用SQLite足够满足需求，正式生产环境建议使用Redis或MySQL保证数据可靠性和并发性能。通过以上方案，即使机器人重启，用户之前的操作状态也可以正常恢复，避免用户重复操作，提升使用体验。
Python_Telegram_Bot用户状态保持持久化存储重启恢复会话管理修改时间：2026-06-21 11:54:20