91 lines
3.8 KiB
Python
91 lines
3.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
数据库架构修复脚本
|
|||
|
|
添加缺失的字段和修复表结构
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
|
|||
|
|
# 添加项目根目录到Python路径
|
|||
|
|
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
|
|||
|
|
from src.core.database import db_manager
|
|||
|
|
from src.core.models import Base
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
logging.basicConfig(level=logging.INFO)
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
def fix_database_schema():
|
|||
|
|
"""修复数据库架构"""
|
|||
|
|
try:
|
|||
|
|
with db_manager.get_session() as session:
|
|||
|
|
# 检查并添加severity字段到alerts表
|
|||
|
|
try:
|
|||
|
|
# 检查字段是否存在
|
|||
|
|
result = session.execute(text("""
|
|||
|
|
SELECT COUNT(*) as count
|
|||
|
|
FROM information_schema.columns
|
|||
|
|
WHERE table_name = 'alerts' AND column_name = 'severity'
|
|||
|
|
"""))
|
|||
|
|
|
|||
|
|
if result.fetchone()[0] == 0:
|
|||
|
|
logger.info("添加severity字段到alerts表...")
|
|||
|
|
session.execute(text("ALTER TABLE alerts ADD COLUMN severity VARCHAR(20) DEFAULT 'medium'"))
|
|||
|
|
session.commit()
|
|||
|
|
logger.info("severity字段添加成功")
|
|||
|
|
else:
|
|||
|
|
logger.info("severity字段已存在")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"添加severity字段失败: {e}")
|
|||
|
|
# 如果是SQLite，尝试不同的方法
|
|||
|
|
try:
|
|||
|
|
session.execute(text("ALTER TABLE alerts ADD COLUMN severity VARCHAR(20) DEFAULT 'medium'"))
|
|||
|
|
session.commit()
|
|||
|
|
logger.info("severity字段添加成功（SQLite）")
|
|||
|
|
except Exception as e2:
|
|||
|
|
logger.error(f"SQLite添加severity字段也失败: {e2}")
|
|||
|
|
|
|||
|
|
# 检查并添加is_verified相关字段到knowledge_entries表
|
|||
|
|
try:
|
|||
|
|
result = session.execute(text("""
|
|||
|
|
SELECT COUNT(*) as count
|
|||
|
|
FROM information_schema.columns
|
|||
|
|
WHERE table_name = 'knowledge_entries' AND column_name = 'is_verified'
|
|||
|
|
"""))
|
|||
|
|
|
|||
|
|
if result.fetchone()[0] == 0:
|
|||
|
|
logger.info("添加is_verified字段到knowledge_entries表...")
|
|||
|
|
session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN is_verified BOOLEAN DEFAULT FALSE"))
|
|||
|
|
session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_by VARCHAR(100)"))
|
|||
|
|
session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_at DATETIME"))
|
|||
|
|
session.commit()
|
|||
|
|
logger.info("is_verified相关字段添加成功")
|
|||
|
|
else:
|
|||
|
|
logger.info("is_verified字段已存在")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.warning(f"添加is_verified字段失败: {e}")
|
|||
|
|
# SQLite方法
|
|||
|
|
try:
|
|||
|
|
session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN is_verified BOOLEAN DEFAULT FALSE"))
|
|||
|
|
session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_by VARCHAR(100)"))
|
|||
|
|
session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_at DATETIME"))
|
|||
|
|
session.commit()
|
|||
|
|
logger.info("is_verified相关字段添加成功（SQLite）")
|
|||
|
|
except Exception as e2:
|
|||
|
|
logger.error(f"SQLite添加is_verified字段也失败: {e2}")
|
|||
|
|
|
|||
|
|
logger.info("数据库架构修复完成")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"数据库架构修复失败: {e}")
|
|||
|
|
raise
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
fix_database_schema()
|