120 lines
4.0 KiB
Python
120 lines
4.0 KiB
Python
"""add profile resolver fields to config profiles and mounts
|
|
|
|
Revision ID: 0014_add_profile_resolver_fields
|
|
Revises: 0013_add_config_profiles
|
|
Create Date: 2026-05-24 14:00:00.000000
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = "0014_add_profile_resolver_fields"
|
|
down_revision: Union[str, None] = "0013_add_config_profiles"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Add fields to config_profiles
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("project_id", postgresql.UUID(as_uuid=True), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("tool_type_id", postgresql.UUID(as_uuid=True), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("environment_variables", sa.JSON(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("start_command", sa.Text(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("working_directory", sa.Text(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("port", sa.Integer(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_profiles",
|
|
sa.Column("is_default", sa.Boolean(), nullable=False, server_default="false"),
|
|
)
|
|
|
|
# Add foreign keys for project and tool_type
|
|
op.create_foreign_key(
|
|
"fk_config_profiles_project",
|
|
"config_profiles",
|
|
"projects",
|
|
["project_id"],
|
|
["id"],
|
|
ondelete="CASCADE",
|
|
)
|
|
op.create_foreign_key(
|
|
"fk_config_profiles_tool_type",
|
|
"config_profiles",
|
|
"tool_types",
|
|
["tool_type_id"],
|
|
["id"],
|
|
ondelete="CASCADE",
|
|
)
|
|
|
|
# Create indices
|
|
op.create_index("idx_config_profiles_project", "config_profiles", ["project_id"])
|
|
op.create_index("idx_config_profiles_tool_type", "config_profiles", ["tool_type_id"])
|
|
|
|
# Alter config_mounts: rename mount_path to target_path, add mode, change content to files JSON
|
|
op.alter_column("config_mounts", "mount_path", new_column_name="target_path")
|
|
op.add_column(
|
|
"config_mounts",
|
|
sa.Column("mode", sa.String(length=10), nullable=False, server_default="rw"),
|
|
)
|
|
op.add_column(
|
|
"config_mounts",
|
|
sa.Column("files", sa.JSON(), nullable=True),
|
|
)
|
|
# Drop the source_profile foreign key if it exists
|
|
op.drop_constraint(
|
|
"config_mounts_source_profile_id_fkey",
|
|
"config_mounts",
|
|
type_="foreignkey",
|
|
)
|
|
op.drop_column("config_mounts", "content")
|
|
op.drop_column("config_mounts", "source_profile_id")
|
|
|
|
|
|
def downgrade() -> None:
|
|
# Restore config_mounts
|
|
op.add_column(
|
|
"config_mounts",
|
|
sa.Column("source_profile_id", postgresql.UUID(as_uuid=True), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"config_mounts",
|
|
sa.Column("content", sa.Text(), nullable=True),
|
|
)
|
|
op.drop_column("config_mounts", "files")
|
|
op.drop_column("config_mounts", "mode")
|
|
op.alter_column("config_mounts", "target_path", new_column_name="mount_path")
|
|
|
|
# Restore config_profiles
|
|
op.drop_index("idx_config_profiles_tool_type", table_name="config_profiles")
|
|
op.drop_index("idx_config_profiles_project", table_name="config_profiles")
|
|
op.drop_constraint("fk_config_profiles_tool_type", "config_profiles", type_="foreignkey")
|
|
op.drop_constraint("fk_config_profiles_project", "config_profiles", type_="foreignkey")
|
|
op.drop_column("config_profiles", "is_default")
|
|
op.drop_column("config_profiles", "port")
|
|
op.drop_column("config_profiles", "working_directory")
|
|
op.drop_column("config_profiles", "start_command")
|
|
op.drop_column("config_profiles", "environment_variables")
|
|
op.drop_column("config_profiles", "tool_type_id")
|
|
op.drop_column("config_profiles", "project_id")
|