82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""
|
||
Database Migration: Scene Slot Bundle Tables.
|
||
[AC-SCENE-SLOT-01] 场景-槽位映射配置表迁移
|
||
|
||
创建时间: 2025-03-07
|
||
变更说明:
|
||
- 新增 scene_slot_bundles 表用于存储场景槽位包配置
|
||
|
||
执行方式:
|
||
- SQLModel 会自动创建表（通过 init_db）
|
||
- 此脚本用于手动迁移或回滚
|
||
|
||
SQL DDL:
|
||
```sql
|
||
CREATE TABLE scene_slot_bundles (
|
||
id UUID PRIMARY KEY,
|
||
tenant_id VARCHAR NOT NULL,
|
||
scene_key VARCHAR(100) NOT NULL,
|
||
scene_name VARCHAR(100) NOT NULL,
|
||
description TEXT,
|
||
required_slots JSON NOT NULL DEFAULT '[]',
|
||
optional_slots JSON NOT NULL DEFAULT '[]',
|
||
slot_priority JSON,
|
||
completion_threshold FLOAT NOT NULL DEFAULT 1.0,
|
||
ask_back_order VARCHAR NOT NULL DEFAULT 'priority',
|
||
status VARCHAR NOT NULL DEFAULT 'draft',
|
||
version INTEGER NOT NULL DEFAULT 1,
|
||
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
||
updated_at TIMESTAMP NOT NULL DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX ix_scene_slot_bundles_tenant ON scene_slot_bundles(tenant_id);
|
||
CREATE UNIQUE INDEX ix_scene_slot_bundles_tenant_scene ON scene_slot_bundles(tenant_id, scene_key);
|
||
CREATE INDEX ix_scene_slot_bundles_tenant_status ON scene_slot_bundles(tenant_id, status);
|
||
```
|
||
|
||
回滚 SQL:
|
||
```sql
|
||
DROP TABLE IF EXISTS scene_slot_bundles;
|
||
```
|
||
"""
|
||
|
||
SCENE_SLOT_BUNDLES_DDL = """
|
||
CREATE TABLE IF NOT EXISTS scene_slot_bundles (
|
||
id UUID PRIMARY KEY,
|
||
tenant_id VARCHAR NOT NULL,
|
||
scene_key VARCHAR(100) NOT NULL,
|
||
scene_name VARCHAR(100) NOT NULL,
|
||
description TEXT,
|
||
required_slots JSON NOT NULL DEFAULT '[]',
|
||
optional_slots JSON NOT NULL DEFAULT '[]',
|
||
slot_priority JSON,
|
||
completion_threshold FLOAT NOT NULL DEFAULT 1.0,
|
||
ask_back_order VARCHAR NOT NULL DEFAULT 'priority',
|
||
status VARCHAR NOT NULL DEFAULT 'draft',
|
||
version INTEGER NOT NULL DEFAULT 1,
|
||
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
||
updated_at TIMESTAMP NOT NULL DEFAULT NOW()
|
||
);
|
||
"""
|
||
|
||
SCENE_SLOT_BUNDLES_INDEXES = """
|
||
CREATE INDEX IF NOT EXISTS ix_scene_slot_bundles_tenant ON scene_slot_bundles(tenant_id);
|
||
CREATE UNIQUE INDEX IF NOT EXISTS ix_scene_slot_bundles_tenant_scene ON scene_slot_bundles(tenant_id, scene_key);
|
||
CREATE INDEX IF NOT EXISTS ix_scene_slot_bundles_tenant_status ON scene_slot_bundles(tenant_id, status);
|
||
"""
|
||
|
||
SCENE_SLOT_BUNDLES_ROLLBACK = """
|
||
DROP TABLE IF EXISTS scene_slot_bundles;
|
||
"""
|
||
|
||
|
||
async def upgrade(conn):
|
||
"""执行迁移"""
|
||
await conn.execute(SCENE_SLOT_BUNDLES_DDL)
|
||
await conn.execute(SCENE_SLOT_BUNDLES_INDEXES)
|
||
|
||
|
||
async def downgrade(conn):
|
||
"""回滚迁移"""
|
||
await conn.execute(SCENE_SLOT_BUNDLES_ROLLBACK)
|