77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
#!/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()
|