add migration script

ITISFoundation · Jan 7, 2025 · 4573217 · 4573217
1 parent 5807f81
commit 4573217
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/...tabase/src/simcore_postgres_database/migration/versions/e11f9d1e3f44_migrate_workbench.py b/...tabase/src/simcore_postgres_database/migration/versions/e11f9d1e3f44_migrate_workbench.py
@@ -0,0 +1,53 @@
+"""Migrate workbench
+
+Revision ID: e11f9d1e3f44
+Revises: 307017ee1a49
+Create Date: 2025-01-07 10:16:43.305789+00:00
+
+"""
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = "e11f9d1e3f44"
+down_revision = "307017ee1a49"
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column("projects_nodes", sa.Column("key", sa.String(), nullable=False))
+    op.add_column("projects_nodes", sa.Column("version", sa.String(), nullable=False))
+    op.add_column("projects_nodes", sa.Column("label", sa.String(), nullable=False))
+    op.add_column("projects_nodes", sa.Column("progress", sa.Numeric(), nullable=True))
+    op.add_column("projects_nodes", sa.Column("thumbnail", sa.String(), nullable=False))
+    op.add_column(
+        "projects_nodes",
+        sa.Column("inputs", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+    )
+    op.add_column(
+        "projects_nodes",
+        sa.Column("outputs", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+    )
+    op.add_column("projects_nodes", sa.Column("run_hash", sa.String(), nullable=True))
+    op.add_column(
+        "projects_nodes",
+        sa.Column("state", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column("projects_nodes", "state")
+    op.drop_column("projects_nodes", "run_hash")
+    op.drop_column("projects_nodes", "outputs")
+    op.drop_column("projects_nodes", "inputs")
+    op.drop_column("projects_nodes", "thumbnail")
+    op.drop_column("projects_nodes", "progress")
+    op.drop_column("projects_nodes", "label")
+    op.drop_column("projects_nodes", "version")
+    op.drop_column("projects_nodes", "key")
+    # ### end Alembic commands ###