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(