fix workflow generate_script db migration (#3159)

This commit is contained in:
Shuchang Zheng
2025-08-10 18:09:24 -07:00
committed by GitHub
parent 58bd43171e
commit eaf7009d30

View File

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