DB migration

Author: schustmi

src/zenml/zen_stores/migrations/versions/af27025fe19c_dynamic_pipelines.py

@@ -0,0 +1,77 @@
+"""Dynamic pipelines [af27025fe19c].
+
+Revision ID: af27025fe19c
+Revises: 0.91.0
+Create Date: 2025-10-27 13:23:40.442485
+
+"""
+
+import sqlalchemy as sa
+import sqlmodel
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "af27025fe19c"
+down_revision = "0.91.0"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("pipeline_snapshot", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column("is_dynamic", sa.Boolean(), nullable=False)
+        )
+
+    with op.batch_alter_table("step_configuration", schema=None) as batch_op:
+        # TODO: missing check constraint
+        batch_op.add_column(
+            sa.Column(
+                "step_run_id", sqlmodel.sql.sqltypes.GUID(), nullable=True
+            )
+        )
+        batch_op.alter_column(
+            "snapshot_id", existing_type=sa.CHAR(length=32), nullable=True
+        )
+        batch_op.drop_constraint(
+            "unique_step_name_for_snapshot", type_="unique"
+        )
+        batch_op.create_unique_constraint(
+            "unique_step_name_for_snapshot",
+            ["snapshot_id", "step_run_id", "name"],
+        )
+        batch_op.create_foreign_key(
+            "fk_step_configuration_step_run_id_step_run",
+            "step_run",
+            ["step_run_id"],
+            ["id"],
+            ondelete="CASCADE",
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("step_configuration", schema=None) as batch_op:
+        batch_op.drop_constraint(
+            "fk_step_configuration_step_run_id_step_run", type_="foreignkey"
+        )
+        batch_op.drop_constraint(
+            "unique_step_name_for_snapshot", type_="unique"
+        )
+        batch_op.create_unique_constraint(
+            "unique_step_name_for_snapshot", ["snapshot_id", "name"]
+        )
+        batch_op.alter_column(
+            "snapshot_id", existing_type=sa.CHAR(length=32), nullable=False
+        )
+        batch_op.drop_column("step_run_id")
+
+    with op.batch_alter_table("pipeline_snapshot", schema=None) as batch_op:
+        batch_op.drop_column("is_dynamic")
+
+    # ### end Alembic commands ###