86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
数据库迁移脚本
|
||
为tasks表添加新字段：paused_at, retry_count, last_error, articles_only
|
||
"""
|
||
import sqlite3
|
||
import os
|
||
import logging
|
||
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
def migrate_database():
|
||
"""执行数据库迁移"""
|
||
db_path = "data/baijiahao.db"
|
||
|
||
if not os.path.exists(db_path):
|
||
logger.info("数据库文件不存在，无需迁移")
|
||
return
|
||
|
||
try:
|
||
conn = sqlite3.connect(db_path)
|
||
cursor = conn.cursor()
|
||
|
||
# 检查表是否存在
|
||
cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='tasks'")
|
||
if not cursor.fetchone():
|
||
logger.info("tasks表不存在，无需迁移")
|
||
conn.close()
|
||
return
|
||
|
||
# 获取当前表结构
|
||
cursor.execute("PRAGMA table_info(tasks)")
|
||
columns = {row[1]: row for row in cursor.fetchall()}
|
||
|
||
logger.info("开始数据库迁移...")
|
||
|
||
# 添加 paused_at 字段
|
||
if 'paused_at' not in columns:
|
||
logger.info("添加 paused_at 字段...")
|
||
cursor.execute("ALTER TABLE tasks ADD COLUMN paused_at TEXT")
|
||
logger.info("✓ paused_at 字段添加成功")
|
||
else:
|
||
logger.info("✓ paused_at 字段已存在")
|
||
|
||
# 添加 retry_count 字段
|
||
if 'retry_count' not in columns:
|
||
logger.info("添加 retry_count 字段...")
|
||
cursor.execute("ALTER TABLE tasks ADD COLUMN retry_count INTEGER DEFAULT 0")
|
||
logger.info("✓ retry_count 字段添加成功")
|
||
else:
|
||
logger.info("✓ retry_count 字段已存在")
|
||
|
||
# 添加 last_error 字段
|
||
if 'last_error' not in columns:
|
||
logger.info("添加 last_error 字段...")
|
||
cursor.execute("ALTER TABLE tasks ADD COLUMN last_error TEXT")
|
||
logger.info("✓ last_error 字段添加成功")
|
||
else:
|
||
logger.info("✓ last_error 字段已存在")
|
||
|
||
# 添加 articles_only 字段
|
||
if 'articles_only' not in columns:
|
||
logger.info("添加 articles_only 字段...")
|
||
cursor.execute("ALTER TABLE tasks ADD COLUMN articles_only INTEGER DEFAULT 1")
|
||
logger.info("✓ articles_only 字段添加成功")
|
||
else:
|
||
logger.info("✓ articles_only 字段已存在")
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
|
||
logger.info("=" * 50)
|
||
logger.info("✅ 数据库迁移完成！")
|
||
logger.info("=" * 50)
|
||
|
||
except Exception as e:
|
||
logger.error(f"❌ 数据库迁移失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
raise
|
||
|
||
if __name__ == "__main__":
|
||
migrate_database()
|