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:
|
def upgrade() -> None:
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
# ### 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")
|
op.drop_column("workflows", "use_cache")
|
||||||
|
|
||||||
# ### end Alembic commands ###
|
# ### end Alembic commands ###
|
||||||
|
|
||||||
|
|
||||||
def downgrade() -> None:
|
def downgrade() -> None:
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
# ### 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")
|
op.drop_column("workflows", "generate_script")
|
||||||
# ### end Alembic commands ###
|
# ### end Alembic commands ###
|
||||||
|
|||||||
Reference in New Issue
Block a user