Skip to content

Commit 4573217

Browse files
add migration script
1 parent 5807f81 commit 4573217

File tree

1 file changed

+53
-0
lines changed

1 file changed

+53
-0
lines changed
Lines changed: 53 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,53 @@
1+
"""Migrate workbench
2+
3+
Revision ID: e11f9d1e3f44
4+
Revises: 307017ee1a49
5+
Create Date: 2025-01-07 10:16:43.305789+00:00
6+
7+
"""
8+
import sqlalchemy as sa
9+
from alembic import op
10+
from sqlalchemy.dialects import postgresql
11+
12+
# revision identifiers, used by Alembic.
13+
revision = "e11f9d1e3f44"
14+
down_revision = "307017ee1a49"
15+
branch_labels = None
16+
depends_on = None
17+
18+
19+
def upgrade():
20+
# ### commands auto generated by Alembic - please adjust! ###
21+
op.add_column("projects_nodes", sa.Column("key", sa.String(), nullable=False))
22+
op.add_column("projects_nodes", sa.Column("version", sa.String(), nullable=False))
23+
op.add_column("projects_nodes", sa.Column("label", sa.String(), nullable=False))
24+
op.add_column("projects_nodes", sa.Column("progress", sa.Numeric(), nullable=True))
25+
op.add_column("projects_nodes", sa.Column("thumbnail", sa.String(), nullable=False))
26+
op.add_column(
27+
"projects_nodes",
28+
sa.Column("inputs", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
29+
)
30+
op.add_column(
31+
"projects_nodes",
32+
sa.Column("outputs", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
33+
)
34+
op.add_column("projects_nodes", sa.Column("run_hash", sa.String(), nullable=True))
35+
op.add_column(
36+
"projects_nodes",
37+
sa.Column("state", postgresql.JSONB(astext_type=sa.Text()), nullable=True),
38+
)
39+
# ### end Alembic commands ###
40+
41+
42+
def downgrade():
43+
# ### commands auto generated by Alembic - please adjust! ###
44+
op.drop_column("projects_nodes", "state")
45+
op.drop_column("projects_nodes", "run_hash")
46+
op.drop_column("projects_nodes", "outputs")
47+
op.drop_column("projects_nodes", "inputs")
48+
op.drop_column("projects_nodes", "thumbnail")
49+
op.drop_column("projects_nodes", "progress")
50+
op.drop_column("projects_nodes", "label")
51+
op.drop_column("projects_nodes", "version")
52+
op.drop_column("projects_nodes", "key")
53+
# ### end Alembic commands ###

0 commit comments

Comments
 (0)