fix workflow generate_script db migration (#3159)
This commit is contained in:
@@ -21,13 +21,31 @@ depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
def upgrade() -> None:
|
||||
# ### commands auto generated by Alembic - please adjust! ###
|
||||
op.add_column("workflows", sa.Column("generate_script", sa.Boolean(), nullable=False))
|
||||
op.add_column("workflows", sa.Column("generate_script", sa.Boolean(), nullable=False, server_default=sa.false()))
|
||||
|
||||
# Copy data from use_cache to generate_script
|
||||
op.execute("UPDATE workflows SET generate_script = use_cache")
|
||||
|
||||
# Remove server default after data migration
|
||||
op.alter_column("workflows", "generate_script", server_default=None)
|
||||
|
||||
# Drop the old use_cache column
|
||||
op.drop_column("workflows", "use_cache")
|
||||
|
||||
# ### end Alembic commands ###
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
# ### commands auto generated by Alembic - please adjust! ###
|
||||
op.add_column("workflows", sa.Column("use_cache", sa.BOOLEAN(), autoincrement=False, nullable=False))
|
||||
# Add back use_cache column
|
||||
op.add_column("workflows", sa.Column("use_cache", sa.Boolean(), nullable=False, server_default=sa.false()))
|
||||
|
||||
# Copy data from generate_script back to use_cache
|
||||
op.execute("UPDATE workflows SET use_cache = generate_script")
|
||||
|
||||
# Remove server default after data migration
|
||||
op.alter_column("workflows", "use_cache", server_default=None)
|
||||
|
||||
# Drop the new generate_script column
|
||||
op.drop_column("workflows", "generate_script")
|
||||
# ### end Alembic commands ###
|
||||
|
||||
Reference in New Issue
Block a user