#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
TSP助手数据库初始化脚本 - 包含所有数据库操作
"""

import sys
import os
import logging
from sqlalchemy import text
from datetime import datetime

# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from src.config.config import Config
from src.utils.helpers import setup_logging
from src.core.database import db_manager
from src.core.models import Base, WorkOrder, KnowledgeEntry, Conversation, Analytics, Alert, VehicleData

def init_database():
    """初始化数据库 - 包含所有数据库操作"""
    print("=" * 60)
    print("🚀 TSP智能助手数据库初始化")
    print("=" * 60)
    
    try:
        # 设置日志
        setup_logging(Config.LOG_LEVEL, Config.LOG_FILE)
        logger = logging.getLogger(__name__)
        
        # 测试数据库连接
        if not db_manager.test_connection():
            print("❌ 数据库连接失败")
            return False
        
        print("✅ 数据库连接成功")
        
        # 创建所有表
        print("\n📋 创建数据库表...")
        Base.metadata.create_all(bind=db_manager.engine)
        print("✅ 数据库表创建成功")
        
        # 执行数据库迁移（添加新字段和表）
        print("\n🔄 执行数据库迁移...")
        migrate_database()
        
        # 插入初始数据
        print("\n📊 插入初始数据...")
        insert_initial_data()
        
        # 添加示例车辆数据
        print("\n🚗 添加示例车辆数据...")
        add_sample_vehicle_data()
        
        # 验证知识库条目
        print("\n🔍 验证知识库条目...")
        verify_existing_knowledge()
        
        print("\n✅ 数据库初始化完成")
        return True
        
    except Exception as e:
        print(f"❌ 数据库初始化失败: {e}")
        return False

def migrate_database():
    """执行数据库迁移 - 添加新字段和表"""
    try:
        with db_manager.get_session() as session:
            # 检查数据库类型
            db_url = db_manager.engine.url
            is_mysql = 'mysql' in str(db_url)
            is_sqlite = 'sqlite' in str(db_url)
            
            print("   📝 检查知识库验证字段...")
            
            # 检查is_verified字段是否存在
            if is_mysql:
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM INFORMATION_SCHEMA.COLUMNS 
                    WHERE TABLE_SCHEMA = DATABASE() 
                    AND TABLE_NAME = 'knowledge_entries' 
                    AND COLUMN_NAME = 'is_verified'
                """)).fetchone()
            else:  # SQLite
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM pragma_table_info('knowledge_entries') 
                    WHERE name = 'is_verified'
                """)).fetchone()
            
            if result.count == 0:
                print("      ➕ 添加is_verified字段...")
                if is_mysql:
                    session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN is_verified BOOLEAN DEFAULT FALSE"))
                else:
                    session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN is_verified BOOLEAN DEFAULT FALSE"))
                print("      ✅ is_verified字段添加成功")
            else:
                print("      ✅ is_verified字段已存在")
            
            # 检查verified_by字段是否存在
            if is_mysql:
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM INFORMATION_SCHEMA.COLUMNS 
                    WHERE TABLE_SCHEMA = DATABASE() 
                    AND TABLE_NAME = 'knowledge_entries' 
                    AND COLUMN_NAME = 'verified_by'
                """)).fetchone()
            else:  # SQLite
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM pragma_table_info('knowledge_entries') 
                    WHERE name = 'verified_by'
                """)).fetchone()
            
            if result.count == 0:
                print("      ➕ 添加verified_by字段...")
                if is_mysql:
                    session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_by VARCHAR(100)"))
                else:
                    session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_by VARCHAR(100)"))
                print("      ✅ verified_by字段添加成功")
            else:
                print("      ✅ verified_by字段已存在")
            
            # 检查verified_at字段是否存在
            if is_mysql:
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM INFORMATION_SCHEMA.COLUMNS 
                    WHERE TABLE_SCHEMA = DATABASE() 
                    AND TABLE_NAME = 'knowledge_entries' 
                    AND COLUMN_NAME = 'verified_at'
                """)).fetchone()
            else:  # SQLite
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM pragma_table_info('knowledge_entries') 
                    WHERE name = 'verified_at'
                """)).fetchone()
            
            if result.count == 0:
                print("      ➕ 添加verified_at字段...")
                if is_mysql:
                    session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_at DATETIME"))
                else:
                    session.execute(text("ALTER TABLE knowledge_entries ADD COLUMN verified_at DATETIME"))
                print("      ✅ verified_at字段添加成功")
            else:
                print("      ✅ verified_at字段已存在")
            
            # 检查车辆数据表是否存在
            if is_mysql:
                result = session.execute(text("""
                    SELECT COUNT(*) as count 
                    FROM INFORMATION_SCHEMA.TABLES 
                    WHERE TABLE_SCHEMA = DATABASE() 
                    AND TABLE_NAME = 'vehicle_data'
                """)).fetchone()
            else:  # SQLite
                result = session.execute(text("""
                    SELECT name FROM sqlite_master 
                    WHERE type='table' AND name='vehicle_data'
                """)).fetchone()
            
            if (is_mysql and result.count == 0) or (not is_mysql and not result):
                print("      ➕ 创建vehicle_data表...")
                VehicleData.__table__.create(session.bind, checkfirst=True)
                print("      ✅ vehicle_data表创建成功")
            else:
                print("      ✅ vehicle_data表已存在")
            
            session.commit()
            print("   ✅ 数据库迁移完成")
            
    except Exception as e:
        print(f"   ❌ 数据库迁移失败: {e}")
        return False
    
    return True

def insert_initial_data():
    """插入初始数据"""
    try:
        with db_manager.get_session() as session:
            # 检查是否已有数据
            existing_entries = session.query(KnowledgeEntry).count()
            if existing_entries > 0:
                print("   ✅ 数据库中已有数据，跳过初始数据插入")
                return
            
            # 插入示例知识库条目
            initial_knowledge = [
                {
                    "question": "如何重置密码？",
                    "answer": "您可以通过以下步骤重置密码：1. 点击登录页面的'忘记密码'链接 2. 输入您的邮箱地址 3. 检查邮箱并点击重置链接 4. 设置新密码",
                    "category": "账户问题",
                    "confidence_score": 0.9,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "账户被锁定了怎么办？",
                    "answer": "如果您的账户被锁定，请尝试以下解决方案：1. 等待15分钟后重试登录 2. 如果问题持续，请联系客服并提供您的用户ID",
                    "category": "账户问题",
                    "confidence_score": 0.8,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "如何修改个人信息？",
                    "answer": "您可以在个人设置页面修改个人信息：1. 登录后点击右上角的个人头像 2. 选择'个人设置' 3. 修改相关信息并保存",
                    "category": "账户问题",
                    "confidence_score": 0.7,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "支付失败怎么办？",
                    "answer": "如果支付失败，请检查：1. 银行卡余额是否充足 2. 银行卡是否支持在线支付 3. 网络连接是否正常 4. 如果问题持续，请联系支付客服",
                    "category": "支付问题",
                    "confidence_score": 0.8,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "如何申请退款？",
                    "answer": "申请退款流程：1. 在订单详情页面点击'申请退款' 2. 选择退款原因 3. 填写退款说明 4. 提交申请后等待审核",
                    "category": "支付问题",
                    "confidence_score": 0.7,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "系统无法访问怎么办？",
                    "answer": "如果系统无法访问，请尝试：1. 检查网络连接 2. 清除浏览器缓存 3. 尝试使用其他浏览器 4. 如果问题持续，请联系技术支持",
                    "category": "技术问题",
                    "confidence_score": 0.8,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "如何联系客服？",
                    "answer": "您可以通过以下方式联系客服：1. 在线客服：点击页面右下角的客服图标 2. 电话客服：400-123-4567 3. 邮箱客服：support@example.com",
                    "category": "服务问题",
                    "confidence_score": 0.9,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "如何远程启动车辆？",
                    "answer": "远程启动车辆需要满足以下条件：1. 车辆处于P档 2. 手刹拉起 3. 车门已锁 4. 电池电量充足。操作步骤：打开APP → 点击远程启动按钮 → 确认启动条件 → 等待启动完成",
                    "category": "远程控制",
                    "confidence_score": 0.9,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "APP显示车辆信息错误怎么办？",
                    "answer": "如果APP显示车辆信息错误，请尝试：1. 重新登录APP 2. 重启车辆系统 3. 检查网络连接 4. 如果问题持续，请联系客服",
                    "category": "APP功能",
                    "confidence_score": 0.8,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                },
                {
                    "question": "车辆无法远程启动的原因？",
                    "answer": "车辆无法远程启动的常见原因：1. 车辆不在P档 2. 手刹未拉起 3. 车门未锁 4. 电池电量不足 5. 网络信号差 6. 车辆系统故障",
                    "category": "远程控制",
                    "confidence_score": 0.9,
                    "is_verified": True,
                    "verified_by": "system",
                    "verified_at": datetime.now()
                }
            ]
            
            for knowledge in initial_knowledge:
                entry = KnowledgeEntry(**knowledge)
                session.add(entry)
            
            session.commit()
            print(f"   ✅ 成功插入 {len(initial_knowledge)} 条知识库条目")
            
    except Exception as e:
        print(f"   ❌ 插入初始数据失败: {e}")

def add_sample_vehicle_data():
    """添加示例车辆数据"""
    try:
        from src.vehicle.vehicle_data_manager import VehicleDataManager
        
        vehicle_manager = VehicleDataManager()
        success = vehicle_manager.add_sample_vehicle_data()
        
        if success:
            print("   ✅ 示例车辆数据添加成功")
        else:
            print("   ❌ 示例车辆数据添加失败")
            
    except Exception as e:
        print(f"   ❌ 添加示例车辆数据失败: {e}")

def verify_existing_knowledge():
    """验证现有的知识库条目"""
    try:
        with db_manager.get_session() as session:
            # 获取所有未验证的知识库条目
            unverified_entries = session.query(KnowledgeEntry).filter(
                KnowledgeEntry.is_verified == False
            ).all()
            
            if unverified_entries:
                print(f"   📝 发现 {len(unverified_entries)} 条未验证的知识库条目")
                
                # 将现有的知识库条目标记为已验证
                for entry in unverified_entries:
                    entry.is_verified = True
                    entry.verified_by = "system_init"
                    entry.verified_at = datetime.now()
                
                session.commit()
                print(f"   ✅ 成功验证 {len(unverified_entries)} 条知识库条目")
            else:
                print("   ✅ 所有知识库条目已验证")
                
    except Exception as e:
        print(f"   ❌ 验证知识库条目失败: {e}")

def check_database_status():
    """检查数据库状态"""
    print("\n" + "=" * 60)
    print("📊 数据库状态检查")
    print("=" * 60)
    
    try:
        with db_manager.get_session() as session:
            # 检查各表的记录数
            work_orders_count = session.query(WorkOrder).count()
            conversations_count = session.query(Conversation).count()
            knowledge_entries_count = session.query(KnowledgeEntry).count()
            verified_knowledge_count = session.query(KnowledgeEntry).filter(KnowledgeEntry.is_verified == True).count()
            unverified_knowledge_count = session.query(KnowledgeEntry).filter(KnowledgeEntry.is_verified == False).count()
            analytics_count = session.query(Analytics).count()
            alerts_count = session.query(Alert).count()
            vehicle_data_count = session.query(VehicleData).count()
            
            print(f"📋 工单表记录数: {work_orders_count}")
            print(f"💬 对话表记录数: {conversations_count}")
            print(f"📚 知识库表记录数: {knowledge_entries_count}")
            print(f"   - 已验证: {verified_knowledge_count}")
            print(f"   - 未验证: {unverified_knowledge_count}")
            print(f"📊 分析表记录数: {analytics_count}")
            print(f"🚨 预警表记录数: {alerts_count}")
            print(f"🚗 车辆数据表记录数: {vehicle_data_count}")
            
            # 检查车辆数据详情
            if vehicle_data_count > 0:
                vehicle_ids = session.query(VehicleData.vehicle_id).distinct().all()
                print(f"   - 车辆数量: {len(vehicle_ids)}")
                for vehicle_id in vehicle_ids[:3]:  # 显示前3个车辆
                    vehicle_data_types = session.query(VehicleData.data_type).filter(
                        VehicleData.vehicle_id == vehicle_id[0]
                    ).distinct().all()
                    print(f"   - 车辆 {vehicle_id[0]}: {len(vehicle_data_types)} 种数据类型")
            
            print("\n✅ 数据库状态检查完成")
            
    except Exception as e:
        print(f"❌ 数据库状态检查失败: {e}")

def main():
    """主函数"""
    print("🚀 TSP智能助手数据库初始化工具")
    print("=" * 60)
    
    # 初始化数据库
    if init_database():
        # 检查数据库状态
        check_database_status()
        
        print("\n" + "=" * 60)
        print("🎉 数据库初始化成功！")
        print("=" * 60)
        print("✅ 已完成的操作：")
        print("   - 创建所有数据库表")
        print("   - 添加知识库验证字段")
        print("   - 创建车辆数据表")
        print("   - 插入初始知识库数据")
        print("   - 添加示例车辆数据")
        print("   - 验证所有知识库条目")
        print("\n🚀 现在您可以运行以下命令启动系统：")
        print("   python start_dashboard.py")
        print("\n🧪 或运行功能测试：")
        print("   python test_new_features.py")
        print("\n📋 新功能包括：")
        print("   - 知识库分页显示")
        print("   - 知识库验证机制")
        print("   - 车辆实时数据管理")
        print("   - 文件上传生成知识库")
        print("   - 智能对话结合车辆数据")
    else:
        print("\n" + "=" * 60)
        print("❌ 数据库初始化失败！")
        print("=" * 60)
        print("请检查：")
        print("1. 数据库文件权限")
        print("2. SQLite是否已安装")
        print("3. 磁盘空间是否充足")
        print("4. Python依赖库是否完整")

if __name__ == "__main__":
    main()