#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
数据库更新脚本
用于更新现有的product_analysis表，添加follows字段
"""

import sqlite3
import os
from loguru import logger
import sys

# 配置日志
logger.remove()
logger.add(sys.stderr, level="INFO", format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>")

def update_product_analysis_table(db_path):
    """更新product_analysis表，添加follows字段"""
    logger.info(f"开始更新数据库: {db_path}")
    
    try:
        # 连接数据库
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        
        # 检查product_analysis表是否存在
        cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='product_analysis'")
        table_exists = cursor.fetchone() is not None
        
        if not table_exists:
            logger.error("product_analysis表不存在，无法更新")
            conn.close()
            return False
        
        # 检查follows字段是否已经存在
        cursor.execute("PRAGMA table_info(product_analysis)")
        columns = [col[1] for col in cursor.fetchall()]
        
        if 'follows' in columns:
            logger.info("follows字段已经存在，无需更新")
            conn.close()
            return True
        
        # 添加follows字段
        cursor.execute("ALTER TABLE product_analysis ADD COLUMN follows INTEGER")
        conn.commit()
        
        logger.success("成功为product_analysis表添加follows字段")
        conn.close()
        return True
        
    except Exception as e:
        logger.error(f"更新数据库失败: {e}")
        if 'conn' in locals():
            conn.close()
        return False

def main():
    """主函数"""
    # 获取默认数据库路径
    default_db_path = os.path.join(os.path.dirname(__file__), "products.db")
    
    logger.info("=== 数据库更新脚本开始执行 ===")
    
    # 更新数据库
    success = update_product_analysis_table(default_db_path)
    
    if success:
        logger.success("=== 数据库更新成功 ===")
        sys.exit(0)
    else:
        logger.error("=== 数据库更新失败 ===")
        sys.exit(1)

if __name__ == "__main__":
    main()