46 lines
1.4 KiB
Python
Executable File
46 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
数据库迁移：将 template_id 改为可选字段
|
|
"""
|
|
import os
|
|
from sqlalchemy import create_engine, text
|
|
|
|
# 数据库连接
|
|
DATABASE_URL = "postgresql://taiji:By%40123456.@taijipda.postgres.database.azure.com:5432/taijiagnet"
|
|
|
|
print("🔄 开始数据库迁移...")
|
|
print(f"数据库: {DATABASE_URL.split('@')[1]}")
|
|
|
|
engine = create_engine(DATABASE_URL)
|
|
|
|
try:
|
|
with engine.connect() as conn:
|
|
# 修改 template_id 列为可空
|
|
print("\n1️⃣ 修改 template_id 为可空字段...")
|
|
conn.execute(text("""
|
|
ALTER TABLE agents
|
|
ALTER COLUMN template_id DROP NOT NULL;
|
|
"""))
|
|
conn.commit()
|
|
print("✅ template_id 已改为可空")
|
|
|
|
# 验证更改
|
|
print("\n2️⃣ 验证表结构...")
|
|
result = conn.execute(text("""
|
|
SELECT column_name, is_nullable, data_type
|
|
FROM information_schema.columns
|
|
WHERE table_name = 'agents'
|
|
AND column_name = 'template_id';
|
|
"""))
|
|
|
|
for row in result:
|
|
print(f"✅ {row[0]}: nullable={row[1]}, type={row[2]}")
|
|
|
|
print("\n✅ 数据库迁移成功完成！")
|
|
print("\n现在可以创建没有 template_id 的 Agent 了。")
|
|
|
|
except Exception as e:
|
|
print(f"\n❌ 迁移失败: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|