Skip to content

Commit 5ff5563

Browse files
committed
fixup! Restore external_trigger value when downgrade
1 parent b7c925e commit 5ff5563

2 files changed

Lines changed: 10 additions & 1 deletion

File tree

airflow/migrations/versions/0059_3_0_0_remove_external_trigger_field.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,3 +47,12 @@ def downgrade():
4747
"""Unapply remove external_trigger field."""
4848
with op.batch_alter_table("dag_run", schema=None) as batch_op:
4949
batch_op.add_column(sa.Column("external_trigger", sa.BOOLEAN(), autoincrement=False, nullable=True))
50+
# restore external_trigger field based on run_type
51+
dag_run_table = sa.table(
52+
"dag_run", sa.column("external_trigger", sa.BOOLEAN()), sa.column("run_type", sa.String())
53+
)
54+
op.execute(
55+
dag_run_table.update().values(
56+
external_trigger=sa.case([(dag_run_table.c.run_type == "manual", True)], else_=False)
57+
)
58+
)
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
79e5ecaacfd5de52192ca2e28a8586c5fc246c4a6b276930ae149022f5f6d03f
1+
7926c50f5d7526588adbd8017d50097392511315ec78f5a64301dbbf4b166de3

0 commit comments

Comments
 (0)