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