36 lines
859 B
Python
36 lines
859 B
Python
"""
|
|
数据库迁移脚本：为 message 表添加 mentions 字段
|
|
"""
|
|
import sqlite3
|
|
import os
|
|
|
|
db_path = 'instance/database.db'
|
|
|
|
if not os.path.exists(db_path):
|
|
print(f"错误：找不到数据库文件 {db_path}")
|
|
exit(1)
|
|
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
print("开始数据库迁移...\n")
|
|
|
|
# 检查 message 表
|
|
print("检查 message 表...")
|
|
cursor.execute("PRAGMA table_info(message)")
|
|
columns = [row[1] for row in cursor.fetchall()]
|
|
print(f"现有列: {columns}")
|
|
|
|
if 'mentions' not in columns:
|
|
print("添加 mentions 字段...")
|
|
cursor.execute("ALTER TABLE message ADD COLUMN mentions TEXT")
|
|
print("[OK] 已添加 mentions 字段")
|
|
else:
|
|
print("[OK] mentions 字段已存在")
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
print("\n[SUCCESS] 数据库迁移完成！")
|
|
print("现在可以正常使用聊天功能了。")
|