From eaf7009d3016b81407ea44a28902debd0362fc57 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sun, 10 Aug 2025 18:09:24 -0700 Subject: [PATCH] fix workflow generate_script db migration (#3159) --- ...e5a8_workflow_use_cache_generate_script.py | 22 +++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) diff --git a/alembic/versions/2025_08_10_1557-944ef972e5a8_workflow_use_cache_generate_script.py b/alembic/versions/2025_08_10_1557-944ef972e5a8_workflow_use_cache_generate_script.py index 966b915a..0ea80150 100644 --- a/alembic/versions/2025_08_10_1557-944ef972e5a8_workflow_use_cache_generate_script.py +++ b/alembic/versions/2025_08_10_1557-944ef972e5a8_workflow_use_cache_generate_script.py @@ -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 ###