Files
ModelForge/backend/alembic/versions/20260826_0016_m11_project_integrations.py
T

66 lines
3.2 KiB
Python

"""Add operational project integrations and project-fit evidence.
Revision ID: 20260826_0016
Revises: 20260826_0015
"""
from collections.abc import Sequence
import sqlalchemy as sa
from alembic import op
revision: str = "20260826_0016"
down_revision: str | None = "20260826_0015"
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
op.add_column("service_clients", sa.Column("project_binding_id", sa.Uuid(), nullable=True))
op.add_column("service_clients", sa.Column("integration_environment", sa.String(32), nullable=True))
op.add_column("service_clients", sa.Column("purpose", sa.Text(), nullable=True))
op.create_foreign_key(
"fk_service_clients_project_binding",
"service_clients",
"project_bindings",
["project_binding_id"],
["id"],
ondelete="RESTRICT",
)
op.create_index("ix_service_clients_project_binding_id", "service_clients", ["project_binding_id"])
op.create_table(
"project_fit_evidence",
sa.Column("id", sa.Uuid(), nullable=False),
sa.Column("project_binding_id", sa.Uuid(), nullable=False),
sa.Column("capability_deployment_id", sa.Uuid(), nullable=True),
sa.Column("environment", sa.String(32), nullable=False),
sa.Column("recommendation", sa.String(32), nullable=False),
sa.Column("case_count", sa.Integer(), nullable=False),
sa.Column("metric_values", sa.JSON(), nullable=False),
sa.Column("critical_errors", sa.Integer(), nullable=False),
sa.Column("blockers", sa.JSON(), nullable=False),
sa.Column("evidence_digest", sa.String(64), nullable=False),
sa.Column("evidence", sa.JSON(), nullable=False),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
sa.ForeignKeyConstraint(["project_binding_id"], ["project_bindings.id"], ondelete="RESTRICT"),
sa.ForeignKeyConstraint(["capability_deployment_id"], ["capability_deployments.id"], ondelete="SET NULL"),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("evidence_digest", name="uq_project_fit_evidence_digest"),
)
op.create_index("ix_project_fit_evidence_project_binding_id", "project_fit_evidence", ["project_binding_id"])
op.create_index("ix_project_fit_evidence_capability_deployment_id", "project_fit_evidence", ["capability_deployment_id"])
op.create_index("ix_project_fit_evidence_recommendation", "project_fit_evidence", ["recommendation"])
def downgrade() -> None:
op.drop_index("ix_project_fit_evidence_recommendation", table_name="project_fit_evidence")
op.drop_index("ix_project_fit_evidence_capability_deployment_id", table_name="project_fit_evidence")
op.drop_index("ix_project_fit_evidence_project_binding_id", table_name="project_fit_evidence")
op.drop_table("project_fit_evidence")
op.drop_index("ix_service_clients_project_binding_id", table_name="service_clients")
op.drop_constraint("fk_service_clients_project_binding", "service_clients", type_="foreignkey")
op.drop_column("service_clients", "purpose")
op.drop_column("service_clients", "integration_environment")
op.drop_column("service_clients", "project_binding_id")