增加follow字段

product/update_database.py

@@ -0,0 +1,76 @@
+#!/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()