Files
ss-tools/backend/alembic/versions/z6a7b8c9d0e1_add_scenario_edit_proposals.py

46 lines
2.2 KiB
Python

# #region Alembic.ScenarioEditProposals [C:2] [TYPE Module] [SEMANTICS alembic,scenario,editor,proposal,agent]
# @ingroup Alembic
# @BRIEF Add 043 server-stored agent edit proposals table.
# @LAYER Database
"""add server-stored agent edit proposals (043 US5)"""
from collections.abc import Sequence
import sqlalchemy as sa
from alembic import op
revision: str = "z6a7b8c9d0e1"
down_revision: str | Sequence[str] | None = "y4z5a6b7c8d9"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
op.create_table(
"scenario_edit_proposals",
sa.Column("proposal_id", sa.String(length=36), nullable=False),
sa.Column("scenario_id", sa.String(length=36), nullable=False),
sa.Column("base_revision_id", sa.String(length=36), nullable=False),
sa.Column("request_text", sa.Text(), nullable=False),
sa.Column("operations", sa.JSON(), nullable=False),
sa.Column("proposed_graph", sa.JSON(), nullable=False),
sa.Column("digest", sa.String(length=64), nullable=False),
sa.Column("created_by", sa.String(length=128), nullable=False),
sa.Column("agent_action_id", sa.String(length=128), nullable=True),
sa.Column("status", sa.String(length=24), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["scenario_id"], ["scenario_registry_entries.scenario_id"], ondelete="CASCADE"),
sa.PrimaryKeyConstraint("proposal_id"),
)
op.create_index("ix_scenario_edit_proposals_scenario", "scenario_edit_proposals", ["scenario_id"])
op.create_index("ix_scenario_edit_proposals_base_revision", "scenario_edit_proposals", ["base_revision_id"])
op.create_index("ix_scenario_edit_proposals_status", "scenario_edit_proposals", ["status"])
def downgrade() -> None:
op.drop_index("ix_scenario_edit_proposals_status", table_name="scenario_edit_proposals")
op.drop_index("ix_scenario_edit_proposals_base_revision", table_name="scenario_edit_proposals")
op.drop_index("ix_scenario_edit_proposals_scenario", table_name="scenario_edit_proposals")
op.drop_table("scenario_edit_proposals")
# #endregion Alembic.ScenarioEditProposals