From fa4de8a2b7b4161b07ff85a7d3925db2851740d9 Mon Sep 17 00:00:00 2001 From: Luc Didry Date: Mon, 8 Apr 2024 11:21:57 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=97=83=20=E2=80=94=20Fix=20some=20migrati?= =?UTF-8?q?ons=20for=20SQLite?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ...fda3f2952d_add_on_delete_cascade_to_results_task_id.py | 8 ++++++-- ...e99bc35702c9_add_severity_to_task_and_add_severity_.py | 4 +++- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/argos/server/migrations/versions/defda3f2952d_add_on_delete_cascade_to_results_task_id.py b/argos/server/migrations/versions/defda3f2952d_add_on_delete_cascade_to_results_task_id.py index 6767757..71795f4 100644 --- a/argos/server/migrations/versions/defda3f2952d_add_on_delete_cascade_to_results_task_id.py +++ b/argos/server/migrations/versions/defda3f2952d_add_on_delete_cascade_to_results_task_id.py @@ -18,16 +18,20 @@ depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: + bind = op.get_bind() with op.batch_alter_table("results", schema=None) as batch_op: - batch_op.drop_constraint("results_task_id_fkey", type_="foreignkey") + if bind.engine.name != "sqlite": + batch_op.drop_constraint("results_task_id_fkey", type_="foreignkey") batch_op.create_foreign_key( "results_task_id_fkey", "tasks", ["task_id"], ["id"], ondelete="CASCADE" ) def downgrade() -> None: + bind = op.get_bind() with op.batch_alter_table("results", schema=None) as batch_op: - batch_op.drop_constraint("results_task_id_fkey", type_="foreignkey") + if bind.engine.name != "sqlite": + batch_op.drop_constraint("results_task_id_fkey", type_="foreignkey") batch_op.create_foreign_key( "results_task_id_fkey", "tasks", ["task_id"], ["id"] ) diff --git a/argos/server/migrations/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py b/argos/server/migrations/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py index 9f117aa..265c20f 100644 --- a/argos/server/migrations/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py +++ b/argos/server/migrations/versions/e99bc35702c9_add_severity_to_task_and_add_severity_.py @@ -19,7 +19,9 @@ depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: - op.execute("ALTER TYPE severity ADD VALUE 'unknown'") + bind = op.get_bind() + if bind.engine.name != "sqlite": + op.execute("ALTER TYPE severity ADD VALUE 'unknown'") op.add_column( "tasks", sa.Column(