🗃 — Fix some migrations for SQLite

This commit is contained in:
Luc Didry 2024-04-08 11:21:57 +02:00
parent a9b7b8f750
commit fa4de8a2b7
No known key found for this signature in database
GPG key ID: EA868E12D0257E3C
2 changed files with 9 additions and 3 deletions

View file

@ -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"]
)

View file

@ -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(