PCM_Report/add_script_data_column.py

46 lines
1.4 KiB
Python
Raw Normal View History

2025-12-11 14:32:31 +08:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
为experiments表添加script_data字段用于存储动态脚本返回的数据
"""
import sqlite3
from pathlib import Path
def add_script_data_column():
"""添加script_data字段到experiments表"""
db_path = Path(__file__).parent / "experiments.db"
try:
conn = sqlite3.connect(str(db_path))
cursor = conn.cursor()
# 检查字段是否已存在
cursor.execute("PRAGMA table_info(experiments)")
columns = [row[1] for row in cursor.fetchall()]
if 'script_data' in columns:
print("✅ script_data字段已存在无需添加")
else:
# 添加script_data字段
cursor.execute("""
ALTER TABLE experiments
ADD COLUMN script_data TEXT
""")
conn.commit()
print("✅ 成功添加script_data字段到experiments表")
# 显示更新后的表结构
cursor.execute("PRAGMA table_info(experiments)")
print("\n更新后的表结构:")
for row in cursor.fetchall():
print(f" {row[1]}: {row[2]} {row[3] if row[3] else ''}")
conn.close()
except Exception as e:
print(f"❌ 添加字段失败: {e}")
raise
if __name__ == "__main__":
add_script_data_column()