Repository: incubator-airflow Updated Branches: refs/heads/master 1475e6707 -> f271d437a
[AIRFLOW-1744] Make sure max_tries can be set task.retries can be False. Which is not acceptable for and integer field. Closes #2713 from bolkedebruin/AIRFLOW-1744 Project: http://git-wip-us.apache.org/repos/asf/incubator-airflow/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-airflow/commit/f271d437 Tree: http://git-wip-us.apache.org/repos/asf/incubator-airflow/tree/f271d437 Diff: http://git-wip-us.apache.org/repos/asf/incubator-airflow/diff/f271d437 Branch: refs/heads/master Commit: f271d437a6d168521d6d6b06e43ce9262ba3dccf Parents: 1475e67 Author: Bolke de Bruin <[email protected]> Authored: Tue Oct 24 20:45:44 2017 +0200 Committer: Bolke de Bruin <[email protected]> Committed: Tue Oct 24 20:45:44 2017 +0200 ---------------------------------------------------------------------- .../cc1e65623dc7_add_max_tries_column_to_task_instance.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-airflow/blob/f271d437/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py ---------------------------------------------------------------------- diff --git a/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py b/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py index b151e0c..4cb43b1 100644 --- a/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py +++ b/airflow/migrations/versions/cc1e65623dc7_add_max_tries_column_to_task_instance.py @@ -33,6 +33,7 @@ from sqlalchemy.engine.reflection import Inspector BATCH_SIZE = 5000 + def upgrade(): op.add_column('task_instance', sa.Column('max_tries', sa.Integer, server_default="-1")) @@ -69,8 +70,12 @@ def upgrade(): ti.max_tries = ti.try_number else: task = dag.get_task(ti.task_id) - ti.max_tries = task.retries + if task.retries: + ti.max_tries = task.retries + else: + ti.max_tries = ti.try_number session.merge(ti) + session.commit() # Commit the current session. session.commit()
