cr-connect-workflow/migrations/versions/44dd9397c555_new workflow s...

29 lines
812 B
Python

"""new workflow status
Revision ID: 44dd9397c555
Revises: d830959e96c0
Create Date: 2021-12-09 14:03:45.526308
"""
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = '44dd9397c555'
down_revision = 'd830959e96c0'
branch_labels = None
depends_on = None
def upgrade():
op.execute("ALTER TYPE WorkflowStatus ADD VALUE 'erroring'")
def downgrade():
op.execute("UPDATE workflow set status='waiting' WHERE status = 'erroring'")
op.execute('ALTER TYPE WorkflowStatus RENAME TO ws_old;')
op.execute("CREATE TYPE WorkflowStatus AS ENUM('not_started', 'user_input_required', 'waiting', 'complete')")
op.execute("ALTER TABLE workflow ALTER COLUMN status TYPE workflowstatus USING status::text::workflowstatus;")
op.execute('DROP TYPE ws_old;')