|
| 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